62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
|
package ws
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
|
|||
|
|
chatmodel "leke/ws/model"
|
|||
|
|
|
|||
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Client 表示一个在线用户的 WebSocket 连接
|
|||
|
|
// 这里只是框架：先把字段和发送逻辑搭好，读写循环可以后面慢慢加。
|
|||
|
|
type Client struct {
|
|||
|
|
UserId uint64 // 当前用户ID
|
|||
|
|
Nickname string // 昵称（可选）
|
|||
|
|
Conn *ghttp.Request // 底层 WebSocket 连接
|
|||
|
|
|
|||
|
|
// Send 是一个发送队列：其他地方往这里丢 []byte，这个 client 的写协程负责发出去
|
|||
|
|
Send chan []byte
|
|||
|
|
|
|||
|
|
// Rooms 记录当前用户加入的房间（简单用 map 做集合）
|
|||
|
|
Rooms map[string]bool
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewClient 创建一个新的客户端连接对象
|
|||
|
|
func NewClient(userId uint64, nickname string, conn *ghttp.Request) *Client {
|
|||
|
|
return &Client{
|
|||
|
|
UserId: userId,
|
|||
|
|
Nickname: nickname,
|
|||
|
|
Conn: conn,
|
|||
|
|
Send: make(chan []byte, 256), // 简单先给一个缓冲，防止轻微阻塞
|
|||
|
|
Rooms: make(map[string]bool),
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// EnqueueMessage 将 ChatMessage 编码为 JSON 丢到 Send 队列
|
|||
|
|
// 真正 WriteMessage 的动作建议在一个单独的 writeLoop 里做。
|
|||
|
|
func (c *Client) EnqueueMessage(msg *chatmodel.ChatMessage) error {
|
|||
|
|
// 如果需要，补一些默认字段
|
|||
|
|
if msg.FromUserId == 0 {
|
|||
|
|
msg.FromUserId = c.UserId
|
|||
|
|
}
|
|||
|
|
if msg.FromNickname == "" {
|
|||
|
|
msg.FromNickname = c.Nickname
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
data, err := json.Marshal(msg)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
select {
|
|||
|
|
case c.Send <- data:
|
|||
|
|
// 正常入队
|
|||
|
|
default:
|
|||
|
|
// 队列满了，可以考虑：丢弃 / 断开连接 / 打日志
|
|||
|
|
// 这里简单选择丢弃，并返回错误
|
|||
|
|
return ErrSendQueueFull
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|