ThinkChat2.0新版上线,更智能更精彩,支持会话、画图、阅读、搜索等,送10W Token,即刻开启你的AI之旅 广告
用户连接流程: 客户端: ![](https://box.kancloud.cn/7b6831b06ba6a50edc74f63f9d692300_1078x858.png) ~~~ ***获取节点 getScript() http://localhost:8090/1/server/get?k=Terry-Mao&p=2 [先根据用户的key获取连接节点,然后再根据对应的节点创建长连接(参数p=1 websocket,p=2 tcp)] 创建websocket连接 获取离线消息 建立心跳任务 初始化完成(等待接收在线消息 or 心跳应答) ~~~ 服务端: ~~~ 建立连接: SubscribeHandle() 监听websocket请求 addr := ws.Request().RemoteAddr 获取用户ip地址 params := ws.Request().URL.Query() 获取url参数 key、heartbeat、token、version UserChannel.Get(key, true) 在ChannelList上添加一个Channel,并返回这个用户的 Channel c.AuthToken(key, token) token验证(默认不验证) c.AddConn(key, &Connection{Conn: ws, Proto: WebsocketProto, Version: version})创建一个用户连接, 1.判断是否超过最大连接数 2.用户链接应答 3.conn.HandleWrite(key) 开启 goroutine 从 Connection.Buf chan []byte 中读取消息,推送给用户 4.c.conn.PushFront(conn) root *Element 上添加一个 *Element 5.ConnStat.IncrAdd() 总连接数+1 for{} 阻塞等待心跳->心跳应答 获取离线消息: GetOfflineMsg() 获取离线消息 r.Method != "GET" 判断请求方式 params := r.URL.Query() 获取url参数 key、msgId、callback myrpc.MessageRPC.Get() 随机获取一个rpc连接 client.Call(myrpc.MessageServiceGetPrivate, args, reply) rpc 到message GetPrivate()方法 UseStorage.GetPrivate(m.Key, m.MsgId) 根据key和msgId从redis or mysql中获取消息 redis: 1.conn := s.getConn(key) 根据key从连接池获取一个redis连接(hash算法) 2.redis.Values(conn.Do("ZRANGEBYSCORE", key, fmt.Sprintf("(%d", mid), "+inf", "WITHSCORES")) 返回所有符合条件 mid < msgId <= +inf(最大值)) 的成员及成员的 message 3.redis.Scan(values, &b, &cmid) 遍历消息 4.json.Unmarshal(b, rm) 消息反序列化 5.rm.Expire < now 判断消息是否过期 6.s.delCH <- &RedisDelMessage{Key: key, MIds: delMsgs}: 删除unmarshal失败的消息和过期消息(clean() 方法) mysql: 1.s.getConn(key) 获取mysql连接 2.db.Query(getPrivateMsgSQL, key, mid) 查询skey=key and mid>msgId 的消息 "SELECT mid, ttl, msg FROM private_msg WHERE skey=? AND mid>? ORDER BY mid" 3.for rows.Next() 遍历消息 4.now > expire 判断消息是否过期 ~~~ 推送单个私信流程: ![](https://box.kancloud.cn/c0318cdbc8d3deb6b45f24f1aec906f1_1046x1070.png) ~~~ PushPrivate() 推送单个私信 r.Method != "POST" 判断是否是post请求 ioutil.ReadAll(r.Body) 读取请求内容 params := r.URL.Query() 获取url参数 key、expire node := myrpc.GetComet(key) 根据key获取comet节点 ???[判断key连接的comet] client := node.Rpc.Get() 随机获取一个rpc连接 client.Call(myrpc.CometServicePushPrivate, args, &ret) rpc 调用 comet 推送私信 PushPrivate() UserChannel.New(args.Key) 获取用户channel ch.PushMsg(args.Key, m, args.Expire) 推送私信 1.client := myrpc.MessageRPC.Get() 随机获取一个RPC连接 2.m.MsgId = id.Get() 生成msgId (时间戳/100) 3.m.GroupId != myrpc.PublicGroupId && expire > 0 判断是是否需要message保存(私信+过期时间>0) 4.client.Call(myrpc.MessageServiceSavePrivate, args, &ret) rpc 调用 message 模块,保存消息 SavePrivate() 保存私信 UseStorage.SavePrivate(m.Key, m.Msg, m.MsgId, m.Expire) redis: 1.conn := s.getConn(key) 根据key 通过hash算法从连接池获取一个连接 2.conn.Send("ZADD", key, mid, m) 操作写入缓冲区 3.conn.Send("ZREMRANGEBYRANK", key, 0, -1*(Conf.RedisMaxStore+1)) // conn.Send("ZREMRANGEBYRANK", key, 0, -21) 有序集只剩下最后写入的20个成员 4.conn.Flush() 提交操作 5.conn.Receive() 接受redis应答 mysql: 1.db := s.getConn(key) 根据key 通过hash算法从连接池获取一个连接 2.b.Exec(savePrivateMsgSQL, key, mid, now.Unix()+int64(expire), []byte(msg), now, now) 存mysql 5. c.writeMsg(key, m) 推送在线消息 ~~~ 批量推送私信流程: ![](https://box.kancloud.cn/f5a22da3945d948ef58e6c43fe3ba2df_1072x1040.png) ~~~ PushMultiPrivate() 批量推送私信 r.Method != "POST" 判断是否是post请求 ioutil.ReadAll(r.Body) 读取请求内容 parseMultiPrivate(bodyBytes) 获取 keys 、 message 和 ret(错误码) 根据key获取node,通过node获取rpc 链接 rpc -> comet PushPrivates() 向多个key推送私信 UserChannel.New(key) 根据key获取一个channel 和 ChannelBucket 并保存到 bucketMap 遍历 bucketMap 开启 goroutine 存储消息 和 推送消息 1.获取请求消息 2.根据请求消息解析出keys和message 3.根据key匹配节点,存储到 map[node]keys 4.rpc 调用用节点的 PushPrivates() 方法 5.判断 key 所在的bucketChannel 返回 map[*ChannelBucket]*batchChannel 6.遍历 ChannelBucket 给keys 发消息,并rpc 调用 Message模块 存储消息 ~~~ writeMsg: 1.调用每一个 用户 Element 元素 2.将消息写入Connection.Buf chan []byte 3.HandleWrite() 消费 Buf chan 将消息推给用户 MsgId: 通过当前时间的纳秒时间戳(除) / 100 获得 存储时随存储到redis中,用于在用户获取离线消息时,匹配 MsgId 大于 用户所传参的值