errcode

pkg/discovery_registry/discovery_register.go

@@ -0,0 +1,54 @@
+package discoveryRegistry
+
+import (
+	"Open_IM/pkg/common/config"
+	"Open_IM/pkg/utils"
+	"context"
+	"fmt"
+	"github.com/OpenIMSDK/getcdv3"
+	clientv3 "go.etcd.io/etcd/client/v3"
+	"google.golang.org/grpc"
+	"time"
+
+	"gopkg.in/yaml.v3"
+	"strings"
+)
+
+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) {
+	etcdAddr := strings.Join(config.Config.Etcd.EtcdAddr, ",")
+	cli, err := clientv3.New(clientv3.Config{
+		Endpoints: strings.Split(etcdAddr, ","), DialTimeout: 5 * time.Second})
+
+	if err != nil {
+		panic(err.Error())
+	}
+	//lease
+	if _, err := cli.Put(context.Background(), key, conf); err != nil {
+		fmt.Println("panic, params: ")
+		panic(err.Error())
+	}
+}
+
+func RegisterConf() {
+	bytes, err := yaml.Marshal(config.Config)
+	if err != nil {
+		panic(err.Error())
+	}
+	secretMD5 := utils.Md5(config.Config.Etcd.Secret)
+	confBytes, err := utils.AesEncrypt(bytes, []byte(secretMD5[0:16]))
+	if err != nil {
+		panic(err.Error())
+	}
+	fmt.Println("start register", secretMD5, getcdv3.GetPrefix(config.Config.Etcd.EtcdSchema, config.ConfName))
+	registerConf(getcdv3.GetPrefix(config.Config.Etcd.EtcdSchema, config.ConfName), string(confBytes))
+	fmt.Println("etcd register conf ok")
+}