0

0

GolangWebSocket实时通信实现方法

P粉602998670

P粉602998670

发布时间:2025-09-10 12:21:01

|

940人浏览过

|

来源于php中文网

原创

首先通过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 后端服务。

华友协同办公自动化OA系统
华友协同办公自动化OA系统

华友协同办公管理系统(华友OA),基于微软最新的.net 2.0平台和SQL Server数据库,集成强大的Ajax技术,采用多层分布式架构,实现统一办公平台,功能强大、价格便宜,是适用于企事业单位的通用型网络协同办公系统。 系统秉承协同办公的思想,集成即时通讯、日记管理、通知管理、邮件管理、新闻、考勤管理、短信管理、个人文件柜、日程安排、工作计划、工作日清、通讯录、公文流转、论坛、在线调查、

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

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

178

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

226

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

339

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

209

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

391

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

196

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

191

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

192

2025.06.17

高德地图升级方法汇总
高德地图升级方法汇总

本专题整合了高德地图升级相关教程,阅读专题下面的文章了解更多详细内容。

65

2026.01.16

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Node.js 教程
Node.js 教程

共57课时 | 8.8万人学习

CSS3 教程
CSS3 教程

共18课时 | 4.6万人学习

Vue 教程
Vue 教程

共42课时 | 6.7万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号