Go 言語の Websocket 開発実践: 例外を処理して接続を閉じる方法

Go 言語 Websocket 開発の実践: 例外を処理し、接続を閉じる方法

はじめに:
Web 開発では、Websocket を使用して双方向通信を実現できます。リアルタイム データ送信、チャット アプリケーション、その他のシナリオで通常使用されるプロトコル。ただし、Websocket を使用すると、ネットワークの変動やクライアントの異常なシャットダウンなど、予期しない接続の中断が発生しやすくなります。この場合、接続の異常終了をどのように適切に処理し、アプリケーションの安定性と信頼性を確保するかが、開発時に重点的に取り組むべき課題となります。この記事では、サンプル コードを組み合わせ、Go 言語を例として、例外を処理して接続を閉じる方法を紹介します。

1. Websocket サーバーの作成:
まず、単純な Websocket サーバーを作成する必要があります。以下は基本的なサンプル コードです:

package main

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

// 定义Websocket读写的缓冲区大小
const bufferSize = 1024

// Websocket处理函数
func wsHandler(w http.ResponseWriter, r *http.Request) {
    // 升级HTTP连接为Websocket连接
    conn, err := websocket.Upgrade(w, r, nil, bufferSize, bufferSize)
    if err != nil {
        fmt.Println("Failed to upgrade the connection to websocket: ", err)
        return
    }

    for {
        // 读取客户端发送的消息
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            fmt.Println("Failed to read message from client: ", err)
            break
        }

        // 处理接收到的消息
        handleMessage(conn, messageType, message)
    }

    // 关闭连接
    defer conn.Close()
}

// 处理接收到的消息
func handleMessage(conn *websocket.Conn, messageType int, message []byte) {
    // 处理消息的逻辑
    fmt.Println("Received message: ", string(message))
}

2. 接続の異常終了を処理します:
クライアントが異常終了した場合、サーバーは接続が閉じられたことを正しく検出し、処理できる必要があります。それに応じて。以下は、異常終了した接続を処理する方法のサンプル コードです。

// Websocket处理函数
func wsHandler(w http.ResponseWriter, r *http.Request) {
    // 升级HTTP连接为Websocket连接
    conn, err := websocket.Upgrade(w, r, nil, bufferSize, bufferSize)
    if err != nil {
        fmt.Println("Failed to upgrade the connection to websocket: ", err)
        return
    }

    // 异常关闭连接处理协程
    go handleCloseConnection(conn)

    for {
        // 读取客户端发送的消息
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            // 检测到连接异常关闭的错误
            if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
                fmt.Println("Connection closed unexpectedly: ", err)
                break
            }
            // 其他错误处理
            fmt.Println("Failed to read message from client: ", err)
            break
        }

        // 处理接收到的消息
        handleMessage(conn, messageType, message)
    }

    // 关闭连接
    conn.Close()
}

// 异常关闭连接处理协程
func handleCloseConnection(conn *websocket.Conn) {
    // 读取关闭消息，表示客户端主动关闭连接
    _, _, err := conn.ReadMessage()
    if err != nil {
        fmt.Println("Failed to read close message from client: ", err)
    }

    // 处理连接关闭后的逻辑
    fmt.Println("Connection closed by client")
}

上記のサンプル コードでは、IsUnexpectedCloseError 関数を呼び出して、接続が異常終了したかどうかを判断します。接続が異常に閉じられた場合、対応するエラー メッセージが記録され、ループが抜け出して接続が閉じられます。さらに、接続が閉じられた後のロジックを処理するために、handleCloseConnection という名前のコルーチン関数も追加しました。

結論:
Websocket アプリケーションを開発する場合、例外を処理して接続を閉じることは非常に重要な手順です。 IsUnexpectedCloseError 関数とコルーチン処理を使用することで、異常な接続の終了を検出して処理し、接続例外によるアプリケーションのクラッシュを回避できます。実際のアプリケーションでは、ログの記録や通知の送信など、ニーズに応じて対応する拡張を行うことで、アプリケーションの信頼性や保守性を向上させることができます。

参考資料:

• [Gorilla WebSocket](https://github.com/gorilla/websocket)
• [Go 標準ライブラリ ドキュメント](https: / /golang.org/pkg/net/http/)
• [WebSocket プロトコル開発ガイド](https://developer.mozilla.org/en-US/docs/Web/API/WebSocket_API/Writing_WebSocket_servers)

以上がGo 言語の Websocket 開発実践: 例外を処理して接続を閉じる方法の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。