Golang如何使用WebSocket实现实时通信_Golang WebSocket实时通信开发实践-Golang-PHP中文网

Golang如何使用WebSocket实现实时通信_Golang WebSocket实时通信开发实践

P粉602998670

发布： 2025-11-29 15:31:02

原创

550人浏览过

Golang结合gorilla/websocket可高效实现WebSocket通信。首先通过go get安装库，然后创建升级HTTP连接的服务器，处理消息收发；接着用map和channel管理客户端连接并广播消息；前端通过JavaScript建立连接测试；最后建议添加心跳机制与错误监控以提升稳定性。

golang如何使用websocket实现实时通信_golang websocket实时通信开发实践

在构建现代 Web 应用时，实时通信是一个常见需求。Golang 凭借其高并发和轻量级 Goroutine 的优势，非常适合用于实现 WebSocket 服务端。通过 gorilla/websocket 这个广泛使用的第三方库，可以快速搭建稳定高效的 WebSocket 实时通信系统。

1. 安装 gorilla/websocket 库

在项目中使用 WebSocket 前，需要先引入 gorilla 提供的 websocket 包：

go get github.com/gorilla/websocket

该库封装了 WebSocket 协议的底层细节，提供简洁的 API 来处理连接、读取消息和发送数据。

2. 创建 WebSocket 服务端

一个基础的 WebSocket 服务端包括 HTTP 升级处理和消息收发逻辑。以下是一个简单的示例：

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

package main

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

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

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

  for {
    mt, message, err := conn.ReadMessage()
    if err != nil {
      break
    }
    log.Printf("收到: %s", message)
    // 回显消息
    conn.WriteMessage(mt, []byte("echo: "+string(message)))
  }
}

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

上述代码启动了一个监听 /ws 路径的 WebSocket 服务，接受客户端连接并回显收到的消息。

小程序websocket用法示例

该示例将演示在原生微信小程序中使用GoEasy快速实现websocket实时通讯。

查看详情

3. 管理多个连接与广播消息

实际应用中通常需要管理多个客户端连接，并支持广播消息。可以通过维护一个连接池来实现：

var clients = make(map[*websocket.Conn]bool)
var broadcast = make(chan []byte)

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
  conn, err := upgrader.Upgrade(w, r, nil)
  if err != nil {
    log.Println(err)
    return
  }
  defer conn.Close()
  clients[conn] = true

  go func() {
    for {
      _, message, err := conn.ReadMessage()
      if err != nil {
        delete(clients, conn)
        break
      }
      broadcast <- message
    }
  }()
}

// 广播协程 func broadcastMessages() {
  for {
    msg := <-broadcast
    for client := range clients {
      err := client.WriteMessage(websocket.TextMessage, msg)
      if err != nil {
        client.Close()
        delete(clients, client)
      }
    }
  }
}

在 main 函数中启动广播协程：
go broadcastMessages()
这样所有在线客户端都能接收到广播消息。

4. 前端连接测试

使用浏览器 JavaScript 测试连接：

const ws = new WebSocket("ws://localhost:8080/ws");
ws.onopen = () => {
ws.send("Hello Go WebSocket!");
};
ws.onmessage = (event) => {
console.log("收到:", event.data);
};

确保服务运行后打开页面或使用工具（如 wscat）测试通信是否正常。

基本上就这些。Golang 配合 gorilla/websocket 可以高效实现 WebSocket 实时通信，适用于聊天室、通知推送、协同编辑等场景。关键是合理管理连接生命周期和并发安全。不复杂但容易忽略心跳机制和错误处理，建议在生产环境中加入 ping/pong 心跳检测和日志监控。

以上就是Golang如何使用WebSocket实现实时通信_Golang WebSocket实时通信开发实践的详细内容，更多请关注php中文网其它相关文章！