首页 > Golang > Go教程

Golang实现WebSocket通信教程

时间：2025-11-06 08:40:27 276浏览收藏

各位小伙伴们，大家好呀！看看今天我又给各位带来了什么文章？本文标题是《Golang实现WebSocket通信实战教程》，很明显是关于Golang的文章哈哈哈，其中内容主要会涉及到等等，如果能帮到你，觉得很不错的话，欢迎各位多多点评和分享！

使用gorilla/websocket实现Go语言WebSocket通信，首先安装依赖并创建升级器处理连接，通过wsHandler处理读写消息，主函数注册路由启动服务；前端用JavaScript建立连接测试；实现广播机制需维护客户端映射表和消息通道，将收到的消息推送给所有连接；生产环境应设置读写超时、使用锁保护共享资源、处理心跳及优雅关闭。掌握这些核心点可构建实时聊天与通知系统。

如何在Golang中开发简单的WebSocket通信_Golang WebSocket通信项目实战汇总

在Golang中实现WebSocket通信并不复杂，借助标准库和第三方包可以快速搭建实时双向通信服务。下面介绍如何从零开始开发一个简单的WebSocket通信项目，并汇总实战中的关键点。

使用gorilla/websocket搭建基础服务

Go语言标准库没有内置WebSocket支持，社区广泛使用gorilla/websocket包来处理WebSocket连接。先安装依赖：

go get github.com/gorilla/websocket

然后编写基础的WebSocket服务器：

创建一个升级HTTP连接到WebSocket的处理器：

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

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

  // 读取消息循环
  for {
    _, msg, err := conn.ReadMessage()
    if err != nil {
      break
    }
    log.Printf("收到: %s", msg)
    // 回显消息
    conn.WriteMessage(websocket.TextMessage, msg)
  }
}

注册路由并启动服务：

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

前端页面连接测试

写一个简单的HTML页面测试连接：

打开浏览器访问该页面，可在控制台看到返回的消息，说明通信成功。

实现广播机制（群聊功能）

实际项目中常需要向多个客户端发送消息。可以用一个全局map存储连接，并在新消息到来时广播给所有人：

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

在handler中加入广播逻辑：

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

每个连接加入map：

clients[conn] = true
for {
  _, msg, err := conn.ReadMessage()
  if err != nil { break }
  broadcast <- msg
}

错误处理与连接管理

生产环境中需注意连接异常、超时和资源释放：

设置读写超时：conn.SetReadDeadline(time.Now().Add(10 * time.Second))
使用互斥锁保护共享map（如clients），避免并发写入panic
监听关闭信号，优雅退出
定期ping/pong保持连接活跃

gorilla/websocket支持ping/pong自动响应，可通过设置conn.SetPongHandler处理心跳。

基本上就这些。掌握基础通信、广播模式和连接管理后，可扩展出聊天室、实时通知等应用。关键是理解WebSocket的生命周期和Go的并发模型配合使用。

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~