Go WebSocket 如何实现双向通信？-Golang-PHP中文网

Go WebSocket 如何实现双向通信？

WBOY

发布： 2024-05-17 17:39:02

原创

674人浏览过

go 语言内置了对 websocket 的支持，可通过以下步骤实现双向通信：创建 websocket 服务端：创建 http 处理程序函数。创建 upgrader 结构体升级 http 请求。启动 websocket 服务端。创建 websocket 客户端：创建 http 请求并设置 websocket 头。创建 dialer 结构体。使用 dial 方法建立连接。

Go WebSocket 如何实现双向通信？

Go WebSocket：实现双向通信

WebSocket 是一种双向通信协议，允许客户端与服务器在单个持久的 TCP 连接上进行实时数据交换。Go 语言为开发者提供了针对 WebSocket 的内置支持，使其实现双向通信变得简单直接。

创建 WebSocket 服务端

创建 WebSocket 服务端需要三步：

使用 net/http 包创建 http.Handler 函数，该函数处理 WebSocket 连接。
创建 websocket.Upgrader 结构体，该结构体将 HTTP 请求升级为 WebSocket 连接。
使用 http.ServeHTTP 方法启动 WebSocket 服务端。

package main

import (
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

func main() {
    // HTTP 处理程序
    handler := func(w http.ResponseWriter, r *http.Request) {
        upgrader := websocket.Upgrader{}
        conn, err := upgrader.Upgrade(w, r, nil)
        if err != nil {
            log.Fatal(err)
        }
        defer conn.Close()

        for {
            messageType, message, err := conn.ReadMessage()
            if err != nil {
                log.Fatal(err)
            }
            if messageType == websocket.CloseMessage {
                log.Println("Client closed connection")
                return
            }
            log.Printf("Received from client: %s", string(message))
            err = conn.WriteMessage(messageType, []byte("Echoed: "+string(message)))
            if err != nil {
                log.Fatal(err)
            }
        }
    }

    // 启动 WebSocket 服务端
    http.HandleFunc("/websocket", handler)
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal(err)
    }
}

登录后复制

创建 WebSocket 客户端

创建 WebSocket 客户端同样需要三步：

使用 net/http 包创建 http.Request，并设置 "Upgrade: WebSocket" 头。
创建 websocket.Dialer 结构体，该结构体用于建立 WebSocket 连接。
使用 Dial 方法建立连接。

package main

import (
    "fmt"
    "log"

    "github.com/gorilla/websocket"
)

func main() {
    dialers := websocket.Dialer{
        ReadBufferSize:  1024,
        WriteBufferSize: 1024,
    }

    conn, _, err := dialers.Dial("ws://localhost:8080/websocket", nil)
    if err != nil {
        log.Fatal(err)
    }
    defer conn.Close()

    for i := 0; i < 10; i++ {
        message := fmt.Sprintf("Hello from client %d", i)
        err := conn.WriteMessage(websocket.TextMessage, []byte(message))
        if err != nil {
            log.Fatal(err)
        }

        _, message, err = conn.ReadMessage()
        if err != nil {
            log.Fatal(err)
        }
        log.Printf("Received echo: %s", string(message))
    }
}

登录后复制

通过这些步骤，您可以轻松地在 Go 中实现 WebSocket 双向通信，为您的应用程序添加实时交互功能。

以上就是Go WebSocket 如何实现双向通信？的详细内容，更多请关注php中文网其它相关文章！