GolangWebSocket实时通信实现示例-Golang-PHP中文网

GolangWebSocket实时通信实现示例

P粉602998670

发布： 2025-09-17 12:38:01

原创

242人浏览过

首先实现WebSocket服务端，利用Goroutine处理连接与消息广播；接着创建HTML客户端通过JavaScript建立连接并收发消息；最后可选静态文件服务以加载页面，完整实现Golang与前端的实时通信。

golangwebsocket实时通信实现示例

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，非常适合实现实时数据交互。Golang 凭借其轻量级 Goroutine 和高效的网络处理能力，是构建 WebSocket 服务的理想选择。下面是一个基于 gorilla/websocket 库的简单实时通信示例，包含服务端和客户端的基本实现。

1. 安装依赖

Go 官方标准库不包含 WebSocket 支持，我们使用流行的 gorilla/websocket 包：

go get github.com/gorilla/websocket

2. WebSocket 服务端实现

创建一个简单的 HTTP 服务，升级连接为 WebSocket，并实现广播机制。

package main

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

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

// 存储所有连接的客户端 var clients = make(map[*websocket.Conn]bool) var broadcast = make(chan []byte)

func main() { http.HandleFunc("/ws", handleConnections)

// 启动广播监听
go handleMessages()

log.Println("Server started on :8080")
log.Fatal(http.ListenAndServe(":8080", nil))

登录后复制

}

立即学习“go语言免费学习笔记（深入）”；

ViiTor实时翻译

AI实时多语言翻译专家！强大的语音识别、AR翻译功能。

116

查看详情

func handleConnections(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Fatal(err) return } defer conn.Close()

clients[conn] = true

// 实时接收消息
for {
    _, msg, err := conn.ReadMessage()
    if err != nil {
        delete(clients, conn)
        break
    }
    broadcast <- msg
}

登录后复制

}

立即学习“go语言免费学习笔记（深入）”；

func handleMessages() { for { msg := <-broadcast for client := range clients { err := client.WriteMessage(websocket.TextMessage, msg) if err != nil { client.Close() delete(clients, client) } } } }

3. 简单 HTML 客户端测试

创建一个 index.html 页面，连接 WebSocket 并发送/接收消息。

<!DOCTYPE html> <html> <head> <title>WebSocket Test</title> </head> <body> <input type="text" id="msg" placeholder="输入消息" /> <button onclick="send()">发送</button> <div id="output"></div>

<script>
    const ws = new WebSocket("ws://localhost:8080/ws");
    const output = document.getElementById("output");
    const input = document.getElementById("msg");

    ws.onmessage = function(event) {
        const div = document.createElement("div");
        div.textContent = "收到: " + event.data;
        output.appendChild(div);
    };

    function send() {
        if (input.value) {
            ws.send(input.value);
            input.value = "";
        }
    }
</script>

登录后复制

</body> </html>

将该页面放在服务目录下，或通过静态文件服务访问。