fix: player data wasn't updated properly

1 files changed, 4 insertions, 0 deletions

diff --git a/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionHandlers.java b/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionHandlers.java
index 35311ac..3b82305 100644
--- a/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionHandlers.java
+++ b/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionHandlers.java
@@ -119,6 +119,8 @@ public class SessionHandlers {
             return;
         }
 
+        game.getSessionClient().getPlayerDataMap().put(event.getPlayerId(), event.getData());
+
         GameScreen screen = (GameScreen) game.getScreen();
         PlayerEntity entity = new PlayerEntity();
         entity.setPosition(event.getData().getState().getPosition());
@@ -131,6 +133,8 @@ public class SessionHandlers {
             log.log("The screen is not GameScreen, but the session received PlayerLeftRoomEvent");
             return;
         }
+        
+        game.getSessionClient().getPlayerDataMap().remove(event.getPlayerId());
 
         GameScreen screen = (GameScreen) game.getScreen();
         screen.getPlayerEntityMap().remove(event.getPlayerId());