当前位置:

首页 > 编程开发 > Golang WebSocket实时通信实现教程

Golang WebSocket实时通信实现教程

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

本文目录

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

    本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系bd@zhengruan.com
    作者最新文章
    编程开发
    相关文章 更多
    GitHub六个AI开源项目能做什么?软件理解、语音与求职场景
    GitHub六个AI开源项目能做什么?软件理解、语音与求职场景

    解析六个GitHub AI开源项目的具体用途,包括软件静态分析、智能体回答检查、多语言配音、求职材料生成、本地代码检索及动态歌词展示,帮助开发者根据实际需求筛选工具并正确验证结果。

    opencode安装skills教程及本地配置方法
    opencode安装skills教程及本地配置方法

    遇到OpenCode Skills无法加载或配置无效?本文手把手教你完成Skills的安装、环境变量设置及本地调试,避开常见坑点,让AI助手更懂你的代码习惯。

    如何开启Visual Studio2022实时代码分析警告提示
    如何开启Visual Studio2022实时代码分析警告提示

    解决 Visual Studio 2022 编辑时不显示代码分析警告的问题。指南涵盖 .NET 项目实时分析设置、Roslyn 分析器范围调整及错误列表来源筛选,确保开发者能高效利用实时代码诊断功能。

    如何在Visual Studio2022查看变量监视窗口调试程序
    如何在Visual Studio2022查看变量监视窗口调试程序

    详解 Visual Studio 2022 调试技巧:包括断点设置、监视窗口开启方法、变量添加步骤以及通过 F10 单步执行观察变量值变化的全过程。

    Visual Studio2017设置默认编码为UTF-8不带BOM
    Visual Studio2017设置默认编码为UTF-8不带BOM

    针对 Visual Studio 2017 缺乏全局默认编码选项的情况,详解如何利用 .editorconfig 设定项目级 UTF-8 无 BOM 规则,以及手动转换现有文件并校验字节的方法。

    Visual Studio2022F5调试提示启动项目未配置怎么修复
    Visual Studio2022F5调试提示启动项目未配置怎么修复

    解决 Visual Studio 2022 F5 调试报错问题,通过设置启动项目、检查解决方案属性及正确区分类库与应用项目,快速恢复调试功能。

    Visual Studio2022NuGet包还原失败离线处理方案
    Visual Studio2022NuGet包还原失败离线处理方案

    解决Visual Studio 2022离线环境中NuGet包还原失败的难题,涵盖依赖链分析、本地源配置、包集收集及隔离验证步骤,帮助开发者构建稳定的离线开发环境。

    Visual Studio2017大项目加载缓慢,优化加载性能技巧
    Visual Studio2017大项目加载缓慢,优化加载性能技巧

    解决Visual Studio 2017大型项目加载慢的实用指南。涵盖性能监测、扩展管理、文档恢复优化及版本兼容性检查,帮助开发者精准定位瓶颈并提升工作效率。

    Visual Studio2022设置多启动项目同时调试运行
    Visual Studio2022设置多启动项目同时调试运行

    详解 Visual Studio 2022 多启动项目配置方法,包括启动动作选择、顺序调整及调试技巧,帮助开发者高效调试分布式或前后端分离应用。

    fastapi下载文件接口实现教程
    fastapi下载文件接口实现教程

    深入解析在FastAPI框架中实现文件下载接口的完整方案,涵盖基于FileResponse的本地静态文件传输、基于StreamingResponse的分块流式传输、RFC 5987中文文件名编码规范、路径穿越安全防御及BackgroundTasks临时文件清理,并配备详细代码示例与架构解析图。

    查看更多
    精品专题 更多
    装机必备
    装机必备

    正软商城装机必备专区,精选办公、浏览器、安全防护、影音播放、压缩解压、设计创作和系统工具等电脑常用正版软件,帮助用户快速完成新电脑软件配置。

    Windows
    Windows

    正软商城Windows软件专区,汇集适用于Windows电脑的办公、设计、安全防护、影音播放、开发工具和系统优化软件,提供软件介绍、系统要求、正版授权及购买下载服务。

    PDF教程
    PDF教程

    正软商城PDF教程频道提供PDF编辑、转换、合并、拆分、压缩及格式处理方法,同时介绍常用PDF软件和工具的使用技巧。

    Mac软件 更多
    讯飞智作 网页版
    讯飞智作 网页版
    Windows/ macOS/ Linux

    讯飞智作是一款基于语音合成技术、虚拟人驱动技术与数字人形象声音构建能力的 AI 数字人音视频创作平台。

    火山引擎 网页版
    火山引擎 网页版
    Windows/ macOS/ Linux

    火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力,依托字节自研豆包多模态大模型,凭借火山方舟一站式平台,提供模型精调、推理部署、智能应用搭建等全链路服务,结合高性能AI算力,助力各行业快速落地大模型数字化与智能化应用。

    Aperty macOS版
    Aperty macOS版
    macOS

    Aperty 是 Skylum 推出的 AI 人像修图软件,支持自然肤质修饰、瑕疵处理、面部与身体调整、预设及批量同步。

    WINDOWS 更多
    讯飞智作 网页版
    讯飞智作 网页版
    Windows/ macOS/ Linux

    讯飞智作是一款基于语音合成技术、虚拟人驱动技术与数字人形象声音构建能力的 AI 数字人音视频创作平台。

    火山引擎 网页版
    火山引擎 网页版
    Windows/ macOS/ Linux

    火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力,依托字节自研豆包多模态大模型,凭借火山方舟一站式平台,提供模型精调、推理部署、智能应用搭建等全链路服务,结合高性能AI算力,助力各行业快速落地大模型数字化与智能化应用。

    PixTV 网页版
    PixTV 网页版
    Windows

    PixTV 是一款在浏览器中使用的 AI 视觉内容创作平台,提供图片、视频、音频、文字生成、AI 分镜、局部重绘、提示词反推和无限画布等工具,适合短剧、漫剧、广告及社交媒体内容制作。