本文共 2374 字，大约阅读时间需要 7 分钟。
本文将介绍如何利用 Go 语言优化一个简单的 TCP 服务器实现，通过增加客户端连接超时机制，实现客户端 10 秒无响应时自动断开连接的功能。
以下是在本次改造中的主要内容和目标：
服务器基于 Go 语言实现，采用标准库 net 包中的 Listen、Accept 和 Close 等函数进行操作。主要代码逻辑包括以下几个部分：
上线监听
用 net.Listen("tcp", ":8040") 开启一个 TCP 监听端口 8040
处理连接
使用 Accept 回调接收新的连接请求，并启动 goroutine 处理每个连接
客户端输入处理
通过 bufio.Scanner 实现对客户端输入的读取
超时机制实现
使用 time.After 创建 10 秒超时定时器，实现自动断开连接功能
以下是主要改造的代码片段：
package mainimport ( "bufio" "fmt" "log" "net" "strings" "sync" "time")func main() { listeners, err := net.Listen("tcp", ":8040") if err != nil { log.Fatal("无法绑定监听地址", err) } for { conn, err := listeners.Accept() if err != nil { log.Print("连接拒绝：", err) // 例如客户端主动中断连接 continue } go handleConn(conn) }}// 处理每个连接的 goroutinefunc handleConn(c net.Conn) { input := bufio.NewScanner(c) wg := sync.WaitGroup message := make(chan string, 100) wg.Add(1) // 定时器退出 goroutine go func() { defer wg.Done() for { select { case <-time.After(10 * time.Second): c.Close() // 10 秒后自动断开连接 case mes := <-message: // 收到消息后启动新的 goroutine 处理回复 wg.Add(1) go func(c net.Conn, text string) { defer wg.Done() // 实现 echo 功能 fmt.Fprintln(c, "\t", strings.ToUpper(text)) time.Sleep(1 * time.Second) fmt.Fprintln(c, "\t", text) time.Sleep(1 * time.Second) fmt.Fprintln(c, "\t", strings.ToLower(text)) // 除非出现错误，否则不使用 Ctrl + w 中断 }(c, mes) } } }() // 等待 input 中的所有消息处理完成 for input.Scan() { text := input.Text() if text != "" { message <- text } } <-time.After(time.Millisecond) // 确保所有 goroutine 完成等待 c.Close()} 定时器与 goroutine 结合使用
通过 select 语句配合 time.Afterellyً،实现了客户端 10 秒无响应时自动断开连接功能
消息传递与 goroutine 框架
利用 sync.WaitGroup 进行 goroutine 的同步管理，确保并发操作的顺序性
客户端消息处理
实现了消息的读取与转发，客户端输入内容将被自动转发给另一 goroutine 进行处理
本次改造主要通过增加定时器机制，实现了客户端连接的自动断开功能。在代码实现上，主要采用 goroutine 和 select 语句进行并发控制，并通过 sync.WaitGroup 进行 goroutine 的同步。这一设计既保持了原有 echo 服务器的功能，又增加了更加灵活的服务器控制流程，为后续功能扩展提供了很好的基础。
转载地址：http://stnmz.baihongyu.com/