message-server/wsClient/client.go

// 客户端管理模块
// 处理单个 WebSocket 客户端的连接、消息读取和发送
package wsClient

import (
	"encoding/json"
	"fmt"

	"github.com/gorilla/websocket" // WebSocket 库
)

// Client represents connection to a single chatting user
// Client 表示一个聊天用户的连接
// 每个 WebSocket 连接对应一个 Client 实例
type Client struct {
	// a socket is the web socket for this uer
	// WebSocket 连接对象，用于与浏览器端进行双向通信
	socket *websocket.Conn

	//receive is a channel to receive messages from other clients
	// 用于接收来自其他客户端的消息的 channel
	// 房间会将广播的消息发送到这个 channel
	receive chan []byte

	//room is where the client is chatting in
	// 客户端所在的聊天房间引用
	// 用于将发送的消息转发给房间
	room *Room

	// 客户端的显示名称（用户名）
	// 在消息中用于标识消息发送者
	name string
}

// read 从 WebSocket 连接读取消息并转发给房间 (原作者说： Used to send messages)
// 这个方法会阻塞运行直到连接断开或出现错误
func (c *Client) read() {
	// close the connection when we are done
	defer c.socket.Close()
	
	// endlessly read messages from input
	for {
		// 从 WebSocket 连接读取消息
		// _ 表示忽略消息类型，msg 是消息内容，err 是错误
		_, msg, err := c.socket.ReadMessage()
		// break if there is an error
		if err != nil {
			return
		}

		// 构造要发送的消息对象，包含用户名和消息内容
		outgoing := map[string]string{
			"name":    c.name,         // 发送者用户名
			"message": string(msg),   // 消息内容（转换为字符串）
		}

		// 将消息对象编码为 JSON 格式
		jsMessage, err := json.Marshal(outgoing)
		if err != nil {
			fmt.Println("Enconding failed!")
			continue
		}

		// 将编码后的消息和发送者信息一起发送到房间的 forward channel
		c.room.forward <- &message{
			content:   jsMessage, // 消息内容
			sender: c,         // 发送者（当前客户端）
		}// 房间会将这条消息分发给除发送者外的所有其他客户端（在 room.go 中 run() 方法 case msg := <-r.forward 中）
	}
}

// write 从 receive channel 读取消息并发送给客户端
// 在独立的 goroutine 中运行，处理向客户端发送消息的任务
func (c *Client) write() {
	defer c.socket.Close()

	// 循环从 receive channel 读取消息
	for msg := range c.receive {
		err := c.socket.WriteMessage(websocket.TextMessage, msg)
		if err != nil {
			return
		}
	}
}