Go WebSocket 如何处理断开连接？

go websocket 库提供 closehandler 机制来处理断开连接。当客户端关闭连接时，conn 类型触发 close 事件。使用 defer 语句注册 closehandler 函数可以监听该事件，接收 conn 和关闭原因常量。closehandler 函数被自动调用，提供处理断开连接和恢复应用程序的机会。

Go WebSocket：优雅地处理断开连接

在构建实时 Web 应用程序时，WebSocket 是一个强大的工具，用于在客户端和服务器之间进行双向通信。然而，处理客户端连接断开的情况至关重要，以保证应用程序的稳定性和可靠性。

处理断开连接

Go WebSocket 库为处理断开连接提供了内置机制。当客户端关闭连接时，[Conn](https://pkg.go.dev/github.com/gorilla/websocket#Conn) 类型会触发 Close 事件。

要监听此事件，可以使用 defer 语句将 [CloseHandler](https://pkg.go.dev/github.com/gorilla/websocket#CloseHandler) 函数注册到 WebSocket 连接：

arXiv Xplorer

ArXiv 语义搜索引擎，帮您快速轻松的查找，保存和下载arXiv文章。

下载

import (
    "github.com/gorilla/websocket"
)

func handleConnection(ws *websocket.Conn) {
    defer ws.CloseHandler()(ws, websocket.CloseNormalClosure)
    // ...
}

CloseHandler 函数接受两个参数：*Conn 和一个表示关闭原因的常量（例如 websocket.CloseNormalClosure 表示正常关闭）。当连接被关闭时，该函数将被自动调用。

实战案例

以下是一个处理断开连接的简单实战案例：

package main

import (
    "fmt"
    "github.com/gorilla/websocket"
    "net/http"
)

func main() {
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        ws, err := websocket.Upgrade(w, r, nil, nil, nil)
        if err != nil {
            fmt.Printf("upgrade error: %v", err)
            return
        }
        defer ws.CloseHandler()(ws, websocket.CloseNormalClosure)

        // 监听消息和断开连接事件
        go func() {
            for {
                _, _, err := ws.ReadMessage()
                if err != nil {
                    // 处理错误（例如连接断开）
                    return
                }
            }
        }()
    })

    http.ListenAndServe(":8080", nil)
}

在处理断开连接方面，CloseHandler 提供了一种简单而优雅的方式，可以帮助应用程序从客户端断开事件中恢复并继续正常运行。