250e85732d
new shop system more config options arena & hype per season stats battle pass better variant system complete vbuck & starter pack store fix bugs related to deleting account update launcher endpoints fixed gift loot not deleting
74 lines
1.5 KiB
Go
74 lines
1.5 KiB
Go
package socket
|
|
|
|
import (
|
|
"reflect"
|
|
"sync"
|
|
|
|
"github.com/ectrc/snow/aid"
|
|
"github.com/ectrc/snow/person"
|
|
"github.com/gofiber/contrib/websocket"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type MatchmakerData struct {
|
|
PlaylistID string
|
|
Region string
|
|
}
|
|
|
|
type WebSocket interface {
|
|
WriteMessage(messageType int, data []byte) error
|
|
ReadMessage() (messageType int, p []byte, err error)
|
|
}
|
|
|
|
type Socket[T JabberData | MatchmakerData] struct {
|
|
ID string
|
|
Connection WebSocket
|
|
Data *T
|
|
Person *person.Person
|
|
M sync.Mutex
|
|
}
|
|
|
|
func (s *Socket[T]) Write(payload []byte) {
|
|
s.M.Lock()
|
|
defer s.M.Unlock()
|
|
|
|
err := s.Connection.WriteMessage(websocket.TextMessage, payload)
|
|
if err != nil {
|
|
s.Remove()
|
|
}
|
|
}
|
|
|
|
func (s *Socket[T]) Remove() {
|
|
reflectType := reflect.TypeOf(s.Data).String()
|
|
switch reflectType {
|
|
case "*socket.JabberData":
|
|
JabberSockets.Delete(s.ID)
|
|
case "*socket.MatchmakerData":
|
|
MatchmakerSockets.Delete(s.ID)
|
|
}
|
|
}
|
|
|
|
func newSocket[T JabberData | MatchmakerData](conn WebSocket, data ...T) *Socket[T] {
|
|
additional := data[0]
|
|
|
|
return &Socket[T]{
|
|
ID: uuid.New().String(),
|
|
Connection: conn,
|
|
Data: &additional,
|
|
}
|
|
}
|
|
|
|
func NewJabberSocket(conn WebSocket, id string, data JabberData) *Socket[JabberData] {
|
|
socket := newSocket[JabberData](conn, data)
|
|
socket.ID = id
|
|
return socket
|
|
}
|
|
|
|
func NewMatchmakerSocket(conn *websocket.Conn, data MatchmakerData) *Socket[MatchmakerData] {
|
|
return newSocket[MatchmakerData](conn, data)
|
|
}
|
|
|
|
var (
|
|
JabberSockets = aid.GenericSyncMap[Socket[JabberData]]{}
|
|
MatchmakerSockets = aid.GenericSyncMap[Socket[MatchmakerData]]{}
|
|
) |