如何在 Golang 中实现 WebSocket 的双向通信

WebSockets 是一种在客户端和服务器之间建立持久化连接的通信协议，它允许双向通信，而且在实时性和效率方面表现出色。在 Golang 中，我们可以使用标准库中的 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 是服务器的监听地址和端口号，你可以根据需要修改。

使用以上代码，我们已经成功实现了在 Golang 中通过 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);
});

综上所述，我们详细介绍了如何在 Golang 中实现 WebSocket 的双向通信，并提供了具体的代码示例。通过这些示例代码，你可以快速上手并实现基于 WebSocket 的实时通信功能。