0

0

Golang WebSocket消息队列处理示例

P粉602998670

P粉602998670

发布时间:2025-10-11 18:36:01

|

896人浏览过

|

来源于php中文网

原创

使用消息队列可避免阻塞WebSocket通信,提升系统可靠性;通过Go的channel或RabbitMQ/Kafka实现生产者-消费者模式,确保消息不丢失并支持异步处理。

golang websocket消息队列处理示例

用Go语言做WebSocket服务时,如果想高效处理大量消息,特别是需要异步处理、保证不丢消息或对接数据库、第三方API,结合消息队列是个很实用的做法。核心思路是把WebSocket接收到的消息先推到队列里,再由专门的消费者去处理,避免阻塞实时通信。

为什么用消息队列?

直接在WebSocket的读写循环里处理业务逻辑,比如存数据库、调外部接口,一旦耗时稍长,就可能拖慢整个连接,甚至导致超时断开。引入消息队列后:

  • 生产者(WebSocket Handler)只负责快速接收消息并投递到队列
  • 消费者(后台Worker)从队列取出消息,慢慢处理,不影响前端体验
  • 即使处理服务暂时挂了,消息还在队列里,不会丢失

基础实现:WebSocket + 内存通道模拟队列

对于轻量级应用,可以用Go的channel作为简单的内存队列。下面是一个基于gorilla/websocket和channel的例子:

package main

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

var upgrader = websocket.Upgrader{CheckOrigin: func(r http.Request) bool { return true }} var broadcast = make(chan []byte, 100) // 带缓冲的广播通道,相当于队列 var clients = make(map[websocket.Conn]bool)

// 消费者:从队列读取并广播 func handleMessages() { for { msg := <-broadcast for client := range clients { err := client.WriteMessage(websocket.TextMessage, msg) if err != nil { client.Close() delete(clients, client) } } } }

// 生产者:WebSocket连接处理 func wsHandler(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()

clients[conn] = true

for {
    _, msg, err := conn.ReadMessage()
    if err != nil {
        delete(clients, conn)
        break
    }
    // 收到消息后,快速扔进队列
    broadcast <- msg
}

}

立即学习go语言免费学习笔记(深入)”;

星火作家大神
星火作家大神

星火作家大神是一款面向作家的AI写作工具

下载

func main() { go handleMessages() // 启动消费者 http.HandleFunc("/ws", wsHandler) log.Println("Server on :8080") log.Fatal(http.ListenAndServe(":8080", nil)) }

进阶方案:集成RabbitMQ/Kafka

生产环境推荐使用RabbitMQ、Kafka这类专业消息队列。以RabbitMQ为例,你需要:

  • 安装amqp库:go get github.com/rabbitmq/amqp091-go
  • wsHandler里,把broadcast 换成向RabbitMQ发送消息
  • 另起一个或多个worker进程,监听RabbitMQ队列,执行实际业务逻辑

这样系统更稳定,支持多实例水平扩展,也方便做消息持久化、重试等。

基本上就这些,关键是把“接收”和“处理”拆开,让WebSocket保持轻快。

相关专题

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

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

180

2024.02.23

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

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

228

2024.02.23

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

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

340

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开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

393

2024.05.21

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

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

197

2025.06.09

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

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

191

2025.06.10

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

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

233

2025.06.17

AO3中文版入口地址大全
AO3中文版入口地址大全

本专题整合了AO3中文版入口地址大全,阅读专题下面的的文章了解更多详细内容。

1

2026.01.21

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
WEB前端教程【HTML5+CSS3+JS】
WEB前端教程【HTML5+CSS3+JS】

共101课时 | 8.4万人学习

JS进阶与BootStrap学习
JS进阶与BootStrap学习

共39课时 | 3.2万人学习

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

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