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
|
||
}
|