GolangWebSocket通信实战教程

怎么入门Golang编程？需要学习哪些知识点？这是新手们刚接触编程时常见的问题；下面golang学习网就来给大家整理分享一些知识点，希望能够给初学者一些帮助。本篇文章就来介绍《Golang WebSocket通信实例教程》，涉及到，有需要的可以收藏一下

答案：使用Golang和gorilla/websocket库可实现WebSocket通信。1. 安装依赖：go get github.com/gorilla/websocket；2. 编写服务器端代码，监听/ws路径，升级HTTP连接为WebSocket，接收并回显消息；3. 编写客户端代码，连接本地服务器，发送"Hello, WebSocket!"并接收回显；4. 先运行服务器，再运行客户端，验证双向通信成功。

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，非常适合实时应用，比如聊天、通知推送等。Golang 提供了简洁高效的语言特性来实现 WebSocket 服务端和客户端。下面是一个基于 gorilla/websocket 库的完整示例，展示如何用 Go 编写 WebSocket 服务器与客户端。

1. 安装依赖

go get github.com/gorilla/websocket

2. WebSocket 服务器端实现

server.go

package main

import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
)

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

func echoHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Print("升级 WebSocket 失败:", err)
        return
    }
    defer conn.Close()

    for {
        // 读取消息
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Print("读取消息失败:", err)
            break
        }

        log.Printf("收到消息: %s", message)

        // 回显消息
        if err := conn.WriteMessage(messageType, message); err != nil {
            log.Print("发送消息失败:", err)
            break
        }
    }
}

func main() {
    http.HandleFunc("/ws", echoHandler)
    log.Println("服务器启动在 :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

3. WebSocket 客户端实现

client.go

package main

import (
    "log"
    "net/url"
    "github.com/gorilla/websocket"
)

func main() {
    u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/ws"}
    log.Printf("连接到 %s", u.String())

    conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
    if err != nil {
        log.Fatal("连接失败:", err)
    }
    defer conn.Close()

    // 发送消息
    message := []byte("Hello, WebSocket!")
    err = conn.WriteMessage(websocket.TextMessage, message)
    if err != nil {
        log.Fatal("发送消息失败:", err)
    }

    // 读取回显
    _, msg, err := conn.ReadMessage()
    if err != nil {
        log.Fatal("读取消息失败:", err)
    }

    log.Printf("收到回显: %s", msg)
}

4. 运行步骤

• 运行服务器：go run server.go
• 另起终端运行客户端：go run client.go
• 服务器应输出收到的消息，客户端输出回显内容

基本上就这些。这个例子展示了最基础的 WebSocket 双向通信流程。你可以在此基础上扩展：广播消息、维护连接池、心跳检测等。不复杂但容易忽略。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。