Refactor Importer/Exporter to use multiple threads (#317)

This commit is contained in:
Luck 2017-09-04 22:24:07 +01:00
parent dadd8e2c50
commit 6bc7f1fca2
No known key found for this signature in database
GPG Key ID: EFA9B3EC5FD90F8B
6 changed files with 256 additions and 108 deletions

View File

@ -35,6 +35,7 @@ import me.lucko.luckperms.common.model.User;
import me.lucko.luckperms.common.node.NodeFactory; import me.lucko.luckperms.common.node.NodeFactory;
import me.lucko.luckperms.common.plugin.LuckPermsPlugin; import me.lucko.luckperms.common.plugin.LuckPermsPlugin;
import me.lucko.luckperms.common.storage.Storage; import me.lucko.luckperms.common.storage.Storage;
import me.lucko.luckperms.common.utils.Cycle;
import java.io.BufferedWriter; import java.io.BufferedWriter;
import java.io.IOException; import java.io.IOException;
@ -44,7 +45,6 @@ import java.nio.file.Path;
import java.text.SimpleDateFormat; import java.text.SimpleDateFormat;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Collection; import java.util.Collection;
import java.util.Collections;
import java.util.Date; import java.util.Date;
import java.util.HashSet; import java.util.HashSet;
import java.util.List; import java.util.List;
@ -54,7 +54,6 @@ import java.util.concurrent.CompletableFuture;
import java.util.concurrent.atomic.AtomicInteger; import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.locks.ReentrantLock; import java.util.concurrent.locks.ReentrantLock;
import java.util.function.Consumer; import java.util.function.Consumer;
import java.util.function.Function;
/** /**
* Handles export operations * Handles export operations
@ -62,15 +61,6 @@ import java.util.function.Function;
public class Exporter implements Runnable { public class Exporter implements Runnable {
private static final SimpleDateFormat DATE_FORMAT = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss z"); private static final SimpleDateFormat DATE_FORMAT = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss z");
// number of users --> the value to divide the list by. base value of 1000 per thread, with a max of 10 threads. so, 3000 = 3 threads, 10000+ = 10 threads
private static final Function<Integer, Integer> THREAD_COUNT_FUNCTION = usersCount -> {
// how many threads to make. must be 1 <= x <= 15
int i = Math.max(1, Math.min(15, usersCount / 1000));
// then work out the value to split at. e.g. if we have 1,000 users to export and 2 threads, we should split at every 500 users.
return Math.max(1, usersCount / i);
};
private static void write(BufferedWriter writer, String s) { private static void write(BufferedWriter writer, String s) {
try { try {
writer.write(s); writer.write(s);
@ -109,27 +99,27 @@ public class Exporter implements Runnable {
// Create the actual groups first // Create the actual groups first
write(writer, "# Create groups"); write(writer, "# Create groups");
plugin.getGroupManager().getAll().values().stream()
.sorted((o1, o2) -> {
int i = Integer.compare(o2.getWeight().orElse(0), o1.getWeight().orElse(0));
return i != 0 ? i : o1.getName().compareToIgnoreCase(o2.getName());
})
.filter(g -> !g.getName().equals("default"))
.forEach(group -> {
write(writer, "/luckperms creategroup " + group.getName());
});
write(writer, "");
AtomicInteger groupCount = new AtomicInteger(0); AtomicInteger groupCount = new AtomicInteger(0);
// export groups in order of weight
plugin.getGroupManager().getAll().values().stream() plugin.getGroupManager().getAll().values().stream()
// export groups in order of weight
.sorted((o1, o2) -> { .sorted((o1, o2) -> {
int i = Integer.compare(o2.getWeight().orElse(0), o1.getWeight().orElse(0)); int i = Integer.compare(o2.getWeight().orElse(0), o1.getWeight().orElse(0));
return i != 0 ? i : o1.getName().compareToIgnoreCase(o2.getName()); return i != 0 ? i : o1.getName().compareToIgnoreCase(o2.getName());
}) })
// create all groups initially
.peek(group -> {
if (!group.getName().equals("default")) {
write(writer, "/luckperms creategroup " + group.getName());
}
})
// then export the content of each group
.forEach(group -> { .forEach(group -> {
if (groupCount.get() == 0) {
write(writer, "");
}
write(writer, "# Export group: " + group.getName()); write(writer, "# Export group: " + group.getName());
for (Node node : group.getEnduringNodes().values()) { for (Node node : group.getEnduringNodes().values()) {
write(writer, NodeFactory.nodeAsCommand(node, group.getName(), true, true)); write(writer, NodeFactory.nodeAsCommand(node, group.getName(), true, true));
@ -187,17 +177,13 @@ public class Exporter implements Runnable {
write(writer, "# Export users"); write(writer, "# Export users");
List<List<UUID>> subUsers; // divide into 16 pools.
AtomicInteger userCount = new AtomicInteger(0); Cycle<List<UUID>> userPools = new Cycle<>(Util.nInstances(16, ArrayList::new));
for (UUID uuid : users) {
// not really that many users, so it's not really worth spreading the load. userPools.next().add(uuid);
if (users.size() < 1500) {
subUsers = Collections.singletonList(new ArrayList<>(users));
} else {
subUsers = Util.divideList(users, THREAD_COUNT_FUNCTION.apply(users.size()));
} }
log.log("Split users into " + subUsers.size() + " threads for export."); log.log("Split users into " + userPools.getBacking().size() + " threads for export.");
// Setup a file writing lock. We don't want multiple threads writing at the same time. // Setup a file writing lock. We don't want multiple threads writing at the same time.
// The write function accepts a list of strings, as we want a user's data to be grouped together. // The write function accepts a list of strings, as we want a user's data to be grouped together.
@ -217,8 +203,10 @@ public class Exporter implements Runnable {
// A set of futures, which are really just the threads we need to wait for. // A set of futures, which are really just the threads we need to wait for.
Set<CompletableFuture<Void>> futures = new HashSet<>(); Set<CompletableFuture<Void>> futures = new HashSet<>();
AtomicInteger userCount = new AtomicInteger(0);
// iterate through each user sublist. // iterate through each user sublist.
for (List<UUID> subList : subUsers) { for (List<UUID> subList : userPools.getBacking()) {
// register and start a new thread to process the sublist // register and start a new thread to process the sublist
futures.add(CompletableFuture.runAsync(() -> { futures.add(CompletableFuture.runAsync(() -> {
@ -229,7 +217,7 @@ public class Exporter implements Runnable {
// actually export the user. this output will be fed to the writing function when we have all of the user's data. // actually export the user. this output will be fed to the writing function when we have all of the user's data.
List<String> output = new ArrayList<>(); List<String> output = new ArrayList<>();
plugin.getStorage().loadUser(uuid, "null").join(); plugin.getStorage().loadUser(uuid, null).join();
User user = plugin.getUserManager().getIfLoaded(uuid); User user = plugin.getUserManager().getIfLoaded(uuid);
output.add("# Export user: " + user.getUuid().toString() + " - " + user.getName().orElse("unknown username")); output.add("# Export user: " + user.getUuid().toString() + " - " + user.getName().orElse("unknown username"));

View File

@ -25,7 +25,6 @@
package me.lucko.luckperms.common.backup; package me.lucko.luckperms.common.backup;
import lombok.AccessLevel;
import lombok.Getter; import lombok.Getter;
import lombok.Setter; import lombok.Setter;
@ -37,13 +36,17 @@ import me.lucko.luckperms.common.commands.CommandResult;
import me.lucko.luckperms.common.commands.sender.Sender; import me.lucko.luckperms.common.commands.sender.Sender;
import me.lucko.luckperms.common.commands.utils.Util; import me.lucko.luckperms.common.commands.utils.Util;
import me.lucko.luckperms.common.locale.Message; import me.lucko.luckperms.common.locale.Message;
import me.lucko.luckperms.common.utils.DateUtil; import me.lucko.luckperms.common.plugin.LuckPermsPlugin;
import me.lucko.luckperms.common.utils.Cycle;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.HashMap; import java.util.HashSet;
import java.util.List; import java.util.List;
import java.util.Map;
import java.util.Set; import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import java.util.concurrent.atomic.AtomicInteger; import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@ -51,16 +54,11 @@ import java.util.stream.Collectors;
* Handles import operations * Handles import operations
*/ */
public class Importer implements Runnable { public class Importer implements Runnable {
private static final int PROGRESS_REPORT_SECONDS = 10;
private final CommandManager commandManager; private final CommandManager commandManager;
private final Set<Sender> notify; private final Set<Sender> notify;
private final List<String> commands; private final List<String> commands;
private final Map<Integer, Result> cmdResult; private final List<ImportCommand> toExecute;
private final ImporterSender fake;
private long lastMsg = 0;
private int executing = -1;
public Importer(CommandManager commandManager, Sender executor, List<String> commands) { public Importer(CommandManager commandManager, Sender executor, List<String> commands) {
this.commandManager = commandManager; this.commandManager = commandManager;
@ -77,11 +75,10 @@ public class Importer implements Runnable {
.filter(s -> !s.startsWith("//")) .filter(s -> !s.startsWith("//"))
.map(s -> s.startsWith("/") ? s.substring("/".length()) : s) .map(s -> s.startsWith("/") ? s.substring("/".length()) : s)
.map(s -> s.startsWith("perms ") ? s.substring("perms ".length()) : s) .map(s -> s.startsWith("perms ") ? s.substring("perms ".length()) : s)
.map(s -> s.startsWith("lp ") ? s.substring("lp ".length()) : s)
.map(s -> s.startsWith("luckperms ") ? s.substring("luckperms ".length()) : s) .map(s -> s.startsWith("luckperms ") ? s.substring("luckperms ".length()) : s)
.collect(Collectors.toList()); .collect(Collectors.toList());
this.toExecute = new ArrayList<>();
this.cmdResult = new HashMap<>();
this.fake = new ImporterSender(commandManager.getPlugin(), this::logMessage);
} }
@Override @Override
@ -89,35 +86,83 @@ public class Importer implements Runnable {
long startTime = System.currentTimeMillis(); long startTime = System.currentTimeMillis();
notify.forEach(s -> Message.IMPORT_START.send(s)); notify.forEach(s -> Message.IMPORT_START.send(s));
// form instances for all commands, and register them
int index = 1; int index = 1;
for (String command : commands) { for (String command : commands) {
long time = DateUtil.unixSecondsNow(); toExecute.add(new ImportCommand(commandManager.getPlugin(), index, command));
if (lastMsg < (time - PROGRESS_REPORT_SECONDS)) {
lastMsg = time;
sendProgress(index);
}
executing = index;
try {
CommandResult result = commandManager.onCommand(
fake,
"lp",
Util.stripQuotes(Splitter.on(CommandManager.COMMAND_SEPARATOR_PATTERN).omitEmptyStrings().splitToList(command))
).get();
getResult(index, command).setResult(result);
} catch (Exception e) {
getResult(index, command).setResult(CommandResult.FAILURE);
e.printStackTrace();
}
index++; index++;
} }
// divide commands up into pools
Cycle<List<ImportCommand>> commandPools = new Cycle<>(Util.nInstances(16, ArrayList::new));
String lastTarget = null;
for (ImportCommand cmd : toExecute) {
// if the last target isn't the same, skip to a new pool
if (lastTarget == null || !lastTarget.equals(cmd.getTarget())) {
commandPools.next();
}
commandPools.current().add(cmd);
lastTarget = cmd.getTarget();
}
// A set of futures, which are really just the threads we need to wait for.
Set<CompletableFuture<Void>> futures = new HashSet<>();
AtomicInteger processedCount = new AtomicInteger(0);
// iterate through each user sublist.
for (List<ImportCommand> subList : commandPools.getBacking()) {
// register and start a new thread to process the sublist
futures.add(CompletableFuture.runAsync(() -> {
// iterate through each user in the sublist, and grab their data.
for (ImportCommand cmd : subList) {
try {
CommandResult result = commandManager.onCommand(
cmd,
"lp",
Util.stripQuotes(Splitter.on(CommandManager.COMMAND_SEPARATOR_PATTERN).omitEmptyStrings().splitToList(cmd.getCommand()))
).get();
cmd.setResult(result);
} catch (Exception e) {
cmd.setResult(CommandResult.FAILURE);
e.printStackTrace();
}
processedCount.incrementAndGet();
}
}, commandManager.getPlugin().getScheduler().async()));
}
// all of the threads have been scheduled now and are running. we just need to wait for them all to complete
CompletableFuture<Void> overallFuture = CompletableFuture.allOf(futures.toArray(new CompletableFuture[futures.size()]));
while (true) {
try {
overallFuture.get(10, TimeUnit.SECONDS);
} catch (InterruptedException | ExecutionException e) {
// abnormal error - just break
e.printStackTrace();
break;
} catch (TimeoutException e) {
// still executing - send a progress report and continue waiting
sendProgress(processedCount.get());
continue;
}
// process is complete
break;
}
long endTime = System.currentTimeMillis(); long endTime = System.currentTimeMillis();
double seconds = (endTime - startTime) / 1000; double seconds = (endTime - startTime) / 1000;
int errors = (int) cmdResult.values().stream().filter(v -> v.getResult() != null && !v.getResult().asBoolean()).count(); int errors = (int) toExecute.stream().filter(v -> !v.getResult().asBoolean()).count();
if (errors == 0) { if (errors == 0) {
notify.forEach(s -> Message.IMPORT_END_COMPLETE.send(s, seconds)); notify.forEach(s -> Message.IMPORT_END_COMPLETE.send(s, seconds));
@ -128,11 +173,11 @@ public class Importer implements Runnable {
} }
AtomicInteger errIndex = new AtomicInteger(1); AtomicInteger errIndex = new AtomicInteger(1);
for (Map.Entry<Integer, Result> e : cmdResult.entrySet()) { for (ImportCommand e : toExecute) {
if (e.getValue().getResult() != null && !e.getValue().getResult().asBoolean()) { if (e.getResult() != null && !e.getResult().asBoolean()) {
notify.forEach(s -> { notify.forEach(s -> {
Message.IMPORT_END_ERROR_HEADER.send(s, errIndex.get(), e.getKey(), e.getValue().getCommand(), e.getValue().getResult().toString()); Message.IMPORT_END_ERROR_HEADER.send(s, errIndex.get(), e.getId(), e.getCommand(), e.getResult().toString());
for (String out : e.getValue().getOutput()) { for (String out : e.getOutput()) {
Message.IMPORT_END_ERROR_CONTENT.send(s, out); Message.IMPORT_END_ERROR_CONTENT.send(s, out);
} }
Message.IMPORT_END_ERROR_FOOTER.send(s); Message.IMPORT_END_ERROR_FOOTER.send(s);
@ -143,50 +188,77 @@ public class Importer implements Runnable {
} }
} }
private void sendProgress(int executing) { private void sendProgress(int processedCount) {
int percent = (executing * 100) / commands.size(); int percent = (processedCount * 100) / commands.size();
int errors = (int) cmdResult.values().stream().filter(v -> v.getResult() != null && !v.getResult().asBoolean()).count(); int errors = (int) toExecute.stream().filter(v -> !v.getResult().asBoolean()).count();
if (errors == 1) { if (errors == 1) {
notify.forEach(s -> Message.IMPORT_PROGRESS_SIN.send(s, percent, executing, commands.size(), errors)); notify.forEach(s -> Message.IMPORT_PROGRESS_SIN.send(s, percent, processedCount, commands.size(), errors));
} else { } else {
notify.forEach(s -> Message.IMPORT_PROGRESS.send(s, percent, executing, commands.size(), errors)); notify.forEach(s -> Message.IMPORT_PROGRESS.send(s, percent, processedCount, commands.size(), errors));
}
}
private Result getResult(int executing, String command) {
return cmdResult.compute(executing, (i, r) -> {
if (r == null) {
r = new Result(command);
} else {
if (!command.equals("") && r.getCommand().equals("")) {
r.setCommand(command);
}
}
return r;
});
}
private void logMessage(String msg) {
if (executing != -1) {
getResult(executing, "").getOutput().add(Util.stripColor(msg));
} }
} }
@Getter @Getter
@Setter private static class ImportCommand extends ImporterSender {
private static class Result { private static final Splitter SPACE_SPLIT = Splitter.on(" ");
private final int id;
private final String command;
private final String target;
@Setter(AccessLevel.NONE)
private final List<String> output = new ArrayList<>(); private final List<String> output = new ArrayList<>();
private String command; @Setter
private CommandResult result = CommandResult.FAILURE; private CommandResult result = CommandResult.FAILURE;
private Result(String command) { ImportCommand(LuckPermsPlugin plugin, int id, String command) {
super(plugin);
this.id = id;
this.command = command; this.command = command;
this.target = determineTarget(command);
} }
@Override
protected void consumeMessage(String s) {
output.add(s);
}
private static String determineTarget(String command) {
if (command.startsWith("user ") && command.length() > "user ".length()) {
String subCmd = command.substring("user ".length());
if (!subCmd.contains(" ")) {
return null;
}
String targetUser = SPACE_SPLIT.split(subCmd).iterator().next();
return "u:" + targetUser;
}
if (command.startsWith("group ") && command.length() > "group ".length()) {
String subCmd = command.substring("group ".length());
if (!subCmd.contains(" ")) {
return null;
}
String targetGroup = SPACE_SPLIT.split(subCmd).iterator().next();
return "g:" + targetGroup;
}
if (command.startsWith("track ") && command.length() > "track ".length()) {
String subCmd = command.substring("track ".length());
if (!subCmd.contains(" ")) {
return null;
}
String targetTrack = SPACE_SPLIT.split(subCmd).iterator().next();
return "t:" + targetTrack;
}
return null;
}
} }
} }

View File

@ -37,12 +37,12 @@ import me.lucko.luckperms.common.utils.TextUtils;
import net.kyori.text.Component; import net.kyori.text.Component;
import java.util.UUID; import java.util.UUID;
import java.util.function.Consumer;
@AllArgsConstructor @AllArgsConstructor
public class ImporterSender implements Sender { public abstract class ImporterSender implements Sender {
private final LuckPermsPlugin plugin; private final LuckPermsPlugin plugin;
private final Consumer<String> messageConsumer;
protected abstract void consumeMessage(String s);
@Override @Override
public LuckPermsPlugin getPlatform() { public LuckPermsPlugin getPlatform() {
@ -61,13 +61,13 @@ public class ImporterSender implements Sender {
@Override @Override
public void sendMessage(String s) { public void sendMessage(String s) {
messageConsumer.accept(s); consumeMessage(s);
} }
@SuppressWarnings("deprecation") @SuppressWarnings("deprecation")
@Override @Override
public void sendMessage(Component message) { public void sendMessage(Component message) {
messageConsumer.accept(TextUtils.toLegacy(message)); consumeMessage(TextUtils.toLegacy(message));
} }
@Override @Override

View File

@ -40,6 +40,7 @@ import java.util.List;
import java.util.ListIterator; import java.util.ListIterator;
import java.util.Map; import java.util.Map;
import java.util.UUID; import java.util.UUID;
import java.util.function.Supplier;
import java.util.regex.Pattern; import java.util.regex.Pattern;
@UtilityClass @UtilityClass
@ -111,6 +112,14 @@ public class Util {
return s == null ? null : STRIP_COLOR_PATTERN.matcher(s).replaceAll(""); return s == null ? null : STRIP_COLOR_PATTERN.matcher(s).replaceAll("");
} }
public static <T> List<T> nInstances(int count, Supplier<T> supplier) {
List<T> ret = new ArrayList<>(count);
for (int i = 0; i < count; i++) {
ret.add(supplier.get());
}
return ret;
}
public static <T> List<List<T>> divideList(Iterable<T> source, int size) { public static <T> List<List<T>> divideList(Iterable<T> source, int size) {
List<List<T>> lists = new ArrayList<>(); List<List<T>> lists = new ArrayList<>();
Iterator<T> it = source.iterator(); Iterator<T> it = source.iterator();

View File

@ -38,7 +38,7 @@ import java.util.Set;
@RequiredArgsConstructor @RequiredArgsConstructor
public class ProgressLogger { public class ProgressLogger {
private static final int NOTIFY_FREQUENCY = 100; private static final int NOTIFY_FREQUENCY = 500;
private final String pluginName; private final String pluginName;
private final Message logMessage; private final Message logMessage;

View File

@ -0,0 +1,79 @@
/*
* This file is part of LuckPerms, licensed under the MIT License.
*
* Copyright (c) lucko (Luck) <luck@lucko.me>
* Copyright (c) contributors
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
package me.lucko.luckperms.common.utils;
import com.google.common.collect.ImmutableList;
import java.util.List;
/**
* A cycle of elements, backed by a list. All operations are thread safe.
*
* @param <E> the element type
*/
public class Cycle<E> {
protected final List<E> objects;
protected int index = 0;
public Cycle(List<E> objects) {
if (objects == null || objects.isEmpty()) {
throw new IllegalArgumentException("List of objects cannot be null/empty.");
}
this.objects = ImmutableList.copyOf(objects);
}
public int getIndex() {
return index;
}
public E current() {
synchronized (this) {
return objects.get(index);
}
}
public E next() {
synchronized (this) {
index++;
index = index > objects.size() - 1 ? 0 : index;
return objects.get(index);
}
}
public E back() {
synchronized (this) {
index--;
index = index == -1 ? objects.size() - 1 : index;
return objects.get(index);
}
}
public List<E> getBacking() {
return objects;
}
}