商城首页欢迎来到中国正版软件门户
文章教程　|　产品大全　|　软件问答

您的位置：首页 >Golang WebSocket实时通信实现教程

Golang WebSocket实时通信实现教程

　　发布于2025-10-28　阅读（0）

扫一扫，手机访问

首先通过gorilla/websocket库升级HTTP连接为WebSocket，再用map和互斥锁管理客户端连接，最后通过广播机制实现消息实时推送。

GolangWebSocket实时通信实现方法

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，非常适合实现实时数据交互。在 Go 语言中，可以通过 gorilla/websocket 库快速实现 WebSocket 实时通信功能。以下是基于 Golang 实现 WebSocket 通信的核心步骤和代码示例。

建立 WebSocket 服务端

使用 gorilla/websocket 包可以轻松创建 WebSocket 服务。首先需要安装依赖：

go get github.com/gorilla/websocket

然后编写一个简单的 WebSocket 服务端，用于接收和广播消息：

初始化 WebSocket 升级器：HTTP 请求需要通过升级器（Upgrader）转换为 WebSocket 连接。

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

处理客户端连接：将 HTTP 请求升级为 WebSocket，并启动读写协程。

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

  // 启动读取消息循环
  for {
    _, msg, err := conn.ReadMessage()
    if err != nil {
      break
    }
    log.Printf("Received: %s", msg)
    // 回显消息或广播给其他客户端
    broadcastMessage(msg)
  }
}

管理客户端连接与广播消息

为了实现多用户实时通信，需要维护所有活跃的连接，并支持消息广播。

使用全局 map 存储连接：可以用 map 保存所有 WebSocket 连接，配合互斥锁保证并发安全。

var clients = make(map[*websocket.Conn]bool)
var broadcast = make(chan []byte)
var mutex = sync.Mutex{}

注册与注销连接：在建立和断开时更新 clients 集合。

// 注册
mutex.Lock()
clients[conn] = true
mutex.Unlock()

// 注销（在读循环退出后）
mutex.Lock()
delete(clients, conn)
mutex.Unlock()

广播消息到所有客户端：启动一个独立的 goroutine 处理广播。

go func() {
  for msg := range broadcast {
    mutex.Lock()
    for conn := range clients {
      err := conn.WriteMessage(websocket.TextMessage, msg)
      if err != nil {
        conn.Close()
        delete(clients, conn)
      }
    }
    mutex.Unlock()
  }
}()

前端建立 WebSocket 连接

前端使用原生 WebSocket API 连接 Go 后端服务。

const socket = new WebSocket("ws://localhost:8080/ws");

socket.onopen = function(event) {
  console.log("Connected to server");
};

socket.onmessage = function(event) {
  console.log("Received:", event.data);
};

// 发送消息
socket.send("Hello, Server!");

确保后端路由 /ws 能正确处理 WebSocket 升级请求。

启动 HTTP 服务并监听端口

将 WebSocket 处理函数挂载到指定路由，并启动服务。

func main() {
  http.HandleFunc("/ws", handleConnection)
  log.Println("Server started on :8080")
  err := http.ListenAndServe(":8080", nil)
  if err != nil {
    log.Fatal("ListenAndServe error:", err)
  }
}

运行程序后，前端即可通过 ws://localhost:8080/ws 建立连接，实现双向实时通信。

基本上就这些。核心是升级 HTTP 到 WebSocket、管理连接状态、安全地广播消息。gorilla/websocket 简洁高效，适合构建聊天室、通知系统等实时应用。注意处理连接错误和并发访问，避免资源泄漏。不复杂但容易忽略细节。

本文转载于：互联网 如有侵犯，请联系zhengruancom@outlook.com删除。
免责声明：正软商城发布此文仅为传递信息，不代表正软商城认同其观点或证实其描述。

上一篇：谷歌浏览器开发者工具怎么切换到手机模式

下一篇：程序关闭后进程仍在运行的原因可能有以下几点：后台服务未停止：有些程序在关闭主界面后，仍然会继续运行后台服务或守护进程，以提供持续的功能支持（如消息推送、数据同步

产品推荐

最新发布

相关推荐

热门关注
网站备案号： 苏ICP备2026018738号-1 联系邮箱：zhengruancom@outlook.com
Copyright ©2018-2020