74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
|
package models
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
cmap "github.com/orcaman/concurrent-map"
|
|||
|
|
"github.com/toolkits/pkg/logger"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type AliasTime struct {
|
|||
|
|
Alias string
|
|||
|
|
Time int64
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var AliasMapper = cmap.New()
|
|||
|
|
|
|||
|
|
func UpdateAlias() error {
|
|||
|
|
mapper := AliasMapper.Items()
|
|||
|
|
if len(mapper) == 0 {
|
|||
|
|
logger.Warning("alias mapper is nil, no points push?")
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
now := time.Now().Unix()
|
|||
|
|
|
|||
|
|
// 先清理数据，只保留最近15s的数据
|
|||
|
|
for key, at := range mapper {
|
|||
|
|
if at.(*AliasTime).Time < now-15 {
|
|||
|
|
AliasMapper.Remove(key)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 从数据库获取所有的ident->alias对应关系
|
|||
|
|
dbmap, err := ResourceAliasMapper()
|
|||
|
|
if err != nil {
|
|||
|
|
logger.Warningf("ResourceAliasMapper fail: %v", err)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 从内存里拿到最新的ident->alias对应关系
|
|||
|
|
upmap := AliasMapper.Items()
|
|||
|
|
|
|||
|
|
for key, upval := range upmap {
|
|||
|
|
dbval, has := dbmap[key]
|
|||
|
|
if !has {
|
|||
|
|
// 数据库里没有，写入
|
|||
|
|
err = DBInsertOne(Resource{
|
|||
|
|
Ident: key,
|
|||
|
|
Alias: upval.(*AliasTime).Alias,
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
logger.Errorf("mysql.error: insert resource(ident=%s, alias=%s) fail: %v", key, upval.(*AliasTime).Alias, err)
|
|||
|
|
} else {
|
|||
|
|
// 新资源，默认绑定到id为1的classpath，方便用户管理
|
|||
|
|
if err = ClasspathResourceAdd(1, key); err != nil {
|
|||
|
|
logger.Errorf("bind resource(%s) to classpath(1) fail: %v", key, err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if upval.(*AliasTime).Alias != dbval {
|
|||
|
|
// alias 的值与 DB 中不同，更新
|
|||
|
|
_, err = DB.Exec("UPDATE resource SET alias=? WHERE ident=?", upval.(*AliasTime).Alias, key)
|
|||
|
|
if err != nil {
|
|||
|
|
logger.Errorf("mysql.error: update resource(ident=%s, alias=%s) fail: %v", key, upval.(*AliasTime).Alias, err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|