58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package timer
|
||
|
||
import (
|
||
"fmt"
|
||
"math/rand"
|
||
"time"
|
||
|
||
"github.com/didi/nightingale/v5/cache"
|
||
"github.com/didi/nightingale/v5/models"
|
||
|
||
"github.com/toolkits/pkg/logger"
|
||
)
|
||
|
||
// user_group_id->user_group 将数据库中的用户信息缓存在内存里，
|
||
// 在生成告警事件的时候，根据用户ID快速找到用户的详情
|
||
func SyncUserGroups() {
|
||
err := syncUserGroups()
|
||
if err != nil {
|
||
fmt.Println("timer: sync users fail:", err)
|
||
exit(1)
|
||
}
|
||
|
||
go loopSyncUserGroups()
|
||
}
|
||
|
||
func loopSyncUserGroups() {
|
||
randtime := rand.Intn(9000)
|
||
fmt.Printf("timer: sync users: random sleep %dms\n", randtime)
|
||
time.Sleep(time.Duration(randtime) * time.Millisecond)
|
||
|
||
for {
|
||
time.Sleep(time.Second * time.Duration(9))
|
||
err := syncUserGroups()
|
||
if err != nil {
|
||
logger.Warning("timer: sync users fail:", err)
|
||
}
|
||
}
|
||
}
|
||
|
||
func syncUserGroups() error {
|
||
start := time.Now()
|
||
|
||
userGroups, err := models.UserGroupGetAll()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
userGroupsMap := make(map[int64]*models.UserGroup)
|
||
for i := range userGroups {
|
||
userGroupsMap[userGroups[i].Id] = &userGroups[i]
|
||
}
|
||
|
||
cache.UserGroupCache.SetAll(userGroupsMap)
|
||
logger.Debugf("timer: sync userGroups done, cost: %dms", time.Since(start).Milliseconds())
|
||
|
||
return nil
|
||
}
|