[go-websocket 搭建一对一，一对多的聊天室] 第二篇：websocket间的通信 ...

源码地址
https://gitee.com/bin-0821/chat-room-demo-go-websocket
关于websocket,上一篇文章讲述了如何通过websocket进行服务端与客户端的通信，本篇将会带领大家把各个websocket进行相互通信，在开始之前，请确保有理解
1 go的通道
2 go的线程
3 gin基础
事实上，websocket与websocket之间是无法进行直接相互通信的，需要我们将数据接收后，发送给另一个websocket链接，可以理解为

1. conn1.ReadJson(&data)
2. conn2.WriteJson(data)

复制代码

而建立一个类似微信聊天一样的，能进行多群聊，一对多，一对一的聊天,需要对websocket进行管理，本篇文章的重点便是如何管理好所有用户的websocket连接，主要有以下方面
1，一个根据业务进行设计的数据结构
2，用户上下线后conn的处理
3，用户发送信息的群发或单发
首先，要搞清楚我们在做什么，聊天室要实现的功能类似微信
a,b,c三人，1,2,3 三个聊天室

1. 人员        加入的聊天室
2. a        1,2,3
3. b        1,2
4. c        1

复制代码

a在1发送信息，全部人都能收到
a在2发送信息，c收不到，以此类推
a可以与c单独发送信息，接收方不在线时，系统能正常运行
1. 目录结构

相比上篇文章 多了manage_socket_conn，service两个模块，重点在于多了manage_socket_conn模块中

1. D:.
2. │  go.mod
3. │  go.sum
4. │  main.go
5. │  msg.json
6. ├─api
7. │      socket_conn.go
8. ├─manage_socket_conn        //用户的websocket管理模块
9. │      char_room_thread.go        //线程 主要负责对信息的群发
10. │      room_set.go        //聊天室房间管理，房间的创建，销毁 存储房间内的用户id
11. │      user_set.go        //用户websocket链接管理，信息的发送，存储所有在线的webscoket链接，用户上下线
12. ├─middleware
13. │      Cros.go
14. ├─model
15. │      socket_msg_form_front.go
16. │      to_front.go
17. ├─route
18. │      route.go
19. └─service
20.         chat_room.go        //数据层，模拟用户加入了那些聊天室

复制代码

2. 代码内容

user_set.go

1. package manage_socket_conn
3. import (
4.         "WebSocketDemo/model"
5.         "errors"
6.         "fmt"
7.         "github.com/gorilla/websocket"
8.         "sync"
9. )
11. func init() {
12.         GetUserSet()
13. }
14. //用户map 用来存储每个在线的用户id与对应的conn
15. type userSet struct {
16.         //        用户链接集  用户id => 链接对象
17.         users map[int]*websocket.Conn
18.         lock  sync.Mutex
19.         once  sync.Once
20. }
22. var us = new(userSet)
24. //        单例模式
25. func GetUserSet() *userSet {
26.         us.once.Do(func() {
28.                 us.users = make(map[int]*websocket.Conn)
29.                 us.users[-1] = nil
30.                 us.lock = sync.Mutex{}
31.         })
32.         return us
33. }
35. //        用户创建发起websocket连接
36. // join_type 加入模式
37. //                1 正常加入 占线无法加入
38. //                2 强制加入 即踢下线前者
39. func (u *userSet) ConnConnect(user_id, join_type int, conn *websocket.Conn) (int, error) {
40.         u.lock.Lock()
41.         defer u.lock.Unlock()
42.         if join_type == 1 {
43.                 //        用户id是否已经在线
44.                 if _, ok := u.users[user_id]; ok {
45.                         return 1, errors.New("该账号已被登陆")
46.                 }
47.         } else if join_type == 2 {
48.                 //        如果原用户id 已经存在map内 进行销毁挤出
49.                 if conn2, ok := u.users[user_id]; ok {
50.                         err := conn2.Close()
51.                         if err != nil {
52.                                 fmt.Println(err)
53.                         }
54.                         delete(u.users, user_id)
55.                 }
56.                 //        重新加入
57.                 u.users[user_id] = conn
58.         }
59.         return -1, nil
60. }
62. // 链接断开
63. func (u *userSet) ConnDisconnect(user_id int, conn *websocket.Conn) error {
64.         u.lock.Lock()
65.         defer u.lock.Unlock()
66.         if conn2, ok := u.users[user_id]; ok {
67.                 if conn == conn2 {
68.                         delete(u.users, user_id)
69.                 }
70.         } else {
71.                 //        Log  不存在的链接申请断开
72.         }
73.         return nil
74. }
76. //        对单个链接发送信息
77. func (u *userSet) SendMsgByUid(user_id int, msg interface{}) error {
78.         var err error
79.         if conn, ok := u.users[user_id]; ok {
80.                 err = conn.WriteJSON(msg)
81.         } else {
82.                 err = errors.New("不存在的链接")
83.         }
84.         return err
85. }
87. //        对多个连接发送信息
88. func (u *userSet) SendMsgByUidList(user_id_list []int, msg interface{}) (id_list []int, err_list []error) {
90.         for _, user_id := range user_id_list {
91.                 // 这里判断用户是否自己,是自己就跳过
92.                 c := msg.(model.ChatMsg)
93.                 if c.ChatMsgType == 1 {
94.                         if (c.Data["form_user_id"].(int)) == user_id {
95.                                 continue
96.                         }
97.                 }
98.                 if conn, ok := u.users[user_id]; ok {
99.                         err := conn.WriteJSON(msg)
100.                         if err != nil {
101.                                 id_list = append(id_list, user_id)
102.                                 err_list = append(err_list, err)
103.                         }
104.                 } else {
105.                         id_list = append(id_list, user_id)
106.                         err_list = append(err_list, errors.New("不存在的链接"))
107.                 }
108.         }
109.         return
110. }

复制代码

room_set.go
[code]package manage_socket_connimport (        "sync")//群map 用来存储每个群在线的用户idtype roomSet struct {        //                 群id                群内的用户id        rooms map[int]map[int]struct{}        lock sync.Mutex        once sync.Once}var rs = new(roomSet)//        单例func GetRoomSet() *roomSet{        rs.once.Do(func() {                rs.rooms = make(map[int]map[int]struct{})                rs.lock = sync.Mutex{}        })        return rs}//        向用户发送func (r *roomSet)SendMsgToUserList (r_id int ,msg interface{}){        userS := GetUserSet()        r.lock.Lock()        defer r.lock.Unlock()        var user_id_list []int        for key, _ := range r.rooms[r_id] {                user_id_list = append(user_id_list, key)        }        userS.SendMsgByUidList(user_id_list,msg)}//        用户下线/退群 退出聊天室链接集合func (r *roomSet) UserQuitRooms(room_id_list []int ,user_id int)  {        r.lock.Lock()        defer r.lock.Unlock()        for _, room_id := range room_id_list {                if v ,ok := r.rooms[room_id];ok {                        delete(v,user_id)                        //        房间没人就销毁                        if len(r.rooms[room_id])