yanweidong
dbf68c38c1
docs(readme): 重构 README 文档结构并补充模块说明 - 重新组织 README 内容，明确划分私有仓库设置、功能模块等章节 - 补充 crypto、cache、database、middleware、queue 和 utils 模块的功能简介 - 统一代码示例格式，增加 bash 和 go 语言标识 feat(crypto): 优化 PKCS7 填充与去填充函数实现 - 新增 PKCS7Padding 和 PKCS7UnPadding 函数的详细注释 - 添加对输入参数的有效性校验，提升健壮性 - 修复可能引发越界 panic 的潜在问题 feat(database): 完善数据库连接及初始化逻辑 - 为 NewDatabase、NewMysql 和 NewPostgres 函数添加完整注释 - 修复 MaxOpenConns 配置未正确赋值的问题 - 在获取 *sql.DB 实例时增加错误处理逻辑 - 支持通过 Init 变量在连接建立后执行自定义初始化函数 feat(go.mod): 初始化项目依赖管理并引入核心组件 - 添加项目所需的主要依赖包，包括 gin、gorm、redis、nats 等 - 引入常用的工具库如 uuid、ulid、gopsutil 等 - 自动拉取并锁定所有间接依赖版本 ```
133 lines
2.9 KiB
Go
133 lines
2.9 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"time"
|
|
|
|
"git.apinb.com/bsm-sdk/core/print"
|
|
"git.apinb.com/bsm-sdk/core/vars"
|
|
clientv3 "go.etcd.io/etcd/client/v3"
|
|
)
|
|
|
|
// ServiceRegister 创建租约注册服务
|
|
type ServiceRegister struct {
|
|
cli *clientv3.Client //etcd client
|
|
leaseID clientv3.LeaseID //租约ID
|
|
//租约keepalieve相应chan
|
|
keepAliveChan <-chan *clientv3.LeaseKeepAliveResponse
|
|
key string //key
|
|
val string //value
|
|
}
|
|
|
|
// NewRegister 注册服务至路由表.
|
|
//
|
|
// ec:EtcdConfig;
|
|
//
|
|
// routeKey: ServiceRouteRootPrefix + ServiceKey + "/" + utils.GetLocationIP() + addr;
|
|
//
|
|
// gs:grpc.Server;
|
|
func RegisterService(cli *clientv3.Client, routeKey string, methods []string, lease int64) (*ServiceRegister, error) {
|
|
|
|
ser := &ServiceRegister{
|
|
cli: cli,
|
|
key: routeKey,
|
|
val: strings.Join(methods, ","),
|
|
}
|
|
|
|
//申请租约设置时间keepalive
|
|
if err := ser.putKeyWithLease(lease); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ser, nil
|
|
}
|
|
|
|
// 设置租约
|
|
func (s *ServiceRegister) putKeyWithLease(lease int64) error {
|
|
//设置租约时间
|
|
resp, err := s.cli.Grant(context.Background(), lease)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
//注册服务并绑定租约
|
|
_, err = s.cli.Put(context.Background(), s.key, s.val, clientv3.WithLease(resp.ID))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
//设置续租 定期发送需求请求
|
|
leaseRespChan, err := s.cli.KeepAlive(context.Background(), resp.ID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
s.leaseID = resp.ID
|
|
s.keepAliveChan = leaseRespChan
|
|
return nil
|
|
}
|
|
|
|
// ListenLeaseRespChan 监听 续租情况
|
|
func (s *ServiceRegister) ListenLeaseRespChan() {
|
|
for {
|
|
select {
|
|
case leaseKeepResp := <-s.keepAliveChan:
|
|
if leaseKeepResp == nil {
|
|
//log.Println("close lease.")
|
|
return
|
|
} else {
|
|
goto END
|
|
}
|
|
}
|
|
END:
|
|
time.Sleep(500 * time.Millisecond)
|
|
}
|
|
}
|
|
|
|
// Close 注销服务
|
|
func (s *ServiceRegister) Close() error {
|
|
//撤销租约
|
|
if _, err := s.cli.Revoke(context.Background(), s.leaseID); err != nil {
|
|
return err
|
|
}
|
|
return s.cli.Close()
|
|
}
|
|
|
|
func (s *ServiceRegister) SetAnonymous(key string, urls []string) {
|
|
// remove reppeat, clear service all anonymous uri.
|
|
anonymous, err := s.cli.Get(context.Background(), key)
|
|
if err != nil {
|
|
print.Error("[BSM Register] Get Anonymous Fail: %v", err)
|
|
return
|
|
}
|
|
|
|
var as []string
|
|
if len(anonymous.Kvs) > 0 {
|
|
val := string(anonymous.Kvs[0].Value)
|
|
as = strings.Split(val, ",")
|
|
}
|
|
|
|
as = append(clearService(as), urls...)
|
|
newAnonymous := strings.Join(as, ",")
|
|
|
|
// put anonymous to etcd
|
|
_, err = s.cli.Put(context.Background(), key, newAnonymous)
|
|
|
|
if err != nil {
|
|
print.Error("[BSM Register] Anonymous Fail.")
|
|
} else {
|
|
print.Info("[BSM 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
|
|
}
|