@@ -109,13 +109,14 @@ func (ws *WebSocketHandler) messageHandler(conn *websocket.Conn, msg []byte) {
109
109
case MessageConnectRequest :
110
110
ws .registerUser (conn , v )
111
111
case MessageNewDataRequest :
112
- ws .sendDataToRoom (conn , v )
112
+ ws .sendDataToRoom (v )
113
113
case MessageSetLeaderRequest :
114
- ws .setLeader (conn , v )
114
+ ws .setLeader (v )
115
115
}
116
116
}
117
117
118
- func (ws * WebSocketHandler ) setLeader (conn * websocket.Conn , request MessageSetLeaderRequest ) {
118
+ //nolint:cyclop
119
+ func (ws * WebSocketHandler ) setLeader (request MessageSetLeaderRequest ) {
119
120
// Get the UserID from the JWT token
120
121
userID , err := ws .validateJWT (request .Jwt )
121
122
if err != nil {
@@ -146,13 +147,14 @@ func (ws *WebSocketHandler) setLeader(conn *websocket.Conn, request MessageSetLe
146
147
}
147
148
148
149
// Set the leader
149
- if currentRoom .LeaderID == "0" {
150
+ switch currentRoom .LeaderID {
151
+ case "0" :
150
152
currentRoom .SetLeader (userID )
151
153
ws .logger .Debug ("User set as the leader" , zap .String ("userID" , userID ), zap .String ("boardID" , request .BoardID ))
152
- } else if currentRoom . LeaderID == userID {
154
+ case userID :
153
155
currentRoom .SetLeader ("0" )
154
156
ws .logger .Debug ("Leader removed" , zap .String ("userID" , userID ), zap .String ("boardID" , request .BoardID ))
155
- } else {
157
+ default :
156
158
return
157
159
}
158
160
@@ -177,7 +179,8 @@ func (ws *WebSocketHandler) setLeader(conn *websocket.Conn, request MessageSetLe
177
179
}
178
180
}
179
181
180
- func (ws * WebSocketHandler ) sendDataToRoom (conn * websocket.Conn , request MessageNewDataRequest ) {
182
+ //nolint:cyclop
183
+ func (ws * WebSocketHandler ) sendDataToRoom (request MessageNewDataRequest ) {
181
184
// Get the UserID from the JWT token
182
185
userID , err := ws .validateJWT (request .Jwt )
183
186
if err != nil {
@@ -278,7 +281,7 @@ func (ws *WebSocketHandler) unregisterUser(conn *websocket.Conn) {
278
281
}
279
282
280
283
// Get the users ids
281
- var userIDs []string
284
+ userIDs := make ( []string , 0 )
282
285
for _ , u := range currentRoom .GetUsers () {
283
286
userIDs = append (userIDs , u .ID )
284
287
}
@@ -304,7 +307,8 @@ func (ws *WebSocketHandler) registerUser(conn *websocket.Conn, request MessageCo
304
307
305
308
// Check if the user has access to the board
306
309
if ! ws .validateBoardAccess (request .BoardID , request .Jwt ) {
307
- ws .logger .Debug ("User not allowed to access this board" , zap .String ("userID" , userID ), zap .String ("boardID" , request .BoardID ))
310
+ ws .logger .Debug ("User not allowed to access this board" ,
311
+ zap .String ("userID" , userID ), zap .String ("boardID" , request .BoardID ))
308
312
return
309
313
}
310
314
@@ -335,7 +339,7 @@ func (ws *WebSocketHandler) registerUser(conn *websocket.Conn, request MessageCo
335
339
currentRoom .AddUser (newUser )
336
340
337
341
// Get the users ids
338
- var userIDs []string
342
+ userIDs := make ( []string , 0 )
339
343
for _ , u := range currentRoom .GetUsers () {
340
344
userIDs = append (userIDs , u .ID )
341
345
}
@@ -375,26 +379,6 @@ func (ws *WebSocketHandler) sendUserConnected(request MessageUserConnectedRespon
375
379
}
376
380
}
377
381
378
- func (ws * WebSocketHandler ) sendUserFailedToConnect (conn * websocket.Conn , reason string ) {
379
- failedUser , err := ws .userStorage .GetWhere (func (u * user.User ) bool {
380
- return u .Conn == conn
381
- })
382
- if err != nil {
383
- return
384
- }
385
- err = failedUser .Conn .WriteJSON (MessageUserFailedToConnectResponse {
386
- Message : Message {
387
- Event : EventUserFailedToConnect ,
388
- },
389
- UserID : failedUser .ID ,
390
- Reason : reason ,
391
- })
392
- if err != nil {
393
- return
394
- }
395
- failedUser .Conn .Close ()
396
- }
397
-
398
382
func (ws * WebSocketHandler ) sendUserDisconnected (request MessageUserDisconnectedResponse ) {
399
383
// Get the room
400
384
currentRoom , _ := ws .roomStorage .Get (request .BoardID )
0 commit comments