GolangWebSocket实时通信实现示例

本篇文章向大家介绍《GolangWebSocket实时通信实现示例》，主要包括，具有一定的参考价值，需要的朋友可以参考一下。

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

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

1. 安装依赖

Go 官方标准库不包含 WebSocket 支持，我们使用流行的 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))

}

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
}

}

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 并发送/接收消息。

<input type="text" id="msg" placeholder="输入消息" /> 发送

<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>

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

4. 支持静态文件服务（可选）

修改 main 函数，添加对 HTML 文件的静态服务：

确保 index.html 与服务端代码在同一目录。

基本上就这些。这个示例展示了如何使用 Golang 快速搭建一个支持多客户端消息广播的 WebSocket 服务。实际项目中可扩展用户标识、消息类型区分、连接认证等功能。不复杂但容易忽略的是连接异常处理和资源释放，建议在生产环境中加入日志和心跳机制。

理论要掌握，实操不能落！以上关于《GolangWebSocket实时通信实现示例》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！