连接数据库并部署到Linux上使用Docker Compose编排的go语言网络聊天室

项目概述

这是一个基于Go语言开发的完整聊天室系统，采用客户端-服务器架构，支持用户注册、登录认证、即时聊天、私聊等功能。项目使用MySQL作为数据存储，通过Docker容器化部署，展现了现代分布式应用的典型设计模式。

技术栈

• 后端语言: Go 1.21+
• 网络通信: TCP Socket
• 数据库: MySQL 8.0
• 容器化: Docker & Docker Compose
• 认证系统: 自定义用户认证
• 并发模型: Goroutine + Channel

项目架构

1. 服务器端 (Server)

主要组件：

聊天室核心 (main.go)

go

type ChatRoom struct { clients map[*Client]bool // 在线客户端映射 broadcast chan BroadcastMessage // 广播消息通道 join chan *Client // 加入通道 leave chan *Client // 离开通道 mutex sync.RWMutex // 读写锁 workerPool chan struct{} // 工作池 }

关键特性：

• 并发安全设计: 使用读写锁保护共享状态
• 消息队列: 缓冲通道避免消息丢失
• 工作池: 控制并发广播任务数
• 优雅退出: 使用通道通知协程退出

认证模块 (auth.go)

go

type AuthResult struct { Success bool Message string User *database.User }

• 用户名/密码验证
• 输入格式校验（正则表达式）
• 错误处理友好提示

2. 数据库层 (database/db.go)

数据模型：

go

type User struct { ID int64 Username string Password string // 明文存储（生产环境应加密） Nickname string CreatedAt time.Time }

核心功能：

• 连接池管理
• 用户CRUD操作
• 唯一性约束检查
• 自动创建测试数据

3. 客户端 (client/main.go)

客户端状态机：

go

type ClientState struct { Authenticated bool // 认证状态 Nickname string // 显示昵称 Username string // 登录用户名 }

交互流程：

1. 连接服务器
2. 认证阶段（登录/注册）
3. 聊天阶段（公聊/私聊）
4. 优雅退出

核心功能实现

1. 用户认证流程

go

// 认证状态转换 未认证 -> [登录/注册] -> 已认证 -> [聊天] -> 退出

注册验证规则：

• 用户名：3-20位字母/数字/下划线
• 密码：至少6位
• 昵称：2-20位中文/字母/数字/下划线

2. 消息广播机制

go

// 广播消息给所有在线用户 func (cr *ChatRoom) sendToAll(msg string, exclude *Client) { cr.mutex.RLock() defer cr.mutex.RUnlock() for client := range cr.clients { if exclude != nil && client == exclude { continue } client.channel <- msg } }

3. 私聊功能

go

// 私聊消息格式 /whisper 目标昵称 消息内容 /w 目标昵称 消息内容（简写）

4. 在线用户管理

go

// 获取在线用户列表 func (cr *ChatRoom) getOnlineUsers() []string { users := make([]string, 0, len(cr.clients)) for client := range cr.clients { users = append(users, client.user.Nickname) } return users }

并发模型设计

1. 基于Channel的消息传递

go

// 三个核心通道 broadcast // 处理广播消息 join // 处理用户加入 leave // 处理用户离开

2. 读写锁保护共享状态

go

// 读操作 - 共享锁 cr.mutex.RLock() // ... 读取操作 cr.mutex.RUnlock() // 写操作 - 排他锁 cr.mutex.Lock() // ... 写入操作 cr.mutex.Unlock()

3. 协程生命周期管理

go

go func() { // 接收消息协程 defer wg.Done() // ... 消息处理 }() go func() { // 发送消息协程 defer wg.Done() // ... 输入处理 }()

部署架构

Docker Compose编排

yaml

services: mysql: # 数据库服务 chatroom-server: # 服务器应用 chatroom-client: # 客户端应用

网络配置：

• 自定义bridge网络chatroom-network
• 容器间通过服务名通信
• 端口映射对外暴露服务

健康检查

yaml

healthcheck: test: ["CMD", "mysqladmin", "ping", "-h", "localhost"] interval: 10s timeout: 5s retries: 10

项目亮点

1. 完整的错误处理

• 网络异常恢复
• 数据库连接重试
• 用户输入验证

2. 资源管理

• 数据库连接池
• 内存泄漏防护
• 协程泄露避免

3. 可观测性

• 连接状态日志
• 在线人数统计
• 消息广播计数

4. 扩展性设计

• 模块化架构
• 配置外部化
• 接口预留

使用指南

启动服务

bash

docker-compose up -d

客户端操作

1. 登录：/login username password
2. 注册：/register username password nickname
3. 查看帮助：/help
4. 私聊：/w nickname message
5. 退出：/exit

性能优化点

1. 消息缓冲: 使用缓冲通道减少阻塞
2. 连接复用: 保持TCP长连接
3. 批量处理: 消息批量广播
4. 内存池: 复用消息对象

安全考虑

1. 输入验证: 防止注入攻击
2. 认证限制: 同一用户多处登录限制
3. 资源限制: 连接数控制
4. 超时机制: 空闲连接清理

扩展方向

1. 消息持久化: 聊天记录存储
2. 文件传输: 支持图片/文件
3. 群组功能: 创建聊天群组
4. Web界面: 提供WebSocket支持
5. 消息加密: 端到端加密通信

总结

这个项目展示了如何使用Go语言构建一个完整的实时聊天系统。它涵盖了：

• 网络编程: TCP Socket通信
• 并发处理: Goroutine和Channel
• 数据存储: MySQL数据库操作
• 系统设计: 模块化架构
• 部署运维: Docker容器化

代码结构清晰，功能完整，既适合学习Go语言的高级特性，也展示了生产级应用的设计思路。通过这个项目，开发者可以掌握构建分布式实时系统所需的核心技能。

项目地址: [你的GitHub仓库链接]
技术栈: Go, MySQL, Docker, TCP/IP
适用场景: 学习项目、原型验证、技术演示