首页 > 后端开发 > Golang > 正文

Go WebSocket 如何实现双向通信?

WBOY
发布: 2024-06-05 21:38:00
原创
1088 人浏览过

Go 语言内置了对 WebSocket 的支持,可通过以下步骤实现双向通信:创建 WebSocket 服务端:创建 HTTP 处理程序函数。创建 Upgrader 结构体升级 HTTP 请求。启动 WebSocket 服务端。创建 WebSocket 客户端:创建 HTTP 请求并设置 WebSocket 头。创建 Dialer 结构体。使用 Dial 方法建立连接。

Go WebSocket 如何实现双向通信?

Go WebSocket:实现双向通信

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

创建 WebSocket 服务端

创建 WebSocket 服务端需要三步:

  1. 使用 net/http 包创建 http.Handler 函数,该函数处理 WebSocket 连接。
  2. 创建 websocket.Upgrader 结构体,该结构体将 HTTP 请求升级为 WebSocket 连接。
  3. 使用 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 客户端同样需要三步:

  1. 使用 net/http 包创建 http.Request,并设置 "Upgrade: WebSocket" 头。
  2. 创建 websocket.Dialer 结构体,该结构体用于建立 WebSocket 连接。
  3. 使用 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中文网其他相关文章!

相关标签:
go
来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板