Go语言通过net包实现TCP长连接服务，用于即时通讯等场景。首先使用net.Listen监听端口，Accept接受连接并为每个客户端启动goroutine处理读写。在handleConnection中，开启读协程接收数据，通过SetReadDeadline设置读超时实现心跳检测，收到消息后重置超时，若接收到”ping”则回复”pong”。主协程等待done信号关闭连接。客户端示例通过Dial连接服务器，循环发送消息并读取回显，验证通信正常。整体结构包含连接管理、读写分离与心跳机制，具备扩展性，可结合context、协议编码等提升生产环境稳定性。
在Go语言中实现TCP长连接服务，通常用于即时通讯、设备上报、心跳维持等场景。核心是通过net包监听TCP连接，为每个客户端维护一个持久连接，并支持读写分离、心跳检测、并发安全处理。下面是一个简洁但完整的TCP长连接服务示例。
1. 基础TCP服务器结构
使用net.Listen创建监听，接受客户端连接并为每个连接启动独立的goroutine处理数据收发。
package main
<p>import (
"fmt"
"log"
"net"
)</p><p>func main() {
listener, err := net.Listen("tcp", ":8080")
if err != nil {
log.Fatal("监听端口失败:", err)
}
defer listener.Close()
fmt.Println("TCP服务器启动，监听 :8080")</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">for {
conn, err := listener.Accept()
if err != nil {
log.Println("接受连接失败:", err)
continue
}
go handleConnection(conn)
}
}
2. 处理长连接的读写逻辑
每个连接开启两个goroutine：一个负责读取客户端数据，另一个可选地处理消息发送。使用sync.Mutex或通道协调状态。
func handleConnection(conn net.Conn) {
defer conn.Close()
clientIP := conn.RemoteAddr().String()
fmt.Printf("客户端 %s 已连接\n", clientIP)
<pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">// 模拟存储连接状态（实际可用map+锁或context管理）
done := make(chan bool)
// 读协程
go func() {
buffer := make([]byte, 1024)
for {
n, err := conn.Read(buffer)
if err != nil {
fmt.Printf("客户端 %s 断开: %v\n", clientIP, err)
done <- true
return
}
msg := string(buffer[:n])
fmt.Printf("来自 %s 的消息: %s\n", clientIP, msg)
// 回显消息（可替换为业务逻辑）
conn.Write([]byte("echo: " + msg))
}
}()
// 可扩展：写协程通过channel接收消息主动下发
<-done
}
3. 添加心跳机制保活
TCP连接可能因网络中断静默断开，需加入心跳检测。常见做法是设定读超时，或让客户端定期发送ping。
立即学习“go语言免费学习笔记（深入）”；
一键出成片的录屏演示软件，专为制作产品演示、教学课程和使用教程而设计。
34
查看详情
import (
"time"
)
<p>func handleConnection(conn net.Conn) {
defer conn.Close()
clientIP := conn.RemoteAddr().String()
fmt.Printf("客户端 %s 已连接\n", clientIP)</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">conn.SetReadDeadline(time.Now().Add(30 * time.Second)) // 初始超时
done := make(chan bool)
go func() {
buffer := make([]byte, 1024)
for {
n, err := conn.Read(buffer)
if err != nil {
done <- true
return
}
msg := string(buffer[:n])
fmt.Printf("收到: %s\n", msg)
// 收到消息重置超时时间
conn.SetReadDeadline(time.Now().Add(30 * time.Second))
// 简单协议判断是否为心跳
if msg == "ping" {
conn.Write([]byte("pong"))
} else {
conn.Write([]byte("echo: " + msg))
}
}
}()
// 超时或异常退出
<-done
fmt.Printf("连接 %s 关闭\n", clientIP)
}
4. 客户端测试示例
使用简单客户端连接并发送数据验证服务稳定性。
package main
<p>import (
"fmt"
"net"
"time"
)</p><p>func main() {
conn, _ := net.Dial("tcp", "127.0.0.1:8080")
defer conn.Close()</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">for i := 0; i < 5; i++ {
msg := fmt.Sprintf("hello %d", i)
conn.Write([]byte(msg))
buf := make([]byte, 1024)
n, _ := conn.Read(buf)
fmt.Println("回显:", string(buf[:n]))
time.Sleep(2 * time.Second)
}
// 发送心跳
conn.Write([]byte("ping"))
buf := make([]byte, 1024)
n, _ := conn.Read(buf)
fmt.Println("心跳响应:", string(buf[:n]))
}
基本上就这些。这个示例展示了如何用Go构建一个基础的TCP长连接服务，包含连接管理、读写分离和心跳机制。实际生产环境中可结合context取消、连接池、协议编码（如protobuf）、日志监控等进一步增强健壮性。
大家都在看：
如何在 Go 中解析转义的 JSON 字符串
Go语言中Goroutine与标准库及第三方包的并发使用指南
Go语言子字符串的内存管理与高效复制：避免潜在的内存泄露
Go语言中切片指针的预分配与填充：惯用方法解析
暂无评论内容