68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"git.apinb.com/bsm-sdk/engine/etcd"
|
||
|
|
"git.apinb.com/bsm-sdk/engine/print"
|
||
|
|
"git.apinb.com/bsm-sdk/engine/types"
|
||
|
|
"git.apinb.com/bsm-sdk/engine/vars"
|
||
|
|
"github.com/zeromicro/go-zero/zrpc"
|
||
|
|
)
|
||
|
|
|
||
|
|
func RegisterAnonymous(c *zrpc.RpcServerConf, anonCfg *types.AnonymousConf) {
|
||
|
|
// etcd is has
|
||
|
|
if !c.HasEtcd() {
|
||
|
|
return
|
||
|
|
} else {
|
||
|
|
print.Info("[Blocks Register] %s ETCD Endpoints: %s", vars.ServiceKey, c.Etcd.Hosts)
|
||
|
|
print.Success("[Blocks Register] Service: %s", c.Etcd.Key)
|
||
|
|
}
|
||
|
|
|
||
|
|
// anonymous slice is zero
|
||
|
|
if len(anonCfg.Urls) == 0 || anonCfg.Key == "" {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// connect etcd
|
||
|
|
var client *etcd.Etcd
|
||
|
|
if c.Etcd.HasTLS() {
|
||
|
|
client = etcd.NewEtcd(c.Etcd.Hosts, &types.EtcdTls{
|
||
|
|
Ca: c.Etcd.CACertFile,
|
||
|
|
Cert: c.Etcd.CertFile,
|
||
|
|
CertKey: c.Etcd.CertKeyFile,
|
||
|
|
})
|
||
|
|
} else {
|
||
|
|
client = etcd.NewEtcd(c.Etcd.Hosts, nil)
|
||
|
|
}
|
||
|
|
defer client.Client.Close()
|
||
|
|
|
||
|
|
// remove reppeat, clear service all anonymous uri.
|
||
|
|
anonymous, _, _ := client.Get(anonCfg.Key)
|
||
|
|
as := strings.Split(anonymous, ",")
|
||
|
|
as = append(clearService(as), anonCfg.Urls...)
|
||
|
|
newAnonymous := strings.Join(as, ",")
|
||
|
|
|
||
|
|
// put anonymous to etcd
|
||
|
|
_, err := client.Client.Put(context.Background(), anonCfg.Key, newAnonymous)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
print.Fail("[Blocks Register] Anonymous Fail.")
|
||
|
|
} else {
|
||
|
|
print.Success("[Blocks Register] Anonymous: %s", newAnonymous)
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func clearService(as []string) (out []string) {
|
||
|
|
for _, v := range as {
|
||
|
|
if len(v) >= len(vars.ServiceKey) {
|
||
|
|
if v[0:len(vars.ServiceKey)] != vars.ServiceKey {
|
||
|
|
out = append(out, v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|