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 mainimport ("fmt""log""net")
func main() {listener, err := net.Listen("tcp", ":8080")if err != nil {log.Fatal("监听端口失败:", err)}defer listener.Close()fmt.Println("TCP服务器启动,监听 :8080")
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)// 模拟存储连接状态(实际可用map+锁或context管理)done := make(chan bool)// 读协程go func() { buffer := make([]byte, 1024) for { n, err := conn.Read(buffer) if err != nil { fmt.Printf("客户端 %s 断开: %vn", clientIP, err) done <- true return } msg := string(buffer[:n]) fmt.Printf("来自 %s 的消息: %sn", clientIP, msg) // 回显消息(可替换为业务逻辑) conn.Write([]byte("echo: " + msg)) }}()// 可扩展:写协程通过channel接收消息主动下发<-done
}
3. 添加心跳机制保活
TCP连接可能因网络中断静默断开,需加入心跳检测。常见做法是设定读超时,或让客户端定期发送ping。
立即学习“go语言免费学习笔记(深入)”;
import ( "time")func handleConnection(conn net.Conn) {defer conn.Close()clientIP := conn.RemoteAddr().String()fmt.Printf("客户端 %s 已连接n", clientIP)
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("收到: %sn", msg) // 收到消息重置超时时间 conn.SetReadDeadline(time.Now().Add(30 * time.Second)) // 简单协议判断是否为心跳 if msg == "ping" { conn.Write([]byte("pong")) } else { conn.Write([]byte("echo: " + msg)) } }}()// 超时或异常退出<-donefmt.Printf("连接 %s 关闭n", clientIP)
}
4. 客户端测试示例
使用简单客户端连接并发送数据验证服务稳定性。
package mainimport ("fmt""net""time")
func main() {conn, _ := net.Dial("tcp", "127.0.0.1:8080")defer conn.Close()
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)、日志监控等进一步增强健壮性。
以上就是Golang TCP长连接服务实现示例的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1413483.html
微信扫一扫
支付宝扫一扫