添加推送服务

Author: hero

go_push/gateway/room.go

@@ -20,7 +20,7 @@ func newRoom(id int, title string) *Room {
 }
 
 func (r *Room) JoinRoom(ws *WsConnection) error {
-	if _, ok := r.RConn.Load(ws.connId); ok {
+	if _, ok := r.RConn.Load(ws.GetWsId()); ok {
 		return errors.New("already exists")
 	}
 	r.RConn.Store(ws.GetWsId(), ws)

go_push/gateway/room_manage.go

@@ -55,6 +55,7 @@ func (r *RoomManage) AddRoom(id int, wsId string) error {
 	if err := room.JoinRoom(ws); err != nil {
 		return err
 	}
+	ws.addRoom.Store(id, true)
 	return nil
 }