blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-25 19:22:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

msg cache

Browse Source
This commit is contained in:
wangchuxiao 2022-05-24 18:17:50 +08:00
parent c32b53b3c0
commit c176a5b80a
1 changed files with 29 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
internal/rpc/cache/cache.go vendored
View File

@ -153,73 +153,56 @@ func updateAllGroupMemberListToCache() error {
log.NewWarn("0", utils.GetSelfFuncName(), "getAllGroupIDList failed", err.Error()) log.NewWarn("0", utils.GetSelfFuncName(), "getAllGroupIDList failed", err.Error())
panic(err.Error()) panic(err.Error())
} }
wg := &sync.WaitGroup{}
wg.Add(len(groupIDList))
for _, groupID := range groupIDList { for _, groupID := range groupIDList {
go func() { groupMemberIDList, err := imdb.GetGroupMemberIDListByGroupID(groupID)
defer wg.Done() if err != nil {
groupMemberIDList, err := imdb.GetGroupMemberIDListByGroupID(groupID) log.NewWarn("", utils.GetSelfFuncName(), "GetGroupMemberIDListByGroupID", err.Error())
if err != nil { continue
log.NewWarn("", utils.GetSelfFuncName(), "GetGroupMemberIDListByGroupID", err.Error()) }
return if len(groupMemberIDList) > 0 {
if err := db.DB.AddGroupMemberToCache(groupID, groupMemberIDList...); err != nil {
log.NewWarn("", utils.GetSelfFuncName(), "AddGroupMemberToCache", err.Error())
} }
//log.NewDebug("", utils.GetSelfFuncName(), "groupMemberIDList", groupMemberIDList) }
if len(groupMemberIDList) > 0 {
if err := db.DB.AddGroupMemberToCache(groupID, groupMemberIDList...); err != nil {
log.NewWarn("", utils.GetSelfFuncName(), "AddGroupMemberToCache", err.Error())
}
}
}()
} }
wg.Wait()
log.NewInfo("0", utils.GetSelfFuncName(), "ok") log.NewInfo("0", utils.GetSelfFuncName(), "ok")
return nil return nil
} }
func updateAllFriendToCache(userList []db.User) error { func updateAllFriendToCache(userList []db.User) error {
log.NewInfo("0", utils.GetSelfFuncName()) log.NewInfo("0", utils.GetSelfFuncName())
wg := sync.WaitGroup{}
wg.Add(len(userList))
for _, user := range userList { for _, user := range userList {
go func() { friendIDList, err := imdb.GetFriendIDListByUserID(user.UserID)
defer wg.Done() if err != nil {
friendIDList, err := imdb.GetFriendIDListByUserID(user.UserID) log.NewWarn("0", utils.GetSelfFuncName(), err.Error())
if err != nil { continue
log.NewWarn("0", utils.GetSelfFuncName(), err.Error()) }
return if len(friendIDList) > 0 {
if err := db.DB.AddFriendToCache(user.UserID, friendIDList...); err != nil {
log.NewWarn("0", utils.GetSelfFuncName(), err.Error(), friendIDList, user.UserID)
} }
if len(friendIDList) > 0 { }
if err := db.DB.AddFriendToCache(user.UserID, friendIDList...); err != nil {
log.NewWarn("0", utils.GetSelfFuncName(), err.Error(), friendIDList, user.UserID)
}
}
}()
} }
wg.Wait()
log.NewInfo("0", utils.GetSelfFuncName(), "ok") log.NewInfo("0", utils.GetSelfFuncName(), "ok")
return nil return nil
} }
func updateAllBlackListToCache(userList []db.User) error { func updateAllBlackListToCache(userList []db.User) error {
log.NewInfo("0", utils.GetSelfFuncName()) log.NewInfo("0", utils.GetSelfFuncName())
wg := &sync.WaitGroup{}
wg.Add(len(userList))
for _, user := range userList { for _, user := range userList {
go func() { blackIDList, err := imdb.GetBlackIDListByUserID(user.UserID)
defer wg.Done() if err != nil {
blackIDList, err := imdb.GetBlackIDListByUserID(user.UserID) log.NewWarn("", utils.GetSelfFuncName(), err.Error())
if err != nil { continue
log.NewWarn("", utils.GetSelfFuncName(), err.Error()) }
return if len(blackIDList) > 0 {
if err := db.DB.AddBlackUserToCache(user.UserID, blackIDList...); err != nil {
log.NewWarn("0", utils.GetSelfFuncName(), err.Error())
} }
if len(blackIDList) > 0 { }
if err := db.DB.AddBlackUserToCache(user.UserID, blackIDList...); err != nil {
log.NewWarn("0", utils.GetSelfFuncName(), err.Error())
}
}
}()
} }
wg.Wait()
log.NewInfo("0", utils.GetSelfFuncName(), "ok") log.NewInfo("0", utils.GetSelfFuncName(), "ok")
return nil return nil
} }