0

0

Golang WebSocket多客户端管理示例

P粉602998670

P粉602998670

发布时间:2025-10-11 17:13:01

|

590人浏览过

|

来源于php中文网

原创

答案:使用Golang和gorilla/websocket实现WebSocket多客户端管理,通过Client结构体封装连接,ClientManager进行注册、注销和广播。1. 客户端连接由Client结构体表示,包含Conn和Send通道;2. ClientManager维护客户端集合,处理注册、注销及消息广播;3. ServeHTTP升级HTTP连接并启动读写协程;4. readPump从客户端读取消息,writePump发送消息;5. Run方法监听事件并更新客户端状态。

golang websocket多客户端管理示例

在使用 Golang 构建 WebSocket 服务时,管理多个客户端连接是一个常见需求。下面是一个简洁、实用的多客户端管理示例,使用标准库 net/httpgorilla/websocket 实现。

依赖安装

go get github.com/gorilla/websocket

这是目前最常用的 WebSocket 第三方库。

WebSocket 客户端结构体设计

每个客户端连接封装为一个结构体,便于统一管理。

package main

import (
    "net/http"
    "sync"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        return true // 允许跨域,生产环境建议严格校验
    },
}

type Client struct {
    Conn *websocket.Conn
    Send chan []byte
}

type ClientManager struct {
    clients    map[*Client]bool
    register   chan *Client
    unregister chan *Client
    broadcast  chan []byte
    mutex      sync.RWMutex
}

Client 表示一个 WebSocket 连接,包含连接实例和发送消息的通道。
ClientManager 是核心管理器,用于注册、注销和广播消息。

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

启动管理器并处理连接

实现管理器的运行逻辑和 HTTP 处理函数。

免费红色响应式多语言企业通用模板1.0.0
免费红色响应式多语言企业通用模板1.0.0

该模板源码有公司简介、公司新闻、产品展示、客户案例、留言等企业官网常用页面功能。模板是响应式模板,支持多语言,完善的标签调用修改起来很方便。功能特点:1. 使用的框架采用HkCms开源内容管理系统v2.2.3版本、免费可以商用。2. 所需环境Apache/Nginx,PHP7.2 及以上 + MySQL 5.6 及以上。3. 安装教程: (1) 站点运行路径填写到public目录下。 (2) 浏览

下载
func NewClientManager() *ClientManager {
    return &ClientManager{
        clients:    make(map[*Client]bool),
        register:   make(chan *Client),
        unregister: make(chan *Client),
        broadcast:  make(chan []byte),
    }
}

func (manager *ClientManager) Run() {
    for {
        select {
        case client := <-manager.register:
            manager.mutex.Lock()
            manager.clients[client] = true
            manager.mutex.Unlock()
            println("客户端加入,当前总数:", len(manager.clients))

        case client := <-manager.unregister:
            manager.mutex.Lock()
            if _, ok := manager.clients[client]; ok {
                delete(manager.clients, client)
                close(client.Send)
            }
            manager.mutex.Unlock()
            println("客户端退出,剩余:", len(manager.clients))

        case message := <-manager.broadcast:
            manager.mutex.RLock()
            for client := range manager.clients {
                select {
                case client.Send <- message:
                default:
                    // 发送失败,关闭该连接
                    manager.unregister <- client
                }
            }
            manager.mutex.RUnlock()
        }
    }
}

func (manager *ClientManager) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        http.Error(w, "WebSocket 升级失败", http.StatusBadRequest)
        return
    }

    client := &Client{Conn: conn, Send: make(chan []byte, 10)}
    manager.register <- client

    go manager.readPump(client)
    go manager.writePump(client)
}

readPump 负责从客户端读取消息:

```go func (manager *ClientManager) readPump(client *Client) { defer func() { manager.unregister }

writePump 负责向客户端发送消息:

```go func (manager *ClientManager) writePump(client *Client) { defer func() { manager.unregister <- client client.Conn.Close() }() for message := range client.Send { err := client.Conn.WriteMessage(websocket.TextMessage, message) if err != nil { break } } }

主函数启动服务

完整启动一个 WebSocket 服务,监听 8080 端口。

func main() {
    manager := NewClientManager()
    go manager.Run()

    http.Handle("/ws", manager)
    println("WebSocket 服务启动在 :8080/ws")
    http.ListenAndServe(":8080", nil)
}

访问 ws://localhost:8080/ws 的客户端将被纳入管理,任意客户端发送消息,其他所有客户端都能收到。

基本上就这些。这个示例结构清晰,易于扩展,比如加入用户ID、分组广播、心跳检测等。不复杂但容易忽略的是并发安全和连接异常处理,这里通过互斥锁和 defer 已做基础保障。

相关专题

更多
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数组用法,想了解更多的相关内容,请阅读专题下面的文章。

253

2025.06.17

菜鸟裹裹入口以及教程汇总
菜鸟裹裹入口以及教程汇总

本专题整合了菜鸟裹裹入口地址及教程分享,阅读专题下面的文章了解更多详细内容。

0

2026.01.22

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Git 教程
Git 教程

共21课时 | 2.9万人学习

Git版本控制工具
Git版本控制工具

共8课时 | 1.5万人学习

Git中文开发手册
Git中文开发手册

共0课时 | 0人学习

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

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