Added failsafe to waiting.

AuraDevelopmentTeam · Jun 15, 2017 · ee078ed · ee078ed
1 parent e8094c3
commit ee078ed
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/main/java/world/jnc/invsync/event/PlayerEvents.java b/src/main/java/world/jnc/invsync/event/PlayerEvents.java
@@ -175,6 +175,7 @@ public WaitingForPreviousServerToFinish(Player player, int maxWait) {
 		@Override
 		public void accept(Task task) {
 			boolean timeOk = endTime > System.currentTimeMillis();
+			UUID uuid = player.getUniqueId();
 
 			if (dataSource.isActive(player) && timeOk)
 				return;
@@ -187,13 +188,19 @@ public void accept(Task task) {
 				}
 
 				synchronized (waitingPlayers) {
-					waitingPlayers.remove(player.getUniqueId());
+					waitingPlayers.remove(uuid);
 				}
 
 				loadPlayer(player);
 			} catch (ClassNotFoundException | IOException | DataFormatException e) {
 				InventorySync.getLogger().warn("Loading player " + DataSource.getPlayerString(player) + " failed!", e);
 			} finally {
+				synchronized (waitingPlayers) {
+					if (waitingPlayers.containsKey(uuid)) {
+						waitingPlayers.remove(uuid);
+					}
+				}
+
 				task.cancel();
 			}
 		}