Go语言Websocket开发实践：如何处理连接中断

package main

import (
    "log"
    "net/Http"
    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

func echo(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()

    for {
        // 读取消息
        messageType, p, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            break
        }

        // 处理消息
        err = conn.WriteMessage(messageType, p)
        if err != nil {
            log.Println(err)
            break
        }
    }
}

func main() {
    http.HandleFunc("/echo", echo)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

func echo(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()

    for {
        select {
        case <-stopCh:
            // 处理连接中断
            log.Println("connection interrupted")
            return
        default:
            // 读取消息
            messageType, p, err := conn.ReadMessage()
            if err != nil {
                log.Println(err)
                break
            }

            // 处理消息
            err = conn.WriteMessage(messageType, p)
            if err != nil {
                log.Println(err)
                break
            }
        }
    }
}

func main() {
    http.HandleFunc("/echo", func(w http.ResponseWriter, r *http.Request) {
        // 具体的业务逻辑处理

        // 中断连接
        stopCh <- struct{}{}
    })

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