Added: new scheduling and execution system.

Todo: testing and cleanup
2.x-1.16
szymon 2020-07-14 14:11:31 +02:00
parent 51ad310136
commit c07f80b872
7 changed files with 71 additions and 17 deletions

View File

@ -32,7 +32,7 @@ public class ConfigHandler {
@Comment("\nShould backups be done even if there are no players?\n")
public boolean doBackupsOnEmptyServer = false;
@Comment("\nShould backup be made on server shutdown\n")
@Comment("\nShould backup be made on server shutdown?\n")
public boolean shutdownBackup = true;
@Comment("\nA path to backup folder\n")

View File

@ -23,25 +23,36 @@ import io.github.cottonmc.cotton.config.ConfigManager;
import net.fabricmc.api.ModInitializer;
import net.fabricmc.fabric.api.command.v1.CommandRegistrationCallback;
import net.fabricmc.fabric.api.event.lifecycle.v1.ServerLifecycleEvents;
import net.fabricmc.fabric.api.event.lifecycle.v1.ServerTickEvents;
import net.minecraft.server.command.ServerCommandSource;
import net.szum123321.textile_backup.commands.BlacklistCommand;
import net.szum123321.textile_backup.commands.CleanupCommand;
import net.szum123321.textile_backup.commands.StartBackupCommand;
import net.szum123321.textile_backup.commands.WhitelistCommand;
import net.szum123321.textile_backup.core.BackupScheduler;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
public class TextileBackup implements ModInitializer {
public static final String MOD_ID = "textile_backup";
public static final Logger LOGGER = LogManager.getFormatterLogger("Textile Backup");
public static ConfigHandler config;
public static final BackupScheduler scheduler = new BackupScheduler();
public static final ExecutorService executorSerivece = Executors.newSingleThreadExecutor();
@Override
public void onInitialize() {
config = ConfigManager.loadConfig(ConfigHandler.class);
registerCommands();
ServerTickEvents.END_SERVER_TICK.register(scheduler::tick);
}
private void registerCommands(){

View File

@ -23,6 +23,7 @@ import com.mojang.brigadier.builder.LiteralArgumentBuilder;
import com.mojang.brigadier.context.CommandContext;
import net.minecraft.server.command.CommandManager;
import net.minecraft.server.command.ServerCommandSource;
import net.szum123321.textile_backup.TextileBackup;
import net.szum123321.textile_backup.core.BackupHelper;
public class StartBackupCommand {
@ -34,13 +35,27 @@ public class StartBackupCommand {
}
private static int executeWithComment(CommandContext<ServerCommandSource> source) {
BackupHelper.create(source.getSource().getMinecraftServer(), source.getSource(), true, StringArgumentType.getString(source, "comment").replace("#", ""));
TextileBackup.executorSerivece.submit(
BackupHelper.create(
source.getSource().getMinecraftServer(),
source.getSource(),
true,
StringArgumentType.getString(source, "comment").replace("#", "")
)
);
return 1;
}
private static int execute(ServerCommandSource source){
BackupHelper.create(source.getMinecraftServer(), source,true, null);
TextileBackup.executorSerivece.submit(
BackupHelper.create(
source.getMinecraftServer(),
source,
true,
null
)
);
return 1;
}

View File

@ -32,7 +32,7 @@ import java.util.Comparator;
import java.util.concurrent.atomic.AtomicInteger;
public class BackupHelper {
public static Thread create(MinecraftServer server, ServerCommandSource ctx, boolean save, String comment) {
public static Runnable create(MinecraftServer server, ServerCommandSource ctx, boolean save, String comment) {
LocalDateTime now = LocalDateTime.now();
StringBuilder builder = new StringBuilder();
@ -53,11 +53,7 @@ public class BackupHelper {
if (save)
server.save(true, true, false);
Thread thread = new Thread(new MakeBackupThread(server, ctx, comment));
thread.start();
return thread;
return new MakeBackupRunnable(server, ctx, comment);
}
public static void executeFileLimit(ServerCommandSource ctx, String worldName) {

View File

@ -0,0 +1,35 @@
package net.szum123321.textile_backup.core;
import net.minecraft.server.MinecraftServer;
import net.szum123321.textile_backup.TextileBackup;
import java.time.Instant;
public class BackupScheduler {
private boolean scheduled;
private long nextBackup;
public BackupScheduler() {
scheduled = false;
nextBackup = -1;
}
public void tick(MinecraftServer server) {
long now = Instant.now().getEpochSecond();
if(TextileBackup.config.doBackupsOnEmptyServer || server.getPlayerManager().getCurrentPlayerCount() > 0) {
if(scheduled) {
if(nextBackup >= now) {
TextileBackup.executorSerivece.submit(BackupHelper.create(server, null, true, null));
nextBackup = now + TextileBackup.config.backupInterval;
}
} else {
nextBackup = now + TextileBackup.config.backupInterval;
scheduled = true;
}
} else if(!TextileBackup.config.doBackupsOnEmptyServer && server.getPlayerManager().getCurrentPlayerCount() == 0) {
scheduled = false;
}
}
}

View File

@ -34,12 +34,12 @@ import java.io.File;
import java.io.IOException;
import java.time.LocalDateTime;
public class MakeBackupThread implements Runnable {
public class MakeBackupRunnable implements Runnable {
private final MinecraftServer server;
private final ServerCommandSource ctx;
private final String comment;
public MakeBackupThread(MinecraftServer server, ServerCommandSource ctx, String comment){
public MakeBackupRunnable(MinecraftServer server, ServerCommandSource ctx, String comment){
this.server = server;
this.ctx = ctx;
this.comment = comment;

View File

@ -32,6 +32,7 @@ import java.util.function.BooleanSupplier;
@Mixin(MinecraftServer.class)
public abstract class MinecraftServerMixin {
/*
@Shadow private long timeReference;
@Shadow public abstract PlayerManager getPlayerManager();
@ -49,15 +50,11 @@ public abstract class MinecraftServerMixin {
BackupHelper.create((MinecraftServer)(Object)this, null, true, null);
}
}
*/
@Inject(method = "shutdown", at = @At(value = "INVOKE_ASSIGN", target = "Lnet/minecraft/server/MinecraftServer;save(ZZZ)Z"))
public void onShutdown(CallbackInfo ci){
if(TextileBackup.config.shutdownBackup) {
try {
BackupHelper.create((MinecraftServer) (Object) this, null, false, "shutdown").join();
} catch (InterruptedException e) {
e.printStackTrace();
}
BackupHelper.create((MinecraftServer) (Object) this, null, false, "shutdown").run();
}
}
}