在Golang的网络编程场景中，网络连接关闭是不可避免的环节，无论是客户端主动断开、服务端超时回收，还是网络异常导致的连接中断，都需要开发者编写对应的处理逻辑，避免资源泄漏和程序异常。
网络连接关闭的常见触发场景
首先我们需要了解哪些情况会导致网络连接关闭，这样能更有针对性地编写处理逻辑：
- 客户端主动调用关闭方法，终止当前连接
- 连接长时间无数据交互，触发服务端或客户端的超时机制
- 网络链路出现异常，比如网络中断、对端服务崩溃等
- 程序主动取消连接对应的上下文，触发连接关闭流程
基础连接关闭处理方法
Golang的net.Conn接口提供了Close方法用于关闭连接，我们在使用连接完成读写操作后，需要主动调用该方法回收资源。以下是一个简单的TCP客户端连接关闭示例：
package main
import (
"fmt"
"net"
"time"
)
func main() {
// 建立TCP连接
conn, err := net.DialTimeout("tcp", "127.0.0.1:8080", 5*time.Second)
if err != nil {
fmt.Println("建立连接失败:", err)
return
}
// 使用defer确保连接最终会被关闭
defer func() {
err := conn.Close()
if err != nil {
fmt.Println("关闭连接失败:", err)
} else {
fmt.Println("连接已成功关闭")
}
}()
// 模拟业务读写操作
time.Sleep(2 * time.Second)
}
结合Context处理连接关闭
在实际的网络服务开发中，我们通常会结合context来管理连接的生命周期，当上下文被取消时，自动触发连接关闭，避免连接长时间占用资源。以下是服务端结合Context处理连接关闭的示例：
package main
import (
"context"
"fmt"
"net"
"time"
)
func handleConn(ctx context.Context, conn net.Conn) {
defer conn.Close()
// 启动一个goroutine监听上下文取消信号
go func() {
<-ctx.Done()
// 上下文取消时关闭连接
conn.Close()
}()
buf := make([]byte, 1024)
for {
// 设置读取超时
conn.SetReadDeadline(time.Now().Add(3 * time.Second))
n, err := conn.Read(buf)
if err != nil {
// 捕获连接关闭相关错误
fmt.Println("读取数据异常:", err)
return
}
fmt.Printf("收到数据: %sn", string(buf[:n]))
}
}
func main() {
// 监听端口
listener, err := net.Listen("tcp", "127.0.0.1:8080")
if err != nil {
fmt.Println("监听端口失败:", err)
return
}
defer listener.Close()
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
for {
conn, err := listener.Accept()
if err != nil {
fmt.Println("接收连接失败:", err)
continue
}
// 为每个连接启动处理协程
go handleConn(ctx, conn)
}
}
连接关闭的错误捕获与处理
连接关闭时通常会返回对应的错误，我们需要对常见的错误类型进行判断，区分正常关闭和异常关闭。常见的连接关闭相关错误包括EOF、connection reset by peer等，以下是错误处理的示例：
package main
import (
"fmt"
"io"
"net"
"time"
)
func readData(conn net.Conn) {
buf := make([]byte, 1024)
for {
n, err := conn.Read(buf)
if err != nil {
// 判断是否为正常关闭
if err == io.EOF {
fmt.Println("连接被对端正常关闭")
} else {
fmt.Println("连接异常关闭:", err)
}
return
}
fmt.Printf("读取到数据: %sn", string(buf[:n]))
}
}
func main() {
conn, err := net.DialTimeout("tcp", "127.0.0.1:8080", 5*time.Second)
if err != nil {
fmt.Println("建立连接失败:", err)
return
}
defer conn.Close()
go readData(conn)
time.Sleep(5 * time.Second)
}
注意事项
在处理网络连接关闭时，还需要注意以下几点：
- 所有的
net.Conn连接都需要确保最终被关闭，建议使用defer配合关闭逻辑 - 多个goroutine同时操作同一个连接时，需要做好同步，避免重复关闭连接导致错误
- 设置合理的读写超时时间，避免连接长时间阻塞在读写操作上
- 连接关闭后不要再对其进行读写操作，否则会返回错误