Bagaimana untuk melaksanakan komunikasi dua hala WebSocket dalam golang

Cara melaksanakan komunikasi dua hala dengan WebSocket di Golang

WebSockets ialah protokol komunikasi yang mewujudkan sambungan berterusan antara pelanggan dan pelayan Ia membolehkan komunikasi dua hala dan sangat baik dalam masa nyata dan kecekapan. Di Golang, kita boleh menggunakan pakej net/http dan gorilla/websocket dalam pustaka standard untuk melaksanakan komunikasi dua hala WebSocket. Artikel ini akan memperkenalkan cara melaksanakan komunikasi dua hala WebSocket di Golang dan memberikan contoh kod khusus. net/http 和 gorilla/websocket 包来实现 WebSocket 的双向通信。本文将介绍如何在 Golang 中实现 WebSocket 的双向通信，并提供具体的代码示例。

步骤一：导入依赖包

首先，我们需导入 net/http 和 gorilla/websocket 包。net/http 包提供了 HTTP 服务器和客户端的功能，而 gorilla/websocket 包则是一个 WebSocket 实现库。

import (
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

步骤二：设置 WebSocket 处理函数

接下来，我们需要设置 WebSocket 的处理函数。这个函数将会处理客户端的连接请求，并在连接建立成功后，处理客户端发送的消息。

func websocketHandler(w http.ResponseWriter, r *http.Request) {
    // 将 HTTP 连接升级为 WebSocket 连接
    ws, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Error upgrading HTTP connection to WebSocket:", err)
        return
    }

    // 在连接结束时关闭 WebSocket
    defer ws.Close()

    // 循环处理客户端发来的消息
    for {
        // 读取客户端发来的消息
        messageType, message, err := ws.ReadMessage()
        if err != nil {
            log.Println("Error reading message from WebSocket:", err)
            break
        }

        // 打印收到的消息
        log.Printf("Received message: %s", message)

        // 回复客户端收到的消息
        if err = ws.WriteMessage(messageType, message); err != nil {
            log.Println("Error writing message to WebSocket:", err)
            break
        }
    }
}

步骤三：设置 WebSocket 升级器

在处理函数中，我们需要使用 gorilla/websocket 包的升级器(upgrader)来将 HTTP 连接升级为 WebSocket 连接。升级器还可以设置一些参数，例如允许的域以及握手超时时间。

var upgrader = websocket.Upgrader{
    // 允许跨域访问
    CheckOrigin: func(r *http.Request) bool {
        return true
    },
}

步骤四：设置 HTTP 服务器

最后，我们需要设置 HTTP 服务器，将处理函数映射到相应的路径上，以便客户端可以通过路径连接到 WebSocket。

func main() {
    // 设置 WebSocket 处理函数
    http.HandleFunc("/", websocketHandler)

    // 启动 HTTP 服务器
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal("Error starting HTTP server:", err)
    }
}

代码示例中的 :8080

Langkah 1: Import pakej pergantungan

Mula-mula, kita perlu mengimport pakej net/http dan gorilla/websocket. Pakej net/http menyediakan pelayan HTTP dan kefungsian klien, manakala pakej gorilla/websocket ialah perpustakaan pelaksanaan WebSocket.

const socket = new WebSocket("ws://localhost:8080/");

socket.addEventListener("open", (event) => {
    socket.send("Hello, server!");
});

socket.addEventListener("message", (event) => {
    console.log("Received message from server:", event.data);
});

socket.addEventListener("close", (event) => {
    console.log("Server connection closed:", event);
});

Langkah 2: Tetapkan fungsi pemprosesan WebSocket🎜🎜Seterusnya, kita perlu menetapkan fungsi pemprosesan WebSocket. Fungsi ini akan mengendalikan permintaan sambungan klien dan memproses mesej yang dihantar oleh klien selepas sambungan berjaya diwujudkan. 🎜rrreee🎜Langkah 3: Sediakan penaik taraf WebSocket🎜🎜Dalam fungsi pemprosesan, kita perlu menggunakan penaik taraf (upgrader) pakej gorila/websocket untuk menaik taraf Sambungan HTTP ke sambungan WebSocket. Penaik taraf juga boleh menetapkan beberapa parameter seperti domain yang dibenarkan dan tamat masa jabat tangan. 🎜rrreee🎜Langkah 4: Sediakan pelayan HTTP🎜🎜Akhir sekali, kita perlu menyediakan pelayan HTTP dan memetakan fungsi pemprosesan ke laluan yang sepadan supaya pelanggan boleh menyambung ke WebSocket melalui laluan. 🎜rrreee🎜:8080 dalam contoh kod ialah alamat mendengar dan nombor port pelayan, anda boleh mengubah suainya mengikut keperluan. 🎜🎜Menggunakan kod di atas, kami telah berjaya melaksanakan fungsi komunikasi dua hala melalui WebSocket di Golang. Selepas sambungan pelanggan berjaya, anda boleh menggunakan kod yang serupa dengan yang berikut untuk menghantar dan menerima mesej: 🎜rrreee🎜Ringkasnya, kami telah memperkenalkan secara terperinci cara melaksanakan komunikasi dua hala WebSocket di Golang dan menyediakan contoh kod khusus. Melalui kod sampel ini, anda boleh dengan cepat bermula dan melaksanakan fungsi komunikasi masa nyata berdasarkan WebSocket. 🎜

Atas ialah kandungan terperinci Bagaimana untuk melaksanakan komunikasi dua hala WebSocket dalam golang. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!