errcode

pkg/discovery_registry/discovery_register.go

@@ -15,8 +15,12 @@ import (
 )
 
 type SvcDiscoveryRegistry interface {
+	Register(serviceName, host string, port int, opts ...grpc.DialOption) error
+	UnRegister() error
 	GetConns(serviceName string, opts ...grpc.DialOption) ([]*grpc.ClientConn, error)
 	GetConn(serviceName string, strategy func(slice []*grpc.ClientConn) int, opts ...grpc.DialOption) (*grpc.ClientConn, error)
+	//RegisterConf(conf []byte) error
+	//LoadConf() ([]byte, error)
 }
 
 func registerConf(key, conf string) {

pkg/discovery_registry/startegy.go

@@ -0,0 +1,28 @@
+package discoveryRegistry
+
+import "google.golang.org/grpc"
+
+type Robin struct {
+	next int
+}
+
+func (r *Robin) Robin(slice []*grpc.ClientConn) int {
+	index := r.next
+	r.next += 1
+	if r.next > len(slice)-1 {
+		r.next = 0
+	}
+	return index
+}
+
+type Hash struct {
+}
+
+func (r *Hash) Hash(slice []*grpc.ClientConn) int {
+	index := r.next
+	r.next += 1
+	if r.next > len(slice)-1 {
+		r.next = 0
+	}
+	return index
+}

pkg/discovery_registry/zk.go

@@ -0,0 +1 @@
+package discoveryRegistry