首页 > 后端开发 > Golang > 正文

Golang TCP长连接服务实现示例

P粉602998670
发布: 2025-10-13 21:25:15
原创
669人浏览过
Go语言通过net包实现TCP长连接服务,用于即时通讯等场景。首先使用net.Listen监听端口,Accept接受连接并为每个客户端启动goroutine处理读写。在handleConnection中,开启读协程接收数据,通过SetReadDeadline设置读超时实现心跳检测,收到消息后重置超时,若接收到"ping"则回复"pong"。主协程等待done信号关闭连接。客户端示例通过Dial连接服务器,循环发送消息并读取回显,验证通信正常。整体结构包含连接管理、读写分离与心跳机制,具备扩展性,可结合context、协议编码等提升生产环境稳定性。

golang tcp长连接服务实现示例

在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。

Remove.bg
Remove.bg

AI在线抠图软件,图片去除背景

Remove.bg 102
查看详情 Remove.bg

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

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)、日志监控等进一步增强健壮性。

以上就是Golang TCP长连接服务实现示例的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号