This is an automated email from the ASF dual-hosted git repository.
wangdan pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-pegasus.git
The following commit(s) were added to refs/heads/master by this push:
new 261c95ec4 feat(go-client): Add new HostPort struct according to IDL
(#1946)
261c95ec4 is described below
commit 261c95ec4722384acc46df5f6d8fc068226aa264
Author: Yingchun Lai <[email protected]>
AuthorDate: Thu Mar 14 11:28:07 2024 +0800
feat(go-client): Add new HostPort struct according to IDL (#1946)
https://github.com/apache/incubator-pegasus/issues/1947
---
go-client/idl/base/host_port.go | 104 +++++++++++++++++++++++++++++++++++
go-client/idl/base/host_port_test.go | 51 +++++++++++++++++
2 files changed, 155 insertions(+)
diff --git a/go-client/idl/base/host_port.go b/go-client/idl/base/host_port.go
new file mode 100644
index 000000000..22b721815
--- /dev/null
+++ b/go-client/idl/base/host_port.go
@@ -0,0 +1,104 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package base
+
+import (
+ "fmt"
+
+ "github.com/apache/thrift/lib/go/thrift"
+)
+
+type HostPortType int32
+
+const (
+ HOST_TYPE_INVALID HostPortType = iota
+ HOST_TYPE_IPV4
+ HOST_TYPE_GROUP
+)
+
+type HostPort struct {
+ host string
+ port uint16
+ // TODO(yingchun): Now only support ipv4
+ hpType HostPortType
+}
+
+func NewHostPort(host string, port uint16) *HostPort {
+ return &HostPort{
+ host: host,
+ port: port,
+ hpType: HOST_TYPE_IPV4,
+ }
+}
+
+func (r *HostPort) Read(iprot thrift.TProtocol) error {
+ host, err := iprot.ReadString()
+ if err != nil {
+ return err
+ }
+ port, err := iprot.ReadI16()
+ if err != nil {
+ return err
+ }
+ hpType, err := iprot.ReadByte()
+ if err != nil {
+ return err
+ }
+
+ r.host = host
+ r.port = uint16(port)
+ r.hpType = HostPortType(hpType)
+ return nil
+}
+
+func (r *HostPort) Write(oprot thrift.TProtocol) error {
+ err := oprot.WriteString(r.host)
+ if err != nil {
+ return err
+ }
+ err = oprot.WriteI16(int16(r.port))
+ if err != nil {
+ return err
+ }
+ err = oprot.WriteByte(int8(r.hpType))
+ if err != nil {
+ return err
+ }
+ return nil
+}
+
+func (r *HostPort) GetHost() string {
+ return r.host
+}
+
+func (r *HostPort) GetPort() uint16 {
+ return r.port
+}
+
+func (r *HostPort) String() string {
+ if r == nil {
+ return "<nil>"
+ }
+ return fmt.Sprintf("HostPort(%s:%d)", r.host, r.port)
+}
+
+func (r *HostPort) GetHostPort() string {
+ return fmt.Sprintf("%s:%d", r.host, r.port)
+}
diff --git a/go-client/idl/base/host_port_test.go
b/go-client/idl/base/host_port_test.go
new file mode 100644
index 000000000..e07d3f02f
--- /dev/null
+++ b/go-client/idl/base/host_port_test.go
@@ -0,0 +1,51 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package base
+
+import (
+ "testing"
+
+ "github.com/apache/thrift/lib/go/thrift"
+ "github.com/stretchr/testify/assert"
+)
+
+func TestHostPort(t *testing.T) {
+ testCases := map[string]uint16{
+ "localhost": 8080,
+ }
+
+ for host, port := range testCases {
+ hp := NewHostPort(host, port)
+ assert.Equal(t, host, hp.GetHost())
+ assert.Equal(t, port, hp.GetPort())
+
+ // test HostPort serialize
+ buf := thrift.NewTMemoryBuffer()
+ oprot := thrift.NewTBinaryProtocolTransport(buf)
+ hp.Write(oprot)
+
+ // test HostPort deserialize
+ readHostPort := NewHostPort("", 0)
+ readHostPort.Read(oprot)
+
+ // check equals
+ assert.Equal(t, readHostPort, hp)
+ }
+}
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
