跳转到主内容
极星编程网:以代码为星,赴技术山海!

Golang简单聊天室系统开发实战

答案是使用Golang基于TCP实现聊天室,利用goroutine处理并发连接,服务端通过map维护客户端并广播消息,客户端通过双goroutine实现收发。

用Golang开发一个简单的聊天室系统,核心在于利用其轻量级的goroutine和高效的网络编程能力。整个系统可以基于TCP协议实现,服务端负责管理客户端连接、消息广播,客户端则用于发送和接收消息。下面从结构设计到代码实现一步步带你完成这个小项目。

系统架构与基本流程这个聊天室采用C/S(客户端-服务器)模式:服务器监听指定端口,等待客户端连接每个客户端连接后,启动一个独立的goroutine处理读写服务器维护所有活跃连接,并在收到消息时广播给其他客户端支持用户上线通知、下线提示和群发消息关键点是使用goroutine + channel + net包来实现并发处理,避免阻塞。

服务端实现逻辑服务端主要功能包括:监听端口、接受连接、注册用户、广播消息、处理断开连接。

立即学习“go语言免费学习笔记(深入)

”;

定义全局map保存客户端连接,配合互斥锁保证并发安全:// 客户端集合var clients = make(map[net.Conn]string) var mutex sync.Mutex

主函数中启动监听:

listener, _ := net.Listen("tcp", ":8080") defer listener.Close() fmt.Println("服务器已启动,监听 :8080...")循环接受连接,每来一个客户端就开启一个goroutine:for {conn, _ := listener.Accept() go handleClient(conn)}handleClient函数负责读取用户名、注册、监听消息并广播:func handleClient(conn net.Conn) {// 读取用户名buffer := make([]byte, 1024) n, _ := conn.Read(buffer) username := string(buffer[:n-1]) // 去掉换行mutex.Lock() clients[conn] = username mutex.Unlock() broadcast(fmt.Sprintf("%s 加入聊天\n", username), conn)// 持续读取消息for {n, err := conn.Read(buffer) if err != nil { break }msg := string(buffer[:n]) broadcast(username+": "+msg, conn)}// 断开处理mutex.Lock() delete(clients, conn) mutex.Unlock() broadcast(fmt.Sprintf("%s 离开聊天\n", username), conn) conn.Close()}广播函数遍历所有客户端连接,跳过消息来源:php版微信js-sdk支付接口类php版微信js-sdk支付接口类下载func broadcast(message string, sender net.Conn) {mutex.Lock() defer mutex.Unlock() for conn := range clients {if conn != sender {conn.Write([]byte(message))}}}客户端实现要点客户端相对简单,只需要连接服务器、先发送用户名、再开启两个goroutine分别处理输入和接收消息。

连接服务器:

conn, _ := net.Dial("tcp", "127.0.0.1:8080")

先输入用户名并发送:

fmt.Print("请输入用户名: ") scanner := bufio.NewScanner(os.Stdin) scanner.Scan() username := scanner.Text() conn.Write([]byte(username + "\n"))

开启两个协程:

一个持续读取控制台输入并发送到服务端另一个持续读取服务端广播的消息并打印go func() {for scanner.Scan() {conn.Write([]byte(scanner.Text() + "\n"))}}() go func() {buf := make([]byte, 1024) for {n, err := conn.Read(buf) if err != nil { return }fmt.Print(string(buf[:n]))}}()

保持主函数不退出:

select{}运行与测试编译运行server.go启动服务端,再打开多个终端运行client.go,输入不同用户名即可进入聊天室。任意客户端发送消息,其他人都能实时看到。

可优化方向:

加入私聊功能(通过命令如 /msg user hello)

增加在线用户列表显示使用JSON格式传输数据提升扩展性添加心跳机制检测连接状态基本上就这些。Golang的并发模型让这种网络服务写起来非常直观,几段代码就能跑通一个可用的聊天室原型。

相关文章