AI 辅助两天实现 IM 系统开发实战

3.2 消息中台 (ChatHub) 的实现

这是整个 IM 系统的'心脏'，维护着所有在线 Client 的连接。

type ChatHub struct {
	clients    map[uint]*Client
	broadcast  chan []byte
	register   chan *Client
	unregister chan *Client
	mu         sync.RWMutex
	Redis      *redis.Client
	ChatRepo   *repository.ChatRepository
	ctx        context.Context
}

func NewChatHub(rdb *redis.Client, chatRepo *repository.ChatRepository) *ChatHub {
	return &ChatHub{
		broadcast:  make(chan []byte),
		register:   make(chan *Client),
		unregister: make(chan *Client),
		clients:    make(map[uint]*Client),
		Redis:      rdb,
		ChatRepo:   chatRepo,
		ctx:        context.Background(),
	}
}

3.3 分布式推送与在线状态

多实例部署，使用 Redis 进行消息分发。

// PushToUsers 通过 Redis Pub/Sub 将消息发送给指定用户（支持跨实例）
func (h *ChatHub) PushToUsers(userIDs []uint, msg WSMessage) {
	psMsg := PubSubMessage{
		TargetUsers: userIDs,
		Payload:     msg,
	}
	payload, _ := json.Marshal(psMsg)
	h.Redis.Publish(h.ctx, "chat_channel", payload)
}

// pushToLocalUsers 仅将消息发送给连接到'当前实例'的客户端
func (h *ChatHub) pushToLocalUsers(userIDs []uint, msg WSMessage) {
	// ... 锁定 h.clients 并通过 channel 发送消息 ...
}

3.4 连接生命周期管理 (Read & Write Pumps)

每个 WebSocket 连接都会开启两个并发协程：一个负责读，一个负责写。

• ReadPump：解析前端发来的控制指令（如正在输入）。
• WritePump：处理心跳（Ping/Pong）和下发消息队列。

四、关键功能实现细节

4.1 会话管理

支持私聊和群聊。私聊会自动检查是否已存在会话，避免重复创建。群聊则包含角色权限管理（管理员、普通成员）。

4.2 消息实时性与持久化

当用户发送消息时，流程如下：

1. API 调用：前端调用 /api/chat/conversations/:id/messages 发送消息。
2. 后端落库：消息存入 MySQL，确保不丢失。
3. 实时推送：后端调用 Hub.PushToUsers，Hub 将消息通过 WebSocket 实时推送到目标用户的客户端。
4. 回执通知：支持消息已读/未读状态更新，并通过 WS 通知发送方。

4.3 瞬时状态处理 (Ephemeral Events)

对于不需要存库的状态（如 TYPING），系统直接在 Client.readPump 中解析并转发。

// 处理正在输入状态 (TYPING)
if wsMsg.Type == "TYPING" {
	data, ok := wsMsg.Data.(map[string]interface{})
	if !ok {
		continue
	}
	convID, _ := data["conversationId"].(string)
	if convID == "" {
		continue
	}
	// 获取该会话的其他成员并转发
	c.Hub.HandleTransientEvent(c.UserID, convID, wsMsg)
}

五、落地总结

本次 IM 系统的落地，几大技术挑战：

1. 高性能：通过 Go 协程和 Channel 实现了异步非阻塞的消息分发。
2. 分布式扩展：引入 Redis Pub/Sub 解决了单机连接上限问题，系统可水平扩展。
3. 用户体验：引入心跳机制保证连接稳定性，引入在线状态实时通知和输入指示器，提升了产品的互动感。
4. 安全性：在 WebSocket 升级阶段通过 JWT 进行身份校验，确保只有合法用户能建立连接。

该方案支撑了私聊、群聊、好友申请、实时状态更新等核心 IM 功能。