133 lines
2.6 KiB
Go
133 lines
2.6 KiB
Go
|
package ws
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"sync"
|
|||
|
|
|
|||
|
|
chatmodel "leke/ws/model" // 按实际路径修改
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Hub 聊天中枢：管理所有连接和房间
|
|||
|
|
type Hub struct {
|
|||
|
|
mu sync.RWMutex
|
|||
|
|
|
|||
|
|
// 所有在线用户：userId -> Client
|
|||
|
|
clients map[uint64]*Client
|
|||
|
|
|
|||
|
|
// 房间成员：roomId -> (userId -> Client)
|
|||
|
|
rooms map[string]map[uint64]*Client
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 全局唯一 Hub（简单起见用一个全局变量）
|
|||
|
|
var ChatHub = NewHub()
|
|||
|
|
|
|||
|
|
// NewHub 创建一个新的 Hub 实例
|
|||
|
|
func NewHub() *Hub {
|
|||
|
|
return &Hub{
|
|||
|
|
clients: make(map[uint64]*Client),
|
|||
|
|
rooms: make(map[string]map[uint64]*Client),
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Register 注册新连接
|
|||
|
|
func (h *Hub) Register(c *Client) {
|
|||
|
|
h.mu.Lock()
|
|||
|
|
defer h.mu.Unlock()
|
|||
|
|
|
|||
|
|
h.clients[c.UserId] = c
|
|||
|
|
// 默认认为所有在线用户都在世界频道，这里你可以根据需要扩展
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Unregister 移除连接（断线/退出）
|
|||
|
|
func (h *Hub) Unregister(c *Client) {
|
|||
|
|
h.mu.Lock()
|
|||
|
|
defer h.mu.Unlock()
|
|||
|
|
|
|||
|
|
delete(h.clients, c.UserId)
|
|||
|
|
|
|||
|
|
// 同时把他从所有房间移除
|
|||
|
|
for roomId, members := range h.rooms {
|
|||
|
|
if _, ok := members[c.UserId]; ok {
|
|||
|
|
delete(members, c.UserId)
|
|||
|
|
if len(members) == 0 {
|
|||
|
|
delete(h.rooms, roomId)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// JoinRoom 加入房间
|
|||
|
|
func (h *Hub) JoinRoom(userId uint64, roomId string) {
|
|||
|
|
h.mu.Lock()
|
|||
|
|
defer h.mu.Unlock()
|
|||
|
|
|
|||
|
|
client, ok := h.clients[userId]
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
members, ok := h.rooms[roomId]
|
|||
|
|
if !ok {
|
|||
|
|
members = make(map[uint64]*Client)
|
|||
|
|
h.rooms[roomId] = members
|
|||
|
|
}
|
|||
|
|
members[userId] = client
|
|||
|
|
client.Rooms[roomId] = true
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// LeaveRoom 离开房间
|
|||
|
|
func (h *Hub) LeaveRoom(userId uint64, roomId string) {
|
|||
|
|
h.mu.Lock()
|
|||
|
|
defer h.mu.Unlock()
|
|||
|
|
|
|||
|
|
client, ok := h.clients[userId]
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if members, ok := h.rooms[roomId]; ok {
|
|||
|
|
delete(members, userId)
|
|||
|
|
if len(members) == 0 {
|
|||
|
|
delete(h.rooms, roomId)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
delete(client.Rooms, roomId)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// BroadcastWorld 向所有在线用户发送世界频道消息
|
|||
|
|
func (h *Hub) BroadcastWorld(msg *chatmodel.ChatMessage) {
|
|||
|
|
h.mu.RLock()
|
|||
|
|
defer h.mu.RUnlock()
|
|||
|
|
|
|||
|
|
for _, client := range h.clients {
|
|||
|
|
_ = client.EnqueueMessage(msg)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// BroadcastRoom 向房间内所有用户发送消息
|
|||
|
|
func (h *Hub) BroadcastRoom(roomId string, msg *chatmodel.ChatMessage) {
|
|||
|
|
h.mu.RLock()
|
|||
|
|
defer h.mu.RUnlock()
|
|||
|
|
|
|||
|
|
members, ok := h.rooms[roomId]
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
for _, client := range members {
|
|||
|
|
_ = client.EnqueueMessage(msg)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SendPrivate 发送私聊消息
|
|||
|
|
func (h *Hub) SendPrivate(fromUserId, toUserId uint64, msg *chatmodel.ChatMessage) {
|
|||
|
|
h.mu.RLock()
|
|||
|
|
defer h.mu.RUnlock()
|
|||
|
|
|
|||
|
|
toClient, ok := h.clients[toUserId]
|
|||
|
|
if !ok {
|
|||
|
|
return // 对方不在线，可以后面扩展：离线消息存DB
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
msg.FromUserId = fromUserId
|
|||
|
|
_ = toClient.EnqueueMessage(msg)
|
|||
|
|
}
|