58 lines
1.1 KiB
Go
58 lines
1.1 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"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// userid->user 将数据库中的用户信息缓存在内存里，
|
|||
|
|
// 在生成告警事件的时候，根据用户ID快速找到用户的详情
|
|||
|
|
func SyncUsers() {
|
|||
|
|
err := syncUsers()
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("timer: sync users fail:", err)
|
|||
|
|
exit(1)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
go loopSyncUsers()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func loopSyncUsers() {
|
|||
|
|
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 := syncUsers()
|
|||
|
|
if err != nil {
|
|||
|
|
logger.Warning("timer: sync users fail:", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func syncUsers() error {
|
|||
|
|
start := time.Now()
|
|||
|
|
|
|||
|
|
users, err := models.UserGetAll()
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
usersMap := make(map[int64]*models.User)
|
|||
|
|
for i := range users {
|
|||
|
|
usersMap[users[i].Id] = &users[i]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
cache.UserCache.SetAll(usersMap)
|
|||
|
|
logger.Debugf("timer: sync users done, cost: %dms", time.Since(start).Milliseconds())
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|