Use map to improve message delivery time

This commit is contained in:
2024-05-07 20:54:01 +08:00
parent 35f9580499
commit fe27b0bf1c
7 changed files with 6621 additions and 17 deletions

View File

@ -5,13 +5,18 @@ import (
"github.com/gofiber/contrib/websocket"
)
type WsPushRequest struct {
Payload []byte
RecipientID uint
var wsConn = make(map[uint]map[*websocket.Conn]bool)
func ClientRegister(user models.Account, conn *websocket.Conn) {
if wsConn[user.ID] == nil {
wsConn[user.ID] = make(map[*websocket.Conn]bool)
}
wsConn[user.ID][conn] = true
}
var WsConn = make(map[uint][]*websocket.Conn)
func CheckOnline(user models.Account) bool {
return len(WsConn[user.ID]) > 0
func ClientUnregister(user models.Account, conn *websocket.Conn) {
if wsConn[user.ID] == nil {
wsConn[user.ID] = make(map[*websocket.Conn]bool)
}
delete(wsConn[user.ID], conn)
}

View File

@ -41,7 +41,7 @@ func NewNotification(notification models.Notification) error {
func PushNotification(notification models.Notification) error {
raw, _ := jsoniter.Marshal(notification)
for _, conn := range WsConn[notification.RecipientID] {
for conn := range wsConn[notification.RecipientID] {
_ = conn.WriteMessage(1, raw)
}