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

⚡ Use map to improve message delivery time

Browse Source
This commit is contained in:
LittleSheep 2024-05-07 20:45:02 +08:00
parent 7c5c9d03d1
commit 32be79557a
2 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/server/messages_ws.go
View File

@ -5,14 +5,13 @@ import (
"git.solsynth.dev/hydrogen/messaging/pkg/services"
"github.com/gofiber/contrib/websocket"
jsoniter "github.com/json-iterator/go"
"github.com/samber/lo"
)
func messageGateway(c *websocket.Conn) {
user := c.Locals("principal").(models.Account)
// Push connection
services.WsConn[user.ID] = append(services.WsConn[user.ID], c)
services.ClientRegister(user, c)
// Event loop
var task models.UnifiedCommand
@ -42,7 +41,5 @@ func messageGateway(c *websocket.Conn) {
}
// Pop connection
services.WsConn[user.ID] = lo.Filter(services.WsConn[user.ID], func(item *websocket.Conn, idx int) bool {
return item != c
})
services.ClientUnregister(user, c)
}

18
pkg/services/connections.go
View File

@ -5,14 +5,24 @@ import (
"github.com/gofiber/contrib/websocket"
)
var WsConn = make(map[uint][]*websocket.Conn)
var wsConn = make(map[uint]map[*websocket.Conn]bool)
func CheckOnline(user models.Account) bool {
return len(WsConn[user.ID]) > 0
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
}
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)
}
func PushCommand(userId uint, task models.UnifiedCommand) {
for _, conn := range WsConn[userId] {
for conn := range wsConn[userId] {
_ = conn.WriteMessage(1, task.Marshal())
}
}