HyperNet/Messaging
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

✨ Disconnect websocket auto cleaning up subscribed channels

Browse Source
This commit is contained in:
LittleSheep
2025-03-01 18:02:55 +08:00
parent e3a4988ccf
commit eea9b85745
3 changed files with 31 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
pkg/internal/services/subscribe.go
View File

@ -2,8 +2,8 @@ package services
import "sync"
// ChannelID -> UserID
var subscribeInfo = make(map[uint]map[uint]bool)
// ChannelID -> UserID -> Client ID
var subscribeInfo = make(map[uint]map[uint]string)
var subscribeLock sync.Mutex
// If user subscribed to a channel
@ -19,13 +19,13 @@ func CheckSubscribed(UserID uint, ChannelID uint) bool {
return false
}
func SubscribeChannel(userId uint, channelId uint) {
func SubscribeChannel(userId uint, channelId uint, clientId string) {
subscribeLock.Lock()
defer subscribeLock.Unlock()
if _, ok := subscribeInfo[channelId]; !ok {
subscribeInfo[channelId] = make(map[uint]bool)
subscribeInfo[channelId] = make(map[uint]string)
}
subscribeInfo[channelId][userId] = true
subscribeInfo[channelId][userId] = clientId
}
func UnsubscribeChannel(userId uint, channelId uint) {
@ -49,3 +49,15 @@ func UnsubscribeAllWithChannels(channelId uint) {
defer subscribeLock.Unlock()
delete(subscribeInfo, channelId)
}
func UnsubscribeAllWithClient(clientId string) {
subscribeLock.Lock()
defer subscribeLock.Unlock()
for _, v := range subscribeInfo {
for k, item := range v {
if item == clientId {
delete(v, k)
}
}
}
}