mirror of
https://github.com/oarkflow/mq.git
synced 2025-10-01 06:12:10 +08:00
32 lines
514 B
Go
32 lines
514 B
Go
package dag
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/oarkflow/mq/sio"
|
|
)
|
|
|
|
func WsEvents(s *sio.Server) {
|
|
s.On("join", join)
|
|
s.On("message", message)
|
|
}
|
|
|
|
func join(s *sio.Socket, data []byte) {
|
|
currentRooms := s.GetRooms()
|
|
for _, room := range currentRooms {
|
|
s.Leave(room)
|
|
}
|
|
s.Join(string(data))
|
|
s.Emit("joinedRoom", string(data))
|
|
}
|
|
|
|
type msg struct {
|
|
Room string
|
|
Message string
|
|
}
|
|
|
|
func message(s *sio.Socket, data []byte) {
|
|
var m msg
|
|
json.Unmarshal(data, &m)
|
|
s.ToRoom(m.Room, "message", m.Message)
|
|
}
|