TotalFreedomMC/TF-EssentialsX
1
0
Fork 0
mirror of https://github.com/TotalFreedomMC/TF-EssentialsX.git synced 2025-02-11 20:00:46 +00:00
Code Issues Projects Releases Wiki Activity

[Experimental] Rewrite UUIDMap to use one single ScheduledExecutorService

Browse source
This commit is contained in:
vemacs 2016-03-01 17:45:48 -07:00
parent 526118e7d3
commit b8f7918a4a
2 changed files with 68 additions and 88 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
Essentials/src/com/earth2me/essentials/Essentials.java
View file

@ -304,7 +304,7 @@ public class Essentials extends JavaPlugin implements net.ess3.api.IEssentials {
} }
Economy.setEss(null); Economy.setEss(null);
Trade.closeLog(); Trade.closeLog();
getUserMap().getUUIDMap().forceWriteUUIDMap(); getUserMap().getUUIDMap().shutdown();
HandlerList.unregisterAll(this); HandlerList.unregisterAll(this);
} }

154
Essentials/src/com/earth2me/essentials/UUIDMap.java
View file

@ -9,7 +9,6 @@ import java.util.HashMap;
import java.util.Map; import java.util.Map;
import java.util.UUID; import java.util.UUID;
import java.util.concurrent.*; import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.logging.Level; import java.util.logging.Level;
import java.util.regex.Pattern; import java.util.regex.Pattern;
@ -18,13 +17,28 @@ public class UUIDMap {
private final transient net.ess3.api.IEssentials ess; private final transient net.ess3.api.IEssentials ess;
private File userList; private File userList;
private final transient Pattern splitPattern = Pattern.compile(","); private final transient Pattern splitPattern = Pattern.compile(",");
private static final ExecutorService EXECUTOR_SERVICE = Executors.newSingleThreadExecutor();
private final AtomicInteger pendingDiskWrites = new AtomicInteger(0); private static boolean pendingWrite;
private static final ScheduledExecutorService writeScheduler = Executors.newScheduledThreadPool(1);
private final Runnable writeTaskRunnable;
public UUIDMap(final net.ess3.api.IEssentials ess) { public UUIDMap(final net.ess3.api.IEssentials ess) {
this.ess = ess; this.ess = ess;
userList = new File(ess.getDataFolder(), "usermap.csv"); userList = new File(ess.getDataFolder(), "usermap.csv");
pendingWrite = false;
writeTaskRunnable = new Runnable() {
@Override
public void run() {
if (pendingWrite) {
try {
new WriteRunner(ess.getDataFolder(), userList, ess.getUserMap().getNames()).run();
} catch (Throwable t) { // bad code to prevent task from being suppressed
t.printStackTrace();
}
}
}
};
writeScheduler.scheduleWithFixedDelay(writeTaskRunnable, 5, 5, TimeUnit.SECONDS);
} }
public void loadAllUsers(final ConcurrentSkipListMap<String, UUID> names, final ConcurrentSkipListMap<UUID, ArrayList<String>> history) { public void loadAllUsers(final ConcurrentSkipListMap<String, UUID> names, final ConcurrentSkipListMap<UUID, ArrayList<String>> history) {
@ -33,41 +47,36 @@ public class UUIDMap {
userList.createNewFile(); userList.createNewFile();
} }
synchronized (pendingDiskWrites) { if (ess.getSettings().isDebug()) {
if (ess.getSettings().isDebug()) { ess.getLogger().log(Level.INFO, "Reading usermap from disk");
ess.getLogger().log(Level.INFO, "Reading usermap from disk"); }
}
names.clear(); names.clear();
history.clear(); history.clear();
final BufferedReader reader = new BufferedReader(new FileReader(userList)); try (BufferedReader reader = new BufferedReader(new FileReader(userList))) {
try { while (true) {
while (true) { final String line = reader.readLine();
final String line = reader.readLine(); if (line == null) {
if (line == null) { break;
break; } else {
} else { final String[] values = splitPattern.split(line);
final String[] values = splitPattern.split(line); if (values.length == 2) {
if (values.length == 2) { final String name = values[0];
final String name = values[0]; final UUID uuid = UUID.fromString(values[1]);
final UUID uuid = UUID.fromString(values[1]); names.put(name, uuid);
names.put(name, uuid); if (!history.containsKey(uuid)) {
if (!history.containsKey(uuid)) { final ArrayList<String> list = new ArrayList<>();
final ArrayList<String> list = new ArrayList<String>(); list.add(name);
history.put(uuid, list);
} else {
final ArrayList<String> list = history.get(uuid);
if (!list.contains(name)) {
list.add(name); list.add(name);
history.put(uuid, list);
} else {
final ArrayList<String> list = history.get(uuid);
if (!list.contains(name)) {
list.add(name);
}
} }
} }
} }
} }
} finally {
reader.close();
} }
} }
} catch (IOException ex) { } catch (IOException ex) {
@ -76,87 +85,58 @@ public class UUIDMap {
} }
public void writeUUIDMap() { public void writeUUIDMap() {
_writeUUIDMap(); pendingWrite = true;
} }
public void forceWriteUUIDMap() { public void forceWriteUUIDMap() {
if (ess.getSettings().isDebug()) { if (ess.getSettings().isDebug()) {
ess.getLogger().log(Level.INFO, "Forcing usermap write to disk"); ess.getLogger().log(Level.INFO, "Forcing usermap write to disk");
} }
try { pendingWrite = true;
Future<?> future = _writeUUIDMap(); writeTaskRunnable.run();
if (future != null) {
future.get();
}
} catch (InterruptedException ex) {
ess.getLogger().log(Level.SEVERE, ex.getMessage(), ex);
} catch (ExecutionException ex) {
ess.getLogger().log(Level.SEVERE, ex.getMessage(), ex);
}
} }
public Future<?> _writeUUIDMap() { public void shutdown() {
Map<String, UUID> names = ess.getUserMap().getNames(); writeScheduler.submit(writeTaskRunnable);
if (names.size() < 1) { writeScheduler.shutdown();
return null;
}
// The _names_ Map is being shallowly cloned because of an issue that occurs during restarts/reloads that
// causes UserMap database to be cleared. The assumed culprit is names.clear() when loading in UserMap.
//
// For more information, please refer to #213.
names = new HashMap<>(names);
pendingDiskWrites.incrementAndGet();
Future<?> future = EXECUTOR_SERVICE.submit(new WriteRunner(ess.getDataFolder(), userList, names, pendingDiskWrites));
return future;
} }
private static class WriteRunner implements Runnable { private static class WriteRunner implements Runnable {
private final File location; private final File location;
private final File endFile; private final File endFile;
private final Map<String, UUID> names; private final Map<String, UUID> names;
private final AtomicInteger pendingDiskWrites;
private WriteRunner(final File location, final File endFile, final Map<String, UUID> names, final AtomicInteger pendingDiskWrites) { private WriteRunner(final File location, final File endFile, final Map<String, UUID> names) {
this.location = location; this.location = location;
this.endFile = endFile; this.endFile = endFile;
this.names = names; this.names = new HashMap<>(names);
this.pendingDiskWrites = pendingDiskWrites;
} }
@Override @Override
public void run() { public void run() {
synchronized (pendingDiskWrites) { pendingWrite = false;
if (pendingDiskWrites.get() > 1) { File configFile = null;
pendingDiskWrites.decrementAndGet();
return; try {
configFile = File.createTempFile("usermap", ".tmp.csv", location);
final BufferedWriter bWriter = new BufferedWriter(new FileWriter(configFile));
for (Map.Entry<String, UUID> entry : names.entrySet()) {
bWriter.write(entry.getKey() + "," + entry.getValue().toString());
bWriter.newLine();
} }
File configFile = null; bWriter.close();
Files.move(configFile, endFile);
} catch (IOException ex) {
try { try {
configFile = File.createTempFile("usermap", ".tmp.csv", location); if (configFile != null && configFile.exists()) {
Files.move(configFile, new File(endFile.getParentFile(), "usermap.bak.csv"));
final BufferedWriter bWriter = new BufferedWriter(new FileWriter(configFile));
for (Map.Entry<String, UUID> entry : names.entrySet()) {
bWriter.write(entry.getKey() + "," + entry.getValue().toString());
bWriter.newLine();
} }
} catch (Exception ex2) {
bWriter.close(); Bukkit.getLogger().log(Level.SEVERE, ex2.getMessage(), ex2);
Files.move(configFile, endFile);
} catch (IOException ex) {
try {
if (configFile != null && configFile.exists()) {
Files.move(configFile, new File(endFile.getParentFile(), "usermap.bak.csv"));
}
} catch (Exception ex2) {
Bukkit.getLogger().log(Level.SEVERE, ex2.getMessage(), ex2);
}
Bukkit.getLogger().log(Level.WARNING, ex.getMessage(), ex);
} finally {
pendingDiskWrites.decrementAndGet();
} }
Bukkit.getLogger().log(Level.WARNING, ex.getMessage(), ex);
} }
} }
} }