GolangWebSocket实时通信实现方法-Golang-PHP中文网

GolangWebSocket实时通信实现方法

P粉602998670

发布： 2025-09-10 12:21:01

原创

935人浏览过

首先通过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)
  }
}

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

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

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

使用全局 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 后端服务。

微信二级防封域名

防封域名方法千千种，我们只做最简单且有用的这一种。微信域名防封是指通过技术手段来实现预付措施，一切说自己完全可以防封的那都是不可能的，一切说什么免死域名不会死的那也是吹牛逼的。我们正在做的是让我们的推广域名寿命更长一点，成本更低一点，效果更好一点。本源码采用 ASP+ACCESS 搭建，由于要用到二级域名，所以需要使用独享云虚机或者云服务器，不支持虚拟主机使用，不支持本地测试。目前这是免费测试版，

查看详情

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 简洁高效，适合构建聊天室、通知系统等实时应用。注意处理连接错误和并发访问，避免资源泄漏。不复杂但容易忽略细节。

以上就是GolangWebSocket实时通信实现方法的详细内容，更多请关注php中文网其它相关文章！