Go-Zero 项目开发22:用户群聊功能的实现与完善
纲要消息存储模型基于读扩散一条消息只存一份通过type字段区分私聊/群聊receiver_id在群聊时指向群ID。会话管理用户创建群或加入群时由im服务创建群会话并维护用户与群的会话关系。消息推送与并发优化利用go-zero内置的线程工具实现群消息的并发发送避免因群成员数量大导致的延迟。消息队列处理在taskMQ中增加群聊分支调用社交服务获取群成员列表完成消息扩散与落地。服务协作社交API服务在创建群、申请进群、处理群申请等成功回调中通过RPC调用im服务建立会话。涉及技术栈go-zero、go-zero/core/threading、WebSocket、Redis、MySQL、RPC。消息存储与扩散模型群聊消息采用读扩散方案所有群成员共享同一条消息记录避免为每个用户存储一份副本。与私聊相同消息记录在同一张chat_log表中通过两个字段区分场景type消息类型枚举值为private私聊和group群聊。receiver_id接收者 ID私聊时为对方的用户 ID群聊时替换为群 ID。这样客户端拉取群历史消息时只需按群 ID 和消息类型查询即可获得完整的群聊记录无需在写路径上为每个成员维护独立的收件箱。会话的建立与管理创建时机会话的触发来源于两个入口创建群创建者发起创建群操作后社交服务需要同时为群本身和创建者与群之间建立会话。加入群新成员通过申请并被批准后社交服务需要为该用户与群建立会话。无论在哪个入口最终都通过im服务提供的RPC接口完成会话的初始化。时序梳理数据库IM RPC社交 RPC社交 API客户端数据库IM RPC社交 RPC社交 API客户端alt[会话不存在][会话已存在]创建群/审批加入执行群业务逻辑返回群 IDCreateGroupConversation(groupId, userId)查询群会话是否已存在插入群会话记录为用户插入群会话关系成功直接返回操作完成项目结构速览apps/ ├─ social/ │ ├─ api/ # 社交 API 服务 │ │ ├─ internal/ │ │ │ ├─ config/ │ │ │ ├─ logic/ # 创建群、申请群、处理申请等逻辑 │ │ │ └─ svc/ │ │ └─ social.api │ └─ rpc/ # 社交 RPC 服务 │ ├─ internal/ │ │ ├─ logic/ # GetGroupUserList 等 │ │ └─ svc/ │ └─ social.proto └─ im/ └─ rpc/ # IM RPC 服务 ├─ internal/ │ ├─ config/ │ ├─ logic/ # CreateGroupConversation 等 │ ├─ mq/ # taskMQ 消费者 │ ├─ server/ # WebSocket 连接管理、并发推送 │ └─ svc/ ├─ model/ # 会话、用户会话模型 └─ im.proto代码实现IM 服务中的会话逻辑以下代码位于 im 的 RPC 服务中负责创建群会话并关联用户会话列表。文件internal/logic/creategroupconversationlogic.gopackagelogicimport(contextdatabase/sqlgithub.com/pkg/errorsgo-zero-shop/apps/im/rpc/internal/svcgo-zero-shop/apps/im/rpc/pbgithub.com/zeromicro/go-zero/core/logx)typeCreateGroupConversationLogicstruct{ctx context.Context svcCtx*svc.ServiceContext logx.Logger}funcNewCreateGroupConversationLogic(ctx context.Context,svcCtx*svc.ServiceContext)*CreateGroupConversationLogic{returnCreateGroupConversationLogic{ctx:ctx,svcCtx:svcCtx,Logger:logx.WithContext(ctx),}}// CreateGroupConversation 创建群会话func(l*CreateGroupConversationLogic)CreateGroupConversation(in*pb.CreateGroupConversationReq)(*pb.CreateGroupConversationResp,error){// 1. 检查群会话是否已存在existing,err:l.svcCtx.ConversationModel.FindOneByConversationId(l.ctx,in.GroupId)iferr!nil!errors.Is(err,sql.ErrNoRows){l.Logger.Errorf(查询群会话失败: %v,err)returnnil,errors.Wrap(err,查询会话失败)}ifexisting!nil{returnpb.CreateGroupConversationResp{},nil}// 2. 创建群会话groupConv:model.Conversation{ConversationId:in.GroupId,Type:constant.ChatTypeGroup,}if_,err:l.svcCtx.ConversationModel.Insert(l.ctx,groupConv);err!nil{l.Logger.Errorf(创建群会话失败: %v,err)returnnil,errors.Wrap(err,创建会话失败)}// 3. 为创建者添加群会话关系userConv:model.UserConversation{UserId:in.CreatorId,ConversationId:in.GroupId,Type:constant.ChatTypeGroup,}if_,err:l.svcCtx.UserConversationModel.Insert(l.ctx,userConv);err!nil{l.Logger.Errorf(为用户添加群会话失败: %v,err)returnnil,errors.Wrap(err,添加用户会话失败)}returnpb.CreateGroupConversationResp{},nil}说明代码中ConversationModel和UserConversationModel为 go-zero 生成的 model 层对象constant.ChatTypeGroup是定义在常量包中的枚举值。并发推送消息群聊消息需要推送给所有在线成员如果采用串行方式逐个发送延迟会随着人数线性增长。为此我们引入go-zero提供的线程工具进行并发控制。并发限制与配置在im服务的Server结构体中通过Option模式暴露并发度参数方便运维调整。// internal/config/config.gotypeConfigstruct{// ... 其他配置ConcurrencyLimitintjson:ConcurrencyLimit}// internal/server/option.gotypeOptionstruct{ConcurrencyLimitint}funcWithConcurrencyLimit(limitint)Option{returnfunc(s*Server){s.concurrencyLimitlimit}}消息发送逻辑重构推送方法原先只处理私聊现在通过类型判定的方式分流群聊部分使用TaskRunner并发调用私聊推送方法。// internal/server/message.gopackageserverimport(contextfmtgo-zero-shop/apps/im/rpc/internal/constantgo-zero-shop/apps/im/rpc/internal/svcgo-zero-shop/apps/im/rpc/pbgithub.com/zeromicro/go-zero/core/threading)typeMessageCenterstruct{svcCtx*svc.ServiceContext concurrencyLimitinttaskRunner*threading.TaskRunner}funcNewMessageCenter(svcCtx*svc.ServiceContext,limitint)*MessageCenter{returnMessageCenter{svcCtx:svcCtx,concurrencyLimit:limit,taskRunner:threading.NewTaskRunner(limit),}}// Push 消息推送入口func(m*MessageCenter)Push(ctx context.Context,msg*pb.ChatMessage)error{switchmsg.Type{caseconstant.ChatTypePrivate:returnm.pushPrivate(ctx,msg,msg.ReceiverId)caseconstant.ChatTypeGroup:returnm.pushGroup(ctx,msg)default:returnfmt.Errorf(不支持的消息类型: %d,msg.Type)}}// pushPrivate 私聊推送func(m*MessageCenter)pushPrivate(ctx context.Context,msg*pb.ChatMessage,receiverIdstring)error{conn,err:m.svcCtx.ConnectionManager.Get(receiverId)iferr!nil{// 用户离线可记录日志或丢弃returnnil}// 假设存在 packResponse 将消息序列化为 WebSocket 帧data,err:packResponse(msg)iferr!nil{returnerr}returnconn.WriteMessage(data)}// pushGroup 群聊推送func(m*MessageCenter)pushGroup(ctx context.Context,msg*pb.ChatMessage)error{// msg.Receivers 由上游填充包含剔除发送者后的所有成员 IDfor_,uid:rangemsg.Receivers{uid:uid// 防止闭包引用问题m.taskRunner.Schedule(func(){iferr:m.pushPrivate(ctx,msg,uid);err!nil{logx.WithContext(ctx).Errorf(群聊推送失败, receiver%s, err%v,uid,err)}})}returnnil}注释ConnectionManager是我们实现的局部连接管理组件负责根据用户 ID 查找对应的WebSocket连接。TaskRunner.Schedule使用channel控制并发数当队列满时调用方会被阻塞从而实现反压。消息队列的群聊支持为了提高可靠性消息先被投递到消息队列由taskMQ异步消费并完成持久化与推送。需要在消费端增加群聊类型的处理并通过社交RPC服务获取群成员列表。消费端骨架// internal/mq/task.gopackagemqimport(contextencoding/jsongo-zero-shop/apps/im/rpc/internal/constantgo-zero-shop/apps/im/rpc/internal/svcgo-zero-shop/apps/im/rpc/pbgithub.com/zeromicro/go-zero/core/logx)typeTaskHandlerstruct{svcCtx*svc.ServiceContext pushService*server.MessageCenter}func(h*TaskHandler)Handle(ctx context.Context,raw[]byte)error{varmsg pb.ChatMessageiferr:json.Unmarshal(raw,msg);err!nil{returnerr}switchmsg.Type{caseconstant.ChatTypePrivate:returnh.handlePrivate(ctx,msg)caseconstant.ChatTypeGroup:returnh.handleGroup(ctx,msg)default:returnnil}}func(h*TaskHandler)handlePrivate(ctx context.Context,msg*pb.ChatMessage)error{// 存储消息记录...returnh.pushService.Push(ctx,msg)}func(h*TaskHandler)handleGroup(ctx context.Context,msg*pb.ChatMessage)error{// 1. 获取群成员rpcResp,err:h.svcCtx.SocialRpc.GroupUserList(ctx,social_pb.GroupUserListReq{GroupId:msg.ReceiverId,})iferr!nil{logx.WithContext(ctx).Errorf(获取群成员失败: %v,err)returnerr}// 2. 过滤发送者构建接收列表varreceivers[]stringfor_,user:rangerpcResp.Users{ifuser.UserId!msg.SenderId{receiversappend(receivers,user.UserId)}}msg.Receiversreceivers// 3. 存储消息记录...// 4. 并发推送returnh.pushService.Push(ctx,msg)}配置社交 RPC 客户端在im的config和service context中引入社交RPC客户端。// internal/config/config.gotypeConfigstruct{// ...SocialRpc zrpc.RpcClientConf}// internal/svc/servicecontext.gotypeServiceContextstruct{Config config.Config SocialRpc socialpb.SocialClient// ...其他依赖}funcNewServiceContext(c config.Config)*ServiceContext{returnServiceContext{Config:c,SocialRpc:socialpb.NewSocialClient(zrpc.MustNewClient(c.SocialRpc).Conn()),}}社交服务触发会话建立im服务的会话创建接口需要通过具体业务行为触发。在社交API服务中当创建群、申请入群、处理入群申请成功后应异步回调im RPC建立会话。社交 API 中的调用逻辑以创建群为例其余两个场景类似。// internal/logic/creategrouplogic.go (社交 API)func(l*CreateGroupLogic)CreateGroup(req*types.CreateGroupReq)(*types.CreateGroupResp,error){// ... 创建群业务逻辑获得 groupIdgroupId:xxx// 调用 IM RPC 创建群会话_,err:l.svcCtx.ImRpc.CreateGroupConversation(l.ctx,im_pb.CreateGroupConversationReq{GroupId:groupId,CreatorId:req.CreatorId,})iferr!nil{l.Logger.Errorf(创建群会话失败, groupId%s, err%v,groupId,err)// 通常这里可容忍失败通过定时任务补偿}returntypes.CreateGroupResp{GroupId:groupId},nil}社交服务的 IM RPC 配置// internal/config/config.go (社交 API)typeConfigstruct{// ...ImRpc zrpc.RpcClientConf}// internal/svc/servicecontext.go (社交 API)typeServiceContextstruct{Config config.Config ImRpc impb.ImClient// ...}funcNewServiceContext(c config.Config)*ServiceContext{returnServiceContext{Config:c,ImRpc:impb.NewImClient(zrpc.MustNewClient(c.ImRpc).Conn()),}}总结群聊功能的实现本质上复用了私聊的存储与推送链路核心差异体现在三处会话建模在群创建/加入时通过 im 服务统一管理群会话与用户‑会话关系。消息扩散服务端根据群 ID 查询成员列表借助go-zero的并发工具高效推送。异步处理消息队列消费端区分消息类型调用社交服务获取最新成员列表保证成员变动的实时性。整套方案在保持代码简洁的同时充分利用了go-zero框架的微服务能力RPC调用、线程池、消息队列可以平稳支撑较大规模的群组聊天场景。

相关新闻

如何在macOS上降级老款iPhone和iPad:LeetDown终极指南

如何在macOS上降级老款iPhone和iPad:LeetDown终极指南

如何在macOS上降级老款iPhone和iPad:LeetDown终极指南 【免费下载链接】LeetDown a macOS app that downgrades A6 and A7 iDevices to OTA signed firmwares 项目地址: https://gitcode.com/gh_mirrors/le/LeetDown 还在为老旧的iPhone 5或iPad 4运行缓慢而…

2026/10/11 22:13:56 阅读更多 →
FlashGBX终极指南:如何轻松读写Game Boy/GB Advance游戏卡数据

FlashGBX终极指南:如何轻松读写Game Boy/GB Advance游戏卡数据

FlashGBX终极指南:如何轻松读写Game Boy/GB Advance游戏卡数据 【免费下载链接】FlashGBX Reads and writes Game Boy and Game Boy Advance cartridge data. Supported hardware: GBxCart RW, GBFlash, Joey Jr, Game Bub 项目地址: https://gitcode.com/gh_mirr…

2026/10/10 16:33:44 阅读更多 →
10分钟解决经典游戏兼容性难题:DDrawCompat现代系统兼容层深度解析

10分钟解决经典游戏兼容性难题:DDrawCompat现代系统兼容层深度解析

10分钟解决经典游戏兼容性难题:DDrawCompat现代系统兼容层深度解析 【免费下载链接】DDrawCompat DirectDraw and Direct3D 1-7 compatibility, performance and visual enhancements for Windows Vista, 7, 8, 10 and 11 项目地址: https://gitcode.com/gh_mirro…

2026/9/26 7:22:33 阅读更多 →

最新新闻

SpringBoot2+Vue3+MyBatis-Plus+MySQL8.0学生信息管理系统实战

SpringBoot2+Vue3+MyBatis-Plus+MySQL8.0学生信息管理系统实战

接手过一个学生信息管理系统的小项目,技术栈正好是SpringBoot2 Vue3 MyBatis-Plus MySQL8.0,标题里这套组合一眼看去很常见,但真正从零落地的时候,细节远比想象中多。当初团队里有人建议用老掉牙的JSP那套,也有人建…

2026/10/13 2:34:32 阅读更多 →
SpringBoot+Vue私房菜定制上门服务系统开发实战解析

SpringBoot+Vue私房菜定制上门服务系统开发实战解析

一个 SpringBoot Vue 的私房菜定制上门服务系统,光看标题可能觉得就是一个普通的管理系统,但真正动手做过的人会知道,这类“定制 上门 多角色”的业务模型,远比表面看起来复杂。它不是简单的 CRUD 堆砌,订单状态的流…

2026/10/13 2:34:32 阅读更多 →
ModernZ键盘快捷键绑定教程:input.conf中script-binding与script-message完全指南

ModernZ键盘快捷键绑定教程:input.conf中script-binding与script-message完全指南

音视频UI组件 【免费下载链接】ModernZ A sleek and modern OSC for mpv. This project is a fork of ModernX that enhances functionality by adding more features while preserving the core standards of mpvs OSC. 项目地址: https://gitcode.com/gh_mirrors/…

2026/10/13 2:34:32 阅读更多 →
数据结构第一章重难点全梳理:逻辑结构、存储结构与复杂度分析

数据结构第一章重难点全梳理:逻辑结构、存储结构与复杂度分析

很多人在学一门新课程时,最容易忽略的偏偏是第一张。第一张通常看起来“全是概念、没有代码”,感觉像是空话,于是翻两页就过去了,等后面真正写程序、做算法题时才发现什么都对不上。这次我把自己重新整理的《数据结构》第一章笔记…

2026/10/13 2:34:32 阅读更多 →
修仙AI小镇xianxia-AI-town:大模型驱动的修仙模拟器,NPC与玩家为何能平权?完整解读

修仙AI小镇xianxia-AI-town:大模型驱动的修仙模拟器,NPC与玩家为何能平权?完整解读

【免费下载链接】xianxia-AI-town-AI- 大模型驱动的npc决策,npc和玩家平权 项目地址: https://gitcode.com/gh_mirrors/xi/xianxia-AI-town-AI- 点击查看 免费下载 修仙AI小镇(xianxia-AI-town)是一款大模型驱动的修仙模拟游戏&a…

2026/10/13 2:34:32 阅读更多 →
SpringBoot+Vue学生成绩分析与可视化平台全解析

SpringBoot+Vue学生成绩分析与可视化平台全解析

每年毕业季我都能在技术社区里看到大量和“成绩管理系统”相关的提问,这个题目属于那种“看起来老套、实则非常能打”的典型选题。特别是把Java、Vue、SpringBoot这三个词绑在一起的时候,它考查的已经不只是“会写增删改查”,而是完整的前后端…

2026/10/13 2:33:32 阅读更多 →

日新闻

物联网系统集成避坑指南:从选型到验收的关键要点

物联网系统集成避坑指南:从选型到验收的关键要点

做了这么多年物联网系统集成的项目,我最大的感受是,大部分坑不是出在技术前沿,而是出在一些看起来很平常的地方。设备选了便宜的、协议没问清楚、现场没摸透、验收标准写得含糊——这些细枝末节,最后都会变成延期、超支和扯皮的大…

2026/10/13 0:00:57 阅读更多 →
ESP32舵机控制全攻略:从PWM原理到高级玩法与常见故障排查

ESP32舵机控制全攻略:从PWM原理到高级玩法与常见故障排查

如果你玩ESP32有一阵子了,很难不碰到舵机控制这道坎。做智能小车要转方向,做机械臂要抬关节,做云台要偏航俯仰,背后都是同一个东西在干活:舵机。而我见过太多人卡在“代码能编译、舵机就是不听话”这个阶段&#xff0c…

2026/10/13 0:00:57 阅读更多 →
CAN总线原理与STM32实战:抗干扰、仲裁、错误处理全解析

CAN总线原理与STM32实战:抗干扰、仲裁、错误处理全解析

1. 为什么CAN总线至今仍是工业现场的“硬通货”?——从汽车底盘到智能温室的真实需求你有没有拆过一辆老款燃油车的中控台?或者调试过某高校实验室里那台跑着Modbus协议却频频丢帧的PLC控制柜?又或者,在某个农业物联网项目里&…

2026/10/13 0:00:57 阅读更多 →

周新闻

流感时间序列预测实战:ARIMA/LSTM全流程拆解与避坑指南

流感时间序列预测实战:ARIMA/LSTM全流程拆解与避坑指南

简介:基于 ARIMA、LSTM、Transformer 等模型的流感时间序列预测 Python 源码,面向计算机相关专业课程设计与期末大作业学生,以及项目实战学习者。内容覆盖预处理、平稳性检验、定阶、残差分析、多模型对比预测的完整时序建模流程,…

2026/10/12 0:16:30 阅读更多 →
影刀RPA新手教程:键盘模拟输入实战——输入文本与模拟按键的区别

影刀RPA新手教程:键盘模拟输入实战——输入文本与模拟按键的区别

影刀RPA新手教程:键盘模拟输入实战——输入文本与模拟按键的区别 做影刀RPA自动化,十个新手有八个栽在"往输入框里填东西"这件事上:要么填不进去,要么填了一半,要么直接把原来内容追加在后面。这背后的根因&…

2026/10/12 0:16:38 阅读更多 →
影刀RPA新手教程:阅文起点小说数据采集实战——书籍信息与章节内容

影刀RPA新手教程:阅文起点小说数据采集实战——书籍信息与章节内容

影刀RPA新手教程:阅文起点小说数据采集实战——书籍信息与章节内容 1. 认识影刀:什么场景该用RPA采小说数据 起点中文网的页面结构相对稳定——分类榜单、书籍详情、章节内容三块独立页面,跳转链路清晰。这种场景非常适合影刀自动化&#x…

2026/10/13 1:58:08 阅读更多 →

月新闻

我发现了一个新思路:用 Remotion + Claude Code 像写代码一样自动化生成短视频

我发现了一个新思路:用 Remotion + Claude Code 像写代码一样自动化生成短视频

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

2026/10/12 21:05:09 阅读更多 →
Windows下 Codex 中 Chrome 和 Computer Use 插件不可用问题排查及解决参考方式:TaoToken 统一 Key 配置与验证

Windows下 Codex 中 Chrome 和 Computer Use 插件不可用问题排查及解决参考方式:TaoToken 统一 Key 配置与验证

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

2026/10/12 14:48:49 阅读更多 →
黑夜航拍船只数据集训练YOLOV5模型全流程解析

黑夜航拍船只数据集训练YOLOV5模型全流程解析

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

2026/10/12 18:11:17 阅读更多 →