【WebSocket】Gin利用WebSocket实现在线聊天室

最新推荐文章于 2025-03-26 19:27:16 发布

原创

最新推荐文章于 2025-03-26 19:27:16 发布

· 3.9k 阅读

10 ·

版权

文章标签：

#golang #websocket

本文详细介绍了如何使用Gin和WebSocket构建在线聊天室，包括实时记录用户在线状态、接收并广播消息以及管理客户端连接。通过定义Client结构体、ClientManager以及相关方法，实现了心跳监测、消息读写和客户端状态检查。同时，文章展示了处理用户注册、广播消息和离线用户删除的流程，确保了实时通讯的稳定性和可靠性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Gin利用WebSocket实现在线聊天室

引言

使用WebSocket实现实时的在线聊天室有以下3个技术难点：
1.实时记录每个用户都在线状态，并实时更新在线用户数量
2.实时接收客户端的消息并广播到所有客户端
3.唯一记录和标识每一个客户端

数据结构的定义

客户端结构体定义

type Client struct {
   
	ID         string
	IpAddress  string
	IpSource   string
	UserId     interface{
   }
	Socket     *websocket.Conn
	Send       chan []byte
	Start      time.Time
	ExpireTime time.Duration // 一段时间没有接收到心跳则过期
}

其中Start是记录客户端最后一次给服务端发送消息的时间，ExpirTime则表示是在ExpirTime时间内如果服务端没有接收客户端发送过来的消息就代表对应客户端已经下线。Socket则是用来记录客户端和服务端直接到WebSocket连接。

管理使用在线客户端结构体定义

type ClientManager struct {
   
	Clients    map[string]*Client // 记录在线用户
	Broadcast  chan []byte //触发消息广播
	Register   chan *Client // 触发新用户登陆
	UnRegister chan *Client // 触发用户退出
}

由于在本案例中每个客户端都有自己独立的goroutine，所以利用channel保证多个goroutine之间的正常通信，进而保证每个客户端之间正常的消息通讯

消息体的定义

type WsMessage struct {
   
	Type int         `json:"type"`
	Data interface{
   } `json:"data"`
}

Client对应方法的定义

Read方法的定义

// Read 读取客户端发送过来的消息
func (c *Client) Read() {
   
	// 出现故障后把当前客户端注销
	defer func() {
   
		_ = c.Socket.Close()
		Manager.UnRegister <- c
	}()
	for {
   
		_, data, err := c.Socket.ReadMessage()
		if err != nil {
   
			logger.Error(err.Error())
			break
		}
		var msg WsMessage
		err = json.Unmarshal(data, &msg)
		if err != nil {
   
			logger.Error(err.Error())
			break
		}

		switch msg.Type {
   
		case 6:
			// 如果是心跳监测消息（利用心跳监测来判断对应客户端是否在线）
			resp, _ := json.Marshal(&WsMessage