当前位置:

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

Golang实现WebSocket实时通信教程

使用Go和gorilla/websocket库可实现高性能WebSocket服务,1.通过Upgrade将HTTP升级为WebSocket连接;2.使用Hub结构管理客户端连接池,实现消息广播;3.前端通过WebSocketAPI发送和接收消息,多个浏览器窗口可实现实时群聊功能。

本文目录

    使用Go和gorilla/websocket库可实现高性能WebSocket服务,1. 通过Upgrade将HTTP升级为WebSocket连接;2. 使用Hub结构管理客户端连接池,实现消息广播;3. 前端通过WebSocket API发送和接收消息,多个浏览器窗口可实现实时群聊功能。

    怎样用Golang实现WebSocket 实时通信服务开发

    WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议,非常适合实现实时通信功能,比如聊天室、通知推送、在线协作等。Golang 凭借其高并发和轻量级 goroutine 的特性,非常适合开发高性能的 WebSocket 服务。下面介绍如何使用 Go 和第三方库 gorilla/websocket 实现一个简单的 WebSocket 实时通信服务。

    1. 安装依赖库

    Go 标准库没有内置 WebSocket 支持,我们使用广泛使用的 gorilla/websocket 库:

    go get github.com/gorilla/websocket

    2. 建立 WebSocket 连接

    首先创建一个 HTTP 服务,并升级普通 HTTP 请求为 WebSocket 连接。

    示例代码:

    package main

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

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

    func wsHandler(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Print("Upgrade error:", err) return } defer conn.Close()

    // 连接建立成功,可以开始收发消息
    for {
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Print("Read error:", err)
            break
        }
    
        log.Printf("Received: %s", message)
        // 回显消息
        if err := conn.WriteMessage(messageType, message); err != nil {
            log.Print("Write error:", err)
            break
        }
    }

    }

    func main() { http.HandleFunc("/ws", wsHandler) log.Println("Server starting on :8080") log.Fatal(http.ListenAndServe(":8080", nil)) }

    3. 实现客户端广播(群聊)

    在实际场景中,通常需要将消息广播给多个客户端。为此,可以维护一个客户端连接池。

    定义一个中心化的 Hub 来管理所有连接:

    type Hub struct { clients map[*websocket.Conn]bool broadcast chan []byte register chan *websocket.Conn unregister chan *websocket.Conn }

    func newHub() Hub { return &Hub{ clients: make(map[websocket.Conn]bool), broadcast: make(chan []byte), register: make(chan websocket.Conn), unregister: make(chan websocket.Conn), } }

    func (h *Hub) run() { for { select { case conn := <-h.register: h.clients[conn] = true case conn := <-h.unregister: if _, ok := h.clients[conn]; ok { delete(h.clients, conn) conn.Close() } case message := <-h.broadcast: for conn := range h.clients { if err := conn.WriteMessage(websocket.TextMessage, message); err != nil { log.Print("Broadcast error:", err) conn.Close() delete(h.clients, conn) } } } } }

    修改处理函数,接入 Hub:

    func wsHandler(hub *Hub) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Print("Upgrade error:", err) return } defer conn.Close()
        hub.register <- conn
    
        // 读取消息
        for {
            _, message, err := conn.ReadMessage()
            if err != nil {
                hub.unregister <- conn
                break
            }
            // 广播给所有客户端
            hub.broadcast <- message
        }
    }

    }

    在 main 函数中启动 Hub:

    func main() { hub := newHub() go hub.run()
    http.HandleFunc("/ws", wsHandler(hub))
    log.Println("Server on :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))

    }

    4. 简单的前端测试页面

    创建一个 HTML 页面测试连接:

      打开多个浏览器窗口,输入消息即可实现群聊。

      基本上就这些。通过 gorilla/websocket 和一个 Hub 结构,就能实现一个基础但完整的 WebSocket 实时通信服务。后续可扩展身份认证、私聊、消息持久化等功能。不复杂但容易忽略的是连接管理和错误处理,务必在生产环境中加入日志、超时控制和并发安全机制。

      本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系bd@zhengruan.com
      作者最新文章
      编程开发
      相关文章 更多
      Ponytail在Claude Code中的用途:控制代码规模、审查改动与整理技术债务
      Ponytail在Claude Code中的用途:控制代码规模、审查改动与整理技术债务

      面向使用Claude Code进行开发的程序员,Ponytail插件旨在通过引导AI助手优先复用现有能力来控制新增代码规模。本文介绍其核心用途:在开发任务中减少不必要代码编写,通过“/ponytail-review”命令审查暂存改动的安全性与可维护性,以及利用“/ponytail-debt”整理带有明确限制的开发捷径和技术债务。使用前需根据安装方式确认宿主支持的具体命令和自动加载功能。

      Impeccable适合哪些人使用:开发者与团队的界面设计指南
      Impeccable适合哪些人使用:开发者与团队的界面设计指南

      Impeccable面向已使用AI编程工具并希望优化界面清晰度的个人开发者、前端工程师及产品维护团队。它不要求用户精通设计术语,但需能明确描述用户目标与限制条件。本文介绍三类典型应用场景:个人开发者利用shape命令梳理需求边界,避免开发蔓延;团队通过document命令提取视觉规则至DESIGN.md以保持风格一致;维护者使用harden命令检查表单等界面的边缘状态。适用前提是手头有真实的界面设计或改进需求。

      Graphify离线使用指南:纯代码构图与本地模型配置方法
      Graphify离线使用指南:纯代码构图与本地模型配置方法

      面向需要在无网或受限网络环境下使用Graphify的开发者,明确其离线运行的边界条件。纯代码构图和读取本地图谱可在依赖就绪后完全离线进行;文档语义处理需预先配置本地Ollama模型并指定后端。本文详解如何通过--code-only参数避免联网调用、如何准备本地模型环境,以及区分本地文件读取与远程MCP服务连接的关键步骤,帮助用户实现真正的本地化知识图谱构建。

      命名转换工具推荐:马上默、摩卡工具箱与UniKit对比
      命名转换工具推荐:马上默、摩卡工具箱与UniKit对比

      面向需要在camelCase、snake_case、kebab-case等格式间转换变量名和字段名的开发者,本文对比三款中文界面工具。马上默适合批量处理并支持词元拆分检查,能精细控制缩写与数字边界;摩卡工具箱界面简洁,聚焦四种常用风格快速互转;UniKit则一次性提供十种命名风格结果,便于横向比较。针对缩写切分和数字连接等常见差异,文章说明了各工具的适用场景及核对要点,帮助用户根据项目规范选择合适方案。

      ASCII码表查询与转换工具推荐:小鹿tools、Toolbase与易白工具
      ASCII码表查询与转换工具推荐:小鹿tools、Toolbase与易白工具

      面向需要查询ASCII码值或进行文本编码转换的开发人员,本文对比三款中文在线工具。小鹿tools提供完整的0-127标准码表及控制字符详解,支持多种进制并排查看;Toolbase适合快速筛选特定字符或码值,界面直观;易白工具专注于整段文本与十进制、十六进制等序列的互转,适用于串口数据核对。需注意扩展ASCII并非统一标准,处理中文或Emoji时应使用Unicode工具。

      Graphify安装与配置指南:Python环境搭建及Claude Code技能注册
      Graphify安装与配置指南:Python环境搭建及Claude Code技能注册

      面向需要在本地部署Graphify工具的开发人员,本文详解基于Python 3.10+环境的安装流程。针对用户常遇到的依赖冲突问题,推荐使用uv或pipx创建独立运行环境。内容涵盖graphifyy官方包的安装验证、CLI命令可用性检查,以及针对Claude Code等编码助手的技能注册步骤,帮助读者快速完成从环境准备到功能集成的全过程。

      Impeccable常见用途:评审页面质量、适配多端布局与简化界面
      Impeccable常见用途:评审页面质量、适配多端布局与简化界面

      面向需要优化前端界面的开发者,Impeccable借助AI编程助手辅助改进具体页面。它常用于评审页面技术质量、调整移动端与桌面端布局,以及简化复杂界面以降低认知负荷。使用时需针对不同目标选择专用命令,如audit用于检查问题,adapt用于设备适配,distill用于精简内容。本文介绍如何通过明确指令聚焦单一任务,避免一次性重构整个项目,并在修改后通过实际业务流程验证功能完整性。

      Scientific Agent Skills能做什么:数据整理、统计分析与科研绘图指南
      Scientific Agent Skills能做什么:数据整理、统计分析与科研绘图指南

      面向需要使用AI辅助科研工作的用户,Scientific Agent Skills通过封装常用工具的操作规范,帮助AI Agent完成数据整理、统计分析、科学绘图及专业研究工作流。针对科研中工具调用复杂、流程断层的痛点,本文介绍其核心能力:串联Polars等库进行数据检查与清洗;利用statistical-analysis技能指导组间比较、回归分析及结果报告规范;以及执行如scanpy单细胞测序等专业领域分析流程。注意实际效果取决于所选技能、依赖环境及数据条件,关键结论仍需人工核对。

      DeepSeek SuperAgent与网页聊天区别:工作流程与适用场景对比
      DeepSeek SuperAgent与网页聊天区别:工作流程与适用场景对比

      面向需要区分DeepSeek交互模式的用户,本文对比DeepSeek SuperAgent原型与网页聊天的核心差异。网页聊天提供即时问答入口,适合临时咨询;SuperAgent围绕代码仓库接收事件、判断授权并管理工单,需部署Harness及运行主机。文章从工作流程、权限机制及协作方式解析两者定位,帮助用户根据是否需持续跟进Issue或交付PR选择合适工具,避免混淆模型等级与功能边界。

      Impeccable完整使用教程:从上下文建立到视觉规则验证
      Impeccable完整使用教程:从上下文建立到视觉规则验证

      面向使用AI编程助手进行前端开发的用户,介绍Impeccable工具在维护产品一致性与视觉规范中的工作流。针对开发中容易忽视的设计细节与上下文缺失问题,本文详解如何通过/impeccable init建立产品上下文,利用/impeccable document记录DESIGN.md视觉规则,并通过critique和polish命令评审与优化特定页面。同时涵盖终端命令行检测npx impeccable detect的使用场景,帮助开发者在限定范围内高效完成代码修改与流程验证。

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

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

      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 分镜、局部重绘、提示词反推和无限画布等工具,适合短剧、漫剧、广告及社交媒体内容制作。