50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
|
package server
|
||
|
|
|
||
|
|
import (
|
||
|
|
"git.solsynth.dev/hydrogen/messaging/pkg/models"
|
||
|
|
"git.solsynth.dev/hydrogen/messaging/pkg/services"
|
||
|
|
"github.com/gofiber/contrib/websocket"
|
||
|
|
jsoniter "github.com/json-iterator/go"
|
||
|
|
)
|
||
|
|
|
||
|
|
func unifiedGateway(c *websocket.Conn) {
|
||
|
|
user := c.Locals("principal").(models.Account)
|
||
|
|
|
||
|
|
// Push connection
|
||
|
|
connectionIdx := len(services.WsConn)
|
||
|
|
services.WsConn[user.ID] = append(services.WsConn[user.ID], c)
|
||
|
|
|
||
|
|
// Event loop
|
||
|
|
var task models.UnifiedCommand
|
||
|
|
|
||
|
|
var messageType int
|
||
|
|
var packet []byte
|
||
|
|
var err error
|
||
|
|
|
||
|
|
for {
|
||
|
|
if messageType, packet, err = c.ReadMessage(); err != nil {
|
||
|
|
break
|
||
|
|
} else if err := jsoniter.Unmarshal(packet, &task); err != nil {
|
||
|
|
_ = c.WriteMessage(messageType, models.UnifiedCommand{
|
||
|
|
Action: "error",
|
||
|
|
Message: "unable to unmarshal your command, requires json request",
|
||
|
|
}.Marshal())
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
message := services.DealCommand(task, user)
|
||
|
|
|
||
|
|
if message != nil {
|
||
|
|
if err = c.WriteMessage(messageType, message.Marshal()); err != nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Pop connection
|
||
|
|
services.WsConn[user.ID] = append(
|
||
|
|
services.WsConn[user.ID][:connectionIdx],
|
||
|
|
services.WsConn[user.ID][connectionIdx+1:]...,
|
||
|
|
)
|
||
|
|
}
|