使用Go的goroutine和channel构建TCP聊天室,通过net.Listen监听端口,每个连接启动goroutine处理。2. 客户端结构体包含连接和用户名,用map维护在线用户,全局channel用于广播消息。3. handleConn读取客户端输入,首行为昵称并通知广播,后续消息带前缀发送至广播通道。4. 另一goroutine监听广播channel,遍历所有客户端连接并发消息,出错时清理失效连接。5. main函数启动服务器并接受连接,使用telnet测试多用户聊天。该实现展示Go并发网络编程核心:轻量协程+channel通信,可扩展私聊、WebSocket等。

在Golang中构建一个小型聊天室,核心是利用其强大的并发模型和net包实现TCP或WebSocket通信。整个系统可以基于客户端-服务器架构,使用goroutine处理每个连接,通过channel进行消息广播。下面分步骤说明如何实现一个基础但完整的命令行聊天室。
1. 设计服务器基本结构
聊天室服务器需要监听端口、接受多个客户端连接,并维护当前在线用户。使用net.Listen创建TCP服务,每个客户端连接启动一个独立的goroutine处理。
关键点:
- 使用
map[*Client]bool记录活跃连接 - 用一个全局
chan string作为广播通道 - 每个客户端结构体包含连接本身和用户名等信息
示例代码片段:
立即学习“go语言免费学习笔记(深入)”;
type Client struct {
conn net.Conn
name string
}
var (
clients = make(map[*Client]bool)
broadcast = make(chan string)
)
2. 处理客户端连接与消息读取
每当有新连接接入,启动两个goroutine:一个负责读取消息,另一个可选用于写入(本例简化为直接写)。读取循环持续从conn读数据,收到后发送到广播channel。
注意处理连接断开的情况,确保从clients中删除并关闭资源。
- 使用
bufio.Scanner逐行读取输入 - 首次输入设为用户名
- 后续消息加上用户名前缀广播
处理函数示例:
func handleConn(conn net.Conn) {
client := &Client{conn: conn}
defer func() {
delete(clients, client)
conn.Close()
}()
// 第一次输入为昵称
scanner := bufio.NewScanner(conn)
if scanner.Scan() {
client.name = scanner.Text()
broadcast <- fmt.Sprintf("%s 加入聊天", client.name)
}
// 持续读取消息
for scanner.Scan() {
msg := fmt.Sprintf("%s: %s", client.name, scanner.Text())
broadcast <- msg
}}
3. 广播消息给所有在线用户
启动一个单独的goroutine监听broadcast channel,一旦有新消息,遍历所有客户端连接并发送。
由于写网络可能阻塞或失败,建议对每个写操作设置超时或使用非阻塞方式(生产环境可用带缓冲的channel控制)。
- 主循环监听broadcast channel
- 将消息写入每个客户端的conn
- 捕获写错误并清理失效连接
广播逻辑:
go func() {
for msg := range broadcast {
for client := range clients {
go func(c *Client) {
_, err := c.conn.Write([]byte(msg + "\n"))
if err != nil {
delete(clients, c)
}
}(client)
}
}
}()4. 启动服务器并测试
在main函数中启动监听,接受连接并调用处理函数。使用标准net工具如telnet或nc即可测试。
- 监听本地9000端口
- 每来一个连接,启动handleConn
- 保持主进程运行
main函数:
func main() {
listener, _ := net.Listen("tcp", ":9000")
defer listener.Close()
go func() {
for {
conn, _ := listener.Accept()
clients[&Client{conn: conn}] = true
go handleConn(conn)
}
}()
select {} // 阻塞主进程}
基本上就这些。运行程序后,打开多个终端执行telnet localhost 9000,第一行输入昵称,之后就可以群聊了。这个版本虽简单,但展示了Go并发处理网络服务的核心思想:轻量级协程+channel通信。扩展方向包括支持私聊、WebSocket前端界面、持久化消息等。










