纲要IM 服务核心业务与架构设计即时通讯的核心业务私聊、群聊、消息已读/未读、在线/离线消息微服务拆分API服务、RPC服务、WebSocket服务开发顺序先构建WebSocket服务再完善API与RPCWebSocket协议原理双向通讯能力与浏览器原生支持与HTTP的关系依赖HTTP握手升级典型应用场景即时聊天、实时数据推送、长耗时任务结果通知基于Go-Zero的 IM 服务工程化使用go-zerolatest脚手架初始化API与RPC服务在API服务中集成WebSocket处理模块Go语言WebSocket实践引入gorilla/websocket库定义Upgrader并实现WebSocket读写循环使用 API 测试工具验证握手与消息收发总结IM 服务核心业务与架构设计即时通讯IM服务的核心是聊天它可细分为私聊和群聊两种模式。围绕聊天产生的消息又衍生出已读/未读状态追踪、在线消息与离线消息处理、用户在线/离线状态感知等关键业务。为了在微服务架构下清晰划分职责通常将 IM 系统拆分为三个独立的服务API服务面向客户端提供历史消息查询、离线消息拉取、会话列表等非实时性接口。RPC服务负责内部逻辑调度例如会话管理、消息持久化、已读状态更新等。WebSocket服务承担实时双向通信处理私聊、群聊消息的即时投递。三者的协作关系如下图所示HTTPWebSocketgRPCgRPC客户端API 服务WebSocket 服务RPC 服务数据库缓存在开发阶段推荐优先构建WebSocket服务因为它承载了聊天的核心交互待WebSocket服务能够完成基本的消息收发后再逐步完善API与RPC服务实现历史记录、离线消息等周边功能。WebSocket 协议原理双向通讯与浏览器支持WebSocket是一种在单个TCP连接上进行全双工通信的协议。与传统的HTTP请求-响应模式不同WebSocket允许服务端主动向客户端推送数据同时客户端也可以随时向服务端发送消息。所有现代浏览器均原生支持WebSocket这使得它在 Web 应用中成为实时通信的首选方案。为什么需要 WebSocket假设一个场景用户上传一个大文件后端需要执行保存、解析、计算等一系列耗时操作。若采用HTTP短连接整个处理时间可能超过Nginx或浏览器的超时阈值导致客户端无法获取处理结果。一种传统改进方案是“轮询”服务端将任务交给异步队列客户端每隔几秒发送一次HTTP请求查询任务状态。其流程如下异步队列服务端客户端异步队列服务端客户端loop[轮询]上传文件投递任务任务已提交查询状态未完成任务完成查询状态完成这种方式会产生大量无效请求增加服务端压力。而引入WebSocket后流程变为WebSocket 服务异步队列服务端客户端WebSocket 服务异步队列服务端客户端登录认证建立 WebSocket 连接上传文件投递任务任务完成通知推送结果客户端在登录后即建立WebSocket长连接任务处理完成后由服务端主动推送结果彻底避免了客户端的无效轮询。WebSocket 与 HTTP 的关系WebSocket通过HTTP协议完成握手Upgrade随后将协议切换为WebSocket复用底层的TCP连接。其协议栈关系可表示为传输层应用层UpgradeHTTPWebSocketTCP典型应用场景即时通讯私聊、群聊、客服系统等。实时数据看板监控大屏、股票行情、体育比分等对数据实时性要求高的场景。消息推送异步任务结果通知、系统告警、活动提醒等。基于 Go-Zero 的 IM 服务工程化go-zero是一个集成了API网关、RPC框架、熔断限流等功能的企业级微服务框架。在 IM 项目中我们可以利用go-zero的脚手架快速生成API和RPC服务骨架。项目初始化确保已安装go-zero的最新版工具链goctl。执行以下命令创建项目# 创建 API 服务$ goctl api new im-api# 创建 RPC 服务$ goctl rpc new im-rpc生成后的目录结构大致如下im-service/ ├── im-api/ │ ├── etc/ │ │ └── im-api.yaml │ ├── internal/ │ │ ├── config/ │ │ ├── handler/ │ │ ├── logic/ │ │ ├── svc/ │ │ └── types/ │ ├── imapi.go │ └── go.mod ├── im-rpc/ │ ├── etc/ │ │ └── im-rpc.yaml │ ├── internal/ │ │ ├── config/ │ │ ├── logic/ │ │ ├── server/ │ │ └── svc/ │ ├── imrpc.go │ ├── im.proto │ └── go.mod └── go.workWebSocket服务通常作为API服务的一个独立路由存在因为它需要复用HTTP端口进行协议升级。我们可以在im-api的handler中新增一个WebSocket处理函数。Go 语言 WebSocket 实践Go 标准库并未提供WebSocket实现业界广泛使用的是gorilla/websocket库。它提供了简洁的Upgrader接口能够将HTTP请求升级为WebSocket连接。引入依赖在im-api模块下执行$ go get github.com/gorilla/websocket定义 WebSocket 处理逻辑在im-api/internal/handler下创建ws_handler.go内容如下packagehandlerimport(net/httpgithub.com/gorilla/websocketgithub.com/zeromicro/go-zero/core/logx)varupgraderwebsocket.Upgrader{CheckOrigin:func(r*http.Request)bool{returntrue// 生产环境需根据实际需求校验来源},}funcWebSocketHandler(w http.ResponseWriter,r*http.Request){// 将 HTTP 连接升级为 WebSocketconn,err:upgrader.Upgrade(w,r,nil)iferr!nil{logx.Error(WebSocket upgrade error:,err)return}deferconn.Close()logx.Info(WebSocket client connected)for{// 读取客户端消息msgType,message,err:conn.ReadMessage()iferr!nil{logx.Error(ReadMessage error:,err)break}logx.Infof(Received: type%d, msg%s,msgType,string(message))// 原样回显实际项目中可在此接入 RPC 进行消息分发iferr:conn.WriteMessage(msgType,message);err!nil{logx.Error(WriteMessage error:,err)break}}}注册路由在im-api/internal/handler/routes.go中将WebSocketHandler挂载到指定路径// Code generated by goctl. DO NOT EDIT.packagehandlerimport(net/httpim-api/internal/svcgithub.com/zeromicro/go-zero/rest)funcRegisterHandlers(server*rest.Server,serverCtx*svc.ServiceContext){// ... 其他路由server.AddRoutes([]rest.Route{{Method:http.MethodGet,Path:/ws,Handler:WebSocketHandler,},},)}测试验证启动im-api服务后可使用ApiPost或Postman等工具创建WebSocket请求连接地址为ws://localhost:8888/ws。连接成功后发送文本消息服务端应能正常接收并回显表明WebSocket基础通道已搭建完毕。总结本文从 IM 业务的架构设计出发阐述了API、RPC、WebSocket三大服务的职责划分。随后深入分析了WebSocket协议的双向通信原理及其与HTTP的升级关系并通过异步任务通知的对比案例展现了WebSocket在实时场景中的优势。最后基于go-zero微服务框架和gorilla/websocket库给出了在API服务中集成WebSocket的完整实践步骤。至此IM 服务的实时通信基础已构建完成后续可在此基础上扩展消息路由、存储、已读确认等高级功能。
