如何用Golang实现一个简易的聊天程序基于TCP套接字通信案例-Golang-PHP中文网

如何用Golang实现一个简易的聊天程序基于TCP套接字通信案例

P粉602998670

发布： 2025-06-28 12:39:01

原创

368人浏览过

要实现一个基于tcp的简易聊天程序，golang是理想选择，因其内置强大网络库和高并发支持。1. 搭建tcp服务器端基本结构：使用net.listen监听端口并为每个连接开启goroutine处理；2. 处理客户端连接与消息广播：用全局map记录连接，handleclient函数读取消息并向其他客户端广播；3. 客户端接入与消息发送：通过net.dial连接服务器，并用goroutine实现消息收发；4. 注意事项：加锁保护共享资源、正确处理缓冲区、排除自己防止回声、设置超时机制避免资源占用。这些步骤实现了基础tcp聊天功能，并为后续扩展打下基础。

如何用Golang实现一个简易的聊天程序基于TCP套接字通信案例

要实现一个基于TCP的简易聊天程序，Golang是个不错的选择。它内置了强大的网络库，尤其是对并发的支持非常好，很适合做这种多客户端通信的小项目。

下面我从几个关键点出发，讲讲怎么一步步用Go写一个简单的TCP聊天程序。

1. 搭建TCP服务器端的基本结构

首先，你需要启动一个TCP服务器，监听某个端口，等待客户端连接进来。Go的标准库net提供了非常简洁的接口来完成这个任务。

立即学习“go语言免费学习笔记（深入）”；

核心代码大概是这样的：

listener, err := net.Listen("tcp", ":8080")
if err != nil {
    log.Fatal(err)
}
defer listener.Close()

for {
    conn, err := listener.Accept()
    if err != nil {
        log.Println(err)
        continue
    }
    go handleClient(conn)
}

登录后复制

上面这段代码创建了一个监听在8080端口的TCP服务，并为每个新连接开启一个goroutine去处理。这样做的好处是多个客户端可以同时连接，互不干扰。

2. 处理客户端连接与消息广播

当有多个客户端连接时，服务器需要记录这些连接，并在收到某一个客户端的消息后，把这条消息转发给其他所有在线的客户端。

为了做到这一点，你可以维护一个全局的客户端连接列表，比如：

var clients = make(map[net.Conn]bool)
var mutex sync.Mutex

登录后复制

每次有新的连接加入时，就把它加进map里；断开连接时，就从map中移除。

然后，在handleClient函数中读取消息，并向其他客户端广播：

func handleClient(conn net.Conn) {
    mutex.Lock()
    clients[conn] = true
    mutex.Unlock()

    defer func() {
        mutex.Lock()
        delete(clients, conn)
        mutex.Unlock()
        conn.Close()
    }()

    for {
        message := make([]byte, 1024)
        n, err := conn.Read(message)
        if err != nil || n == 0 {
            break
        }

        // 广播给其他客户端
        for client := range clients {
            if client != conn {
                client.Write(message[:n])
            }
        }
    }
}

登录后复制

这样就能实现实时的群聊功能了。当然这里只是最基础的实现，没有做太多错误处理和性能优化。

3. 客户端如何接入并发送消息

客户端部分也很简单，只需要连接到服务器地址，然后可以在控制台输入消息发出去。

示例代码如下：

conn, _ := net.Dial("tcp", "localhost:8080")
go func() {
    for {
        msg, _ := bufio.NewReader(os.Stdin).ReadString('\n')
        conn.Write([]byte(msg))
    }
}()

for {
    reply := make([]byte, 1024)
    n, _ := conn.Read(reply)
    fmt.Println("收到消息：", string(reply[:n]))
}

登录后复制

这段代码做了两件事：