You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.4 KiB
51 lines
1.4 KiB
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 {
|
|
GetConns(serviceName string, opts ...grpc.DialOption) ([]*grpc.ClientConn, error)
|
|
GetConn(serviceName string, strategy func(slice []*grpc.ClientConn) int, opts ...grpc.DialOption) (*grpc.ClientConn, 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")
|
|
}
|