90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package titansSdk
|
||
|
||
import (
|
||
"context"
|
||
"go-common/app/service/live/resource/api/grpc/v1"
|
||
"go-common/library/ecode"
|
||
"go-common/library/log"
|
||
"go-common/library/net/rpc/warden"
|
||
"sync/atomic"
|
||
"time"
|
||
)
|
||
|
||
// titansConfigSdk 初始化时 单实例对象
|
||
var titansConfigSdk = &titansSdk{}
|
||
|
||
// Config 服务可配置项
|
||
type Config struct {
|
||
// 服务的tree_id
|
||
TreeId int64
|
||
// 缓存的更新间隔，单位为s，不配置则为5s
|
||
Expire int64
|
||
}
|
||
|
||
// titansSdk sdk
|
||
type titansSdk struct {
|
||
client v1.TitansClient
|
||
cache atomic.Value
|
||
}
|
||
|
||
// Init 业务初始化sdk接口
|
||
func Init(titansConfig *Config) {
|
||
conf := &warden.ClientConfig{}
|
||
client := warden.NewClient(conf)
|
||
conn, err := client.Dial(context.Background(), "discovery://default/"+v1.AppID)
|
||
if err != nil {
|
||
panic("依赖TitansConfig, 但是Titans的Client没有创建成功！")
|
||
}
|
||
if titansConfig.TreeId == 0 {
|
||
panic("依赖了TitansConfig, 但是titansConfig的treeId配置为空！")
|
||
}
|
||
titansConfigSdk = &titansSdk{}
|
||
titansConfigSdk.client = v1.NewTitansClient(conn)
|
||
titansConfigSdk.cache.Store(make(map[string]string))
|
||
load(titansConfigSdk, titansConfig.TreeId)
|
||
go update(titansConfigSdk, titansConfig.TreeId, titansConfig.Expire)
|
||
}
|
||
|
||
// update 定时更新
|
||
func update(sdk *titansSdk, treeId int64, expire int64) {
|
||
if expire < 1 {
|
||
expire = 5
|
||
}
|
||
for {
|
||
load(sdk, treeId)
|
||
time.Sleep(time.Duration(expire) * time.Second)
|
||
}
|
||
}
|
||
|
||
// load grpc接口
|
||
func load(sdk *titansSdk, treeId int64) {
|
||
for {
|
||
resp, err := sdk.client.GetByTreeId(context.Background(), &v1.TreeIdReq{TreeId: treeId})
|
||
if err != nil {
|
||
log.Error("[SyncTitansConfig][call resource][error], err:%+v", err)
|
||
} else {
|
||
sdk.cache.Store(resp.List)
|
||
break
|
||
}
|
||
time.Sleep(1 * time.Second)
|
||
}
|
||
}
|
||
|
||
// Get 获取配置接口
|
||
func Get(keyword string) (res string, err error) {
|
||
res = ""
|
||
cache, ok := titansConfigSdk.cache.Load().(map[string]string)
|
||
if !ok {
|
||
log.Error("[GetTitansConfig][cache content exception][error] content assert failed")
|
||
err = ecode.GetConfAdminErr
|
||
return
|
||
}
|
||
res, ok = cache[keyword]
|
||
if !ok || res == "" {
|
||
log.Error("[GetTitansConfig][cache content empty][Warn], keyword:%s, cache:%v", keyword, cache)
|
||
err = ecode.GetConfAdminErr
|
||
return
|
||
}
|
||
return
|
||
}
|