Cara menutup sambungan WebSocket dengan betul dalam Go

Dengan aplikasi meluas teknologi WebSocket, semakin ramai pengaturcara mula menggunakan bahasa Go untuk melaksanakan fungsi WebSocket. Dalam proses melaksanakan fungsi WebSocket, pengaturcara sering menghadapi masalah penting, iaitu, cara menutup sambungan WebSocket dengan betul dalam kod bahasa Go. Artikel ini menerangkan cara menutup sambungan WebSocket dengan betul dalam Go.

Kaedah penutupan sambungan WebSocket

Terdapat dua cara untuk menutup sambungan WebSocket:

1. Kaedah penutupan aktif: klien atau pelayan boleh menutup dengan menghantar mesej rapat WebSocket sambungan.
2. Kaedah penutupan pasif: Apabila sambungan TCP diputuskan atau mesej pengecualian diterima, sambungan WebSocket juga akan ditutup.

Penutupan sambungan WebSocket dalam bahasa GO

Dalam bahasa Go, kita boleh menutup sambungan WebSocket dengan memanggil kaedah Close() sambungan WebSocket. Jika anda ingin menutup sambungan WebSocket secara aktif, anda boleh memanggil kaedah ini pada sisi klien atau pelayan. Jika anda ingin menutup sambungan WebSocket secara pasif, anda boleh mengesan operasi baca dan tulis pada bahagian pelayan Setelah pengecualian sambungan dikesan, panggil kaedah Close() untuk menutup sambungan WebSocket.

Berikut ialah contoh kod untuk menutup sambungan WebSocket dalam Go:

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

Loop:
    for {
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
                log.Printf("error: %v", err)
            }
            break Loop
        }
        // handle message
    }
}

Dalam kod di atas, kami menggunakan kaedah Upgrade() yang disediakan oleh Gorilla WebSocket untuk menaik taraf sambungan HTTP ke WebSocket menyambung. Apabila mengendalikan sambungan WebSocket, kami menggunakan gelung untuk membaca mesej WebSocket dan memanggil kaedah Close() untuk menutup sambungan WebSocket apabila pengecualian sambungan dikesan.

Perlu diingatkan bahawa apabila menutup sambungan WebSocket, kita perlu mengelakkan kebocoran sumber. Dalam kod di atas, kami menggunakan penangguhan untuk menutup sambungan WebSocket apabila keluar dari fungsi Ini memastikan bahawa walaupun pengecualian berlaku semasa pemprosesan fungsi, sambungan WebSocket akan ditutup dalam masa untuk mengelakkan kebocoran sumber.

Ringkasan

Melalui pengenalan artikel ini, saya percaya bahawa pembaca mempunyai pemahaman tertentu tentang cara menutup sambungan WebSocket dengan betul dalam Go. Dalam proses pembangunan sebenar, kita perlu memilih kaedah penutupan sambungan WebSocket mengikut senario aplikasi tertentu. Sama ada penutupan aktif atau penutupan pasif, kami perlu memastikan kebocoran sumber dielakkan semasa menutup sambungan WebSocket dan memastikan operasi program yang stabil.

Atas ialah kandungan terperinci Cara menutup sambungan WebSocket dengan betul dalam Go. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!