Golang WebSocket消息订阅与广播示例

答案:Go通过gorilla/websocket库实现WebSocket订阅与广播,核心为连接升级、客户端管理与消息广播。使用Upgrader将HTTP升级为WebSocket,允许跨域;clients map记录所有活跃连接;broadcast channel接收消息并由handleBroadcast协程向所有客户端推送;handleConnections处理新连接与读取消息,主函数启动广播协程并监听/ws路径;前端用JavaScript创建WebSocket连接,发送和显示消息,实现简单聊天功能;需注意并发安全与异常连接清理。

golang websocket消息订阅与广播示例

用Go实现WebSocket的订阅与广播,核心是管理客户端连接和消息分发。下面是一个简洁实用的示例,基于gorilla/websocket库,能快速搭建一个支持消息广播的基础服务。

基本结构设计

服务端需要几个关键组件:升级HTTP连接的Upgrader、存储所有活跃连接的集合、以及一个广播通道。通过这些,可以实现新用户加入、消息接收和向所有在线用户推送。

Upgrader:负责把普通的HTTP请求“升级”成WebSocket连接,这里设置CheckOrigin: true允许跨域请求。clients:用map保存所有已连接的客户端,key为*websocket.Conn,value通常设为bool表示是否在线。broadcast:一个全局channel,用来接收所有客户端发来的消息,并由后台goroutine统一处理广播。

后端实现代码

主逻辑包含三个部分:处理连接、读取消息和广播消息。启动时开启一个独立协程运行广播函数,主线程监听WebSocket路径。

示例代码:

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

package main

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

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

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

func handleConnections(w http.ResponseWriter, r *http.Request) {conn, err := upgrader.Upgrade(w, r, nil)if err != nil {log.Fatal(err)}defer conn.Close()

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

}

func handleBroadcast() {for {msg := websocket.TextMessage, msg)if err != nil {client.Close()delete(clients, client)}}}}

func main() {go handleBroadcast()

http.HandleFunc("/ws", handleConnections)http.ListenAndServe(":8080", nil)

}

前端页面接入

前端通过原生JavaScript的WebSocket API连接服务端,监听消息并更新页面。用户输入内容后点击发送,消息会通过WebSocket传给服务端,再广播给所有人。

创建new WebSocket("ws://localhost:8080/ws")实例建立连接。监听onmessage事件,将收到的消息动态添加到聊天区域。点击“发送”按钮时调用ws.send()把输入框内容发出去。

基本上就这些,不复杂但容易忽略并发安全和连接清理。后续可扩展用户ID绑定、房间订阅或持久化日志等功能。

以上就是Golang WebSocket消息订阅与广播示例的详细内容,更多请关注创想鸟其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1410965.html

赞 (0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
如何在Golang中实现二进制文件处理
上一篇 2025年12月16日 03:47:53
Golang模块迁移与版本冲突解决技巧
下一篇 2025年12月16日 03:47:58

相关推荐

发表回复

登录后才能评论
关注微信