feat: 예외처리 기능-1 #78

codesquad-members-2023 · Nov 7, 2023 · 7b6786a · 7b6786a
1 parent 9914a66
commit 7b6786a
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/be/src/main/java/codesquad/gaemimarble/game/controller/SocketDataSender.java b/be/src/main/java/codesquad/gaemimarble/game/controller/SocketDataSender.java
@@ -1,8 +1,6 @@
 package codesquad.gaemimarble.game.controller;
 
 import java.io.IOException;
-import java.util.HashSet;
-import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.ConcurrentMap;
 
@@ -42,8 +40,7 @@ public boolean saveSocket(Long gameId, String playerId, WebSocketSession session
 			boolean isDuplicate = socketMap.containsKey(playerId);
 
 			if (!isDuplicate) {
-				session.getAttributes().put("playerId", playerId);
-				sessions.add(session);
+				socketMap.put(playerId, session);
 				return true;
 			} else {
 				session.sendMessage(new TextMessage(objectMapper.writeValueAsString(