refactor StompConnectionListener.java

Chat-Your-Way · Oct 30, 2024 · d54a6dd · d54a6dd
1 parent 01739b8
commit d54a6dd
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/main/java/com/chat/yourway/listener/StompConnectionListener.java b/src/main/java/com/chat/yourway/listener/StompConnectionListener.java
@@ -20,16 +20,14 @@ public class StompConnectionListener {
 
     @EventListener
     public void handleWebSocketConnectListener(SessionConnectEvent event) {
-        String email = getUserEmailFromEvent(event);
-        contactOnlineService.setUserOnline(email);
-        log.info("Contact [{}] is connected", email);
+        contactOnlineService.setUserOnline(getUserEmailFromEvent(event));
+        log.info("Contact [{}] is connected", getUserEmailFromEvent(event));
     }
 
     @EventListener
     public void handleWebSocketDisconnectListener(SessionDisconnectEvent event) {
-        String email = getUserEmailFromEvent(event);
-        contactOnlineService.setUserOffline(email);
-        log.info("Contact [{}] is disconnected", email);
+        contactOnlineService.setUserOffline(getUserEmailFromEvent(event));
+        log.info("Contact [{}] is disconnected", getUserEmailFromEvent(event));
     }
 
     private String getUserEmailFromEvent(AbstractSubProtocolEvent event) {