mirror of
https://github.com/mcMMO-Dev/mcMMO.git
synced 2024-11-25 22:56:45 +01:00
UUID Conversion stuff
This commit is contained in:
parent
dc21e18cc2
commit
d3faff67ad
@ -110,6 +110,8 @@ public interface DatabaseManager {
|
|||||||
*/
|
*/
|
||||||
public void convertUsers(DatabaseManager destination);
|
public void convertUsers(DatabaseManager destination);
|
||||||
|
|
||||||
|
public boolean saveUserUUID(String userName, UUID uuid);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Retrieve the type of database in use. Custom databases should return CUSTOM.
|
* Retrieve the type of database in use. Custom databases should return CUSTOM.
|
||||||
*
|
*
|
||||||
|
@ -471,6 +471,46 @@ public final class FlatfileDatabaseManager implements DatabaseManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public boolean saveUserUUID(String userName, UUID uuid) {
|
||||||
|
boolean worked = false;
|
||||||
|
|
||||||
|
BufferedReader in = null;
|
||||||
|
FileWriter out = null;
|
||||||
|
String usersFilePath = mcMMO.getUsersFilePath();
|
||||||
|
|
||||||
|
synchronized (fileWritingLock) {
|
||||||
|
try {
|
||||||
|
in = new BufferedReader(new FileReader(usersFilePath));
|
||||||
|
StringBuilder writer = new StringBuilder();
|
||||||
|
String line;
|
||||||
|
|
||||||
|
while ((line = in.readLine()) != null) {
|
||||||
|
// Write out the same file but when we get to the player we want to remove, we skip his line.
|
||||||
|
if (!worked && line.split(":")[0].equalsIgnoreCase(userName)) {
|
||||||
|
mcMMO.p.getLogger().info("User found, updating UUID...");
|
||||||
|
line.split(":")[41] = uuid.toString();
|
||||||
|
worked = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.append(line).append("\r\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
out = new FileWriter(usersFilePath); // Write out the new file
|
||||||
|
out.write(writer.toString());
|
||||||
|
}
|
||||||
|
catch (Exception e) {
|
||||||
|
mcMMO.p.getLogger().severe("Exception while reading " + usersFilePath + " (Are you sure you formatted it correctly?)" + e.toString());
|
||||||
|
}
|
||||||
|
finally {
|
||||||
|
tryClose(in);
|
||||||
|
tryClose(out);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
System.out.println("Saving " + userName + " | uuid = " + uuid.toString());
|
||||||
|
return worked;
|
||||||
|
}
|
||||||
|
|
||||||
public List<String> getStoredUsers() {
|
public List<String> getStoredUsers() {
|
||||||
ArrayList<String> users = new ArrayList<String>();
|
ArrayList<String> users = new ArrayList<String>();
|
||||||
BufferedReader in = null;
|
BufferedReader in = null;
|
||||||
|
@ -491,6 +491,10 @@ public final class SQLDatabaseManager implements DatabaseManager {
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public boolean saveUserUUID(String userName, UUID uuid) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Check connection status and re-establish if dead or stale.
|
* Check connection status and re-establish if dead or stale.
|
||||||
* <p/>
|
* <p/>
|
||||||
|
@ -36,6 +36,7 @@ import com.gmail.nossr50.runnables.CheckDateTask;
|
|||||||
import com.gmail.nossr50.runnables.SaveTimerTask;
|
import com.gmail.nossr50.runnables.SaveTimerTask;
|
||||||
import com.gmail.nossr50.runnables.UpdaterResultAsyncTask;
|
import com.gmail.nossr50.runnables.UpdaterResultAsyncTask;
|
||||||
import com.gmail.nossr50.runnables.backups.CleanBackupsTask;
|
import com.gmail.nossr50.runnables.backups.CleanBackupsTask;
|
||||||
|
import com.gmail.nossr50.runnables.database.UUIDUpdateAsyncTask;
|
||||||
import com.gmail.nossr50.runnables.database.UserPurgeTask;
|
import com.gmail.nossr50.runnables.database.UserPurgeTask;
|
||||||
import com.gmail.nossr50.runnables.party.PartyAutoKickTask;
|
import com.gmail.nossr50.runnables.party.PartyAutoKickTask;
|
||||||
import com.gmail.nossr50.runnables.player.PowerLevelUpdatingTask;
|
import com.gmail.nossr50.runnables.player.PowerLevelUpdatingTask;
|
||||||
@ -458,6 +459,9 @@ public class mcMMO extends JavaPlugin {
|
|||||||
long saveIntervalTicks = Config.getInstance().getSaveInterval() * 1200;
|
long saveIntervalTicks = Config.getInstance().getSaveInterval() * 1200;
|
||||||
new SaveTimerTask().runTaskTimer(this, saveIntervalTicks, saveIntervalTicks);
|
new SaveTimerTask().runTaskTimer(this, saveIntervalTicks, saveIntervalTicks);
|
||||||
|
|
||||||
|
// Slowly update every entry in the database with UUIDs
|
||||||
|
new UUIDUpdateAsyncTask(this).runTaskTimerAsynchronously(this, 5 * Misc.TICK_CONVERSION_FACTOR, 30 * Misc.TICK_CONVERSION_FACTOR);
|
||||||
|
|
||||||
// Cleanup the backups folder
|
// Cleanup the backups folder
|
||||||
new CleanBackupsTask().runTaskAsynchronously(mcMMO.p);
|
new CleanBackupsTask().runTaskAsynchronously(mcMMO.p);
|
||||||
|
|
||||||
|
@ -0,0 +1,51 @@
|
|||||||
|
package com.gmail.nossr50.runnables.database;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Map.Entry;
|
||||||
|
import java.util.UUID;
|
||||||
|
|
||||||
|
import org.bukkit.scheduler.BukkitRunnable;
|
||||||
|
|
||||||
|
import com.gmail.nossr50.mcMMO;
|
||||||
|
import com.gmail.nossr50.util.uuid.UUIDFetcher;
|
||||||
|
|
||||||
|
public class UUIDFetcherRunnable extends BukkitRunnable {
|
||||||
|
private List<String> names;
|
||||||
|
|
||||||
|
public UUIDFetcherRunnable(List<String> names) {
|
||||||
|
this.names = names;
|
||||||
|
}
|
||||||
|
|
||||||
|
public UUIDFetcherRunnable(String name) {
|
||||||
|
this.names = new ArrayList<String>();
|
||||||
|
this.names.add(name);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
try {
|
||||||
|
Map<String, UUID> returns = new UUIDFetcher(this.names).call();
|
||||||
|
new CacheReturnedNames(returns).runTask(mcMMO.p);
|
||||||
|
}
|
||||||
|
catch (Exception e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private class CacheReturnedNames extends BukkitRunnable {
|
||||||
|
private Map<String, UUID> returns;
|
||||||
|
|
||||||
|
public CacheReturnedNames(Map<String, UUID> returns) {
|
||||||
|
this.returns = returns;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
for (Entry<String, UUID> entry : this.returns.entrySet()) {
|
||||||
|
mcMMO.getDatabaseManager().saveUserUUID(entry.getKey(), entry.getValue());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,76 @@
|
|||||||
|
package com.gmail.nossr50.runnables.database;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
import org.bukkit.scheduler.BukkitRunnable;
|
||||||
|
|
||||||
|
import com.gmail.nossr50.mcMMO;
|
||||||
|
import com.gmail.nossr50.config.HiddenConfig;
|
||||||
|
import com.gmail.nossr50.database.DatabaseManager;
|
||||||
|
import com.gmail.nossr50.datatypes.player.PlayerProfile;
|
||||||
|
import com.gmail.nossr50.util.Misc;
|
||||||
|
|
||||||
|
public class UUIDUpdateAsyncTask extends BukkitRunnable {
|
||||||
|
private mcMMO plugin;
|
||||||
|
private static final int MAX_LOOKUP = 5;
|
||||||
|
|
||||||
|
private DatabaseManager databaseManager;
|
||||||
|
private List<String> userNames;
|
||||||
|
private int size;
|
||||||
|
int checkedUsers;
|
||||||
|
long startMillis;
|
||||||
|
|
||||||
|
public UUIDUpdateAsyncTask(mcMMO plugin) {
|
||||||
|
this.plugin = plugin;
|
||||||
|
|
||||||
|
this.databaseManager = mcMMO.getDatabaseManager();
|
||||||
|
this.userNames = databaseManager.getStoredUsers();
|
||||||
|
this.size = userNames.size();
|
||||||
|
|
||||||
|
this.checkedUsers = 0;
|
||||||
|
this.startMillis = System.currentTimeMillis();
|
||||||
|
|
||||||
|
plugin.getLogger().info("Starting to check and update UUIDs, total amount of users: " + size);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
if (!conversionNeeded()) {
|
||||||
|
plugin.debug("No need to update database with UUIDs");
|
||||||
|
this.cancel();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
List<String> userNamesSection;
|
||||||
|
|
||||||
|
if (size > MAX_LOOKUP) {
|
||||||
|
userNamesSection = userNames.subList(size - MAX_LOOKUP, size);
|
||||||
|
size -= MAX_LOOKUP;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
userNamesSection = userNames.subList(0, size);
|
||||||
|
size = 0;
|
||||||
|
this.cancel();
|
||||||
|
}
|
||||||
|
|
||||||
|
for (String userName : userNamesSection) {
|
||||||
|
PlayerProfile profile = databaseManager.loadPlayerProfile(userName, false);
|
||||||
|
|
||||||
|
checkedUsers++;
|
||||||
|
|
||||||
|
if (profile == null || !profile.isLoaded() || profile.getUniqueId() != null) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
new UUIDFetcherRunnable(userName).runTaskAsynchronously(mcMMO.p);
|
||||||
|
}
|
||||||
|
|
||||||
|
Misc.printProgress(checkedUsers, DatabaseManager.progressInterval, startMillis);
|
||||||
|
}
|
||||||
|
|
||||||
|
private boolean conversionNeeded() {
|
||||||
|
plugin.debug("Checking if conversion is needed...");
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
87
src/main/java/com/gmail/nossr50/util/uuid/UUIDFetcher.java
Normal file
87
src/main/java/com/gmail/nossr50/util/uuid/UUIDFetcher.java
Normal file
@ -0,0 +1,87 @@
|
|||||||
|
package com.gmail.nossr50.util.uuid;
|
||||||
|
|
||||||
|
import java.io.DataOutputStream;
|
||||||
|
import java.io.InputStreamReader;
|
||||||
|
import java.net.HttpURLConnection;
|
||||||
|
import java.net.URL;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.UUID;
|
||||||
|
import java.util.concurrent.Callable;
|
||||||
|
|
||||||
|
import com.google.common.collect.ImmutableList;
|
||||||
|
import org.json.simple.JSONArray;
|
||||||
|
import org.json.simple.JSONObject;
|
||||||
|
import org.json.simple.JSONValue;
|
||||||
|
import org.json.simple.parser.JSONParser;
|
||||||
|
|
||||||
|
public class UUIDFetcher implements Callable<Map<String, UUID>> {
|
||||||
|
private static final int MAX_SEARCH = 100;
|
||||||
|
private static final String PROFILE_URL = "https://api.mojang.com/profiles/page/";
|
||||||
|
private static final String AGENT = "minecraft";
|
||||||
|
private final JSONParser jsonParser = new JSONParser();
|
||||||
|
private final List<String> names;
|
||||||
|
|
||||||
|
public UUIDFetcher(List<String> names) {
|
||||||
|
this.names = ImmutableList.copyOf(names);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Map<String, UUID> call() throws Exception {
|
||||||
|
Map<String, UUID> uuidMap = new HashMap<String, UUID>();
|
||||||
|
String body = buildBody(names);
|
||||||
|
for (int i = 1; i < MAX_SEARCH; i++) {
|
||||||
|
HttpURLConnection connection = createConnection(i);
|
||||||
|
writeBody(connection, body);
|
||||||
|
JSONObject jsonObject = (JSONObject) jsonParser.parse(new InputStreamReader(connection.getInputStream()));
|
||||||
|
JSONArray array = (JSONArray) jsonObject.get("profiles");
|
||||||
|
Number count = (Number) jsonObject.get("size");
|
||||||
|
|
||||||
|
if (count.intValue() == 0) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (Object profile : array) {
|
||||||
|
JSONObject jsonProfile = (JSONObject) profile;
|
||||||
|
String id = (String) jsonProfile.get("id");
|
||||||
|
String name = (String) jsonProfile.get("name");
|
||||||
|
UUID uuid = UUID.fromString(id.substring(0, 8) + "-" + id.substring(8, 12) + "-" + id.substring(12, 16) + "-" + id.substring(16, 20) + "-" + id.substring(20, 32));
|
||||||
|
uuidMap.put(name, uuid);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return uuidMap;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static void writeBody(HttpURLConnection connection, String body) throws Exception {
|
||||||
|
DataOutputStream writer = new DataOutputStream(connection.getOutputStream());
|
||||||
|
writer.write(body.getBytes());
|
||||||
|
writer.flush();
|
||||||
|
writer.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
private static HttpURLConnection createConnection(int page) throws Exception {
|
||||||
|
URL url = new URL(PROFILE_URL + page);
|
||||||
|
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
|
||||||
|
connection.setRequestMethod("POST");
|
||||||
|
connection.setRequestProperty("Content-Type", "application/json");
|
||||||
|
connection.setUseCaches(false);
|
||||||
|
connection.setDoInput(true);
|
||||||
|
connection.setDoOutput(true);
|
||||||
|
return connection;
|
||||||
|
}
|
||||||
|
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
private static String buildBody(List<String> names) {
|
||||||
|
List<JSONObject> lookups = new ArrayList<JSONObject>();
|
||||||
|
|
||||||
|
for (String name : names) {
|
||||||
|
JSONObject obj = new JSONObject();
|
||||||
|
obj.put("name", name);
|
||||||
|
obj.put("agent", AGENT);
|
||||||
|
lookups.add(obj);
|
||||||
|
}
|
||||||
|
|
||||||
|
return JSONValue.toJSONString(lookups);
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user