Add players to non voted cache before checking if they are online

BenCodez · Feb 4, 2024 · 9f85149 · 9f85149
1 parent d6fa68d
commit 9f85149
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 0 deletions.
diff --git a/VotingPlugin/src/com/bencodez/votingplugin/bungee/NonVotedPlayersCache.java b/VotingPlugin/src/com/bencodez/votingplugin/bungee/NonVotedPlayersCache.java
@@ -28,6 +28,13 @@ public void addPlayer(ProxiedPlayer proxiedPlayer) {
 	public void addPlayer(String uuid, String playerName) {
 		getData().set("NonVotedPlayers." + playerName + ".UUID", uuid);
 		getData().set("NonVotedPlayers." + playerName + ".LastTime", System.currentTimeMillis());
+		save();
+	}
+
+	public void addPlayerCheck(String uuid, String playerName) {
+		if (!bungee.getMysql().containsKeyQuery(uuid)) {
+			addPlayer(uuid, playerName);
+		}
 	}
 
 	public void check() {

diff --git a/VotingPlugin/src/com/bencodez/votingplugin/bungee/VotingPluginBungee.java b/VotingPlugin/src/com/bencodez/votingplugin/bungee/VotingPluginBungee.java
@@ -873,6 +873,9 @@ protected void onMessage(String channel, String[] message) {
 									server = message[3];
 								}
 								debug("Login: " + player + "/" + uuid + " " + server);
+								if (nonVotedPlayersCache != null) {
+									nonVotedPlayersCache.addPlayerCheck(uuid, player);
+								}
 								ProxiedPlayer p = getProxy().getPlayer(player);
 								login(p);
 							}

diff --git a/VotingPlugin/src/com/bencodez/votingplugin/bungee/velocity/NonVotedPlayersCache.java b/VotingPlugin/src/com/bencodez/votingplugin/bungee/velocity/NonVotedPlayersCache.java
@@ -26,6 +26,12 @@ public void addPlayer(String uuid, String playerName) {
 		save();
 	}
 
+	public void addPlayerCheck(String uuid, String playerName) {
+		if (!plugin.getMysql().containsKeyQuery(uuid)) {
+			addPlayer(uuid, playerName);
+		}
+	}
+
 	public void check() {
 		for (String player : getKeys(getNode("NonVotedPlayers"))) {
 			long time = getNode("NonVotedPlayers", player, "LastTime").getLong(0);

diff --git a/VotingPlugin/src/com/bencodez/votingplugin/bungee/velocity/VotingPluginVelocity.java b/VotingPlugin/src/com/bencodez/votingplugin/bungee/velocity/VotingPluginVelocity.java
@@ -952,6 +952,9 @@ protected void onMessage(String channel, String[] message) {
 									serverName = message[3];
 								}
 								debug("Login: " + player + "/" + uuid + " " + serverName);
+								if (nonVotedPlayersCache != null) {
+									nonVotedPlayersCache.addPlayerCheck(uuid, player);
+								}
 								if (server.getPlayer(player).isPresent() && (getGlobalDataHandler() == null
 										|| !getGlobalDataHandler().isTimeChangedHappened())) {
 									Player p = server.getPlayer(player).get();