mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2026-01-01 03:03:10 -06:00
Code Issues Packages Projects Releases Wiki Activity

Refactor SharedUsers

Browse source
This commit is contained in:
Neil Alexander 2022-04-06 14:33:14 +01:00
parent 31fa33cbde
commit 4d2d79b4da
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 14 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
syncapi/notifier/notifier.go
View file

@ -22,7 +22,6 @@ import (
"github.com/matrix-org/dendrite/syncapi/storage"
"github.com/matrix-org/dendrite/syncapi/types"
"github.com/matrix-org/gomatrixserverlib"
"github.com/matrix-org/util"
log "github.com/sirupsen/logrus"
)
@ -253,15 +252,25 @@ func (n *Notifier) OnNewPresence(
n.wakeupUsers(sharedUsers, nil, n.currPos)
}
func (n *Notifier) SharedUsers(userID string) (sharedUsers []string) {
func (n *Notifier) SharedUsers(userID string) []string {
n.mapLock.RLock()
defer n.mapLock.RUnlock()
sharedMap := map[string]struct{}{
userID: {},
}
for roomID, users := range n.roomIDToJoinedUsers {
if _, ok := users[userID]; ok {
sharedUsers = append(sharedUsers, n.JoinedUsers(roomID)...)
if _, ok := users[userID]; !ok {
continue
}
for _, userID := range n.JoinedUsers(roomID) {
sharedMap[userID] = struct{}{}
}
}
return util.UniqueStrings(sharedUsers)
sharedUsers := make([]string, 0, len(sharedMap))
for userID := range sharedMap {
sharedUsers = append(sharedUsers, userID)
}
return sharedUsers
}
func (n *Notifier) IsSharedUser(userA, userB string) bool {