From 021959233f38d1bbcfe6fbcac4c64ab524a70eac Mon Sep 17 00:00:00 2001 From: Keuin Date: Fri, 8 May 2020 15:44:24 +0800 Subject: Added script-based auto reboot support (after restoring). --- .../kbackupfabric/operation/BackupOperation.java | 3 +- .../kbackupfabric/operation/RestoreOperation.java | 12 ++++- .../abstracts/AbstractBlockingOperation.java | 4 +- .../kbackupfabric/util/ServerRestartUtil.java | 58 ++++++++++++++++++++++ 4 files changed, 73 insertions(+), 4 deletions(-) create mode 100644 src/main/java/com/keuin/kbackupfabric/util/ServerRestartUtil.java (limited to 'src/main/java/com/keuin/kbackupfabric') diff --git a/src/main/java/com/keuin/kbackupfabric/operation/BackupOperation.java b/src/main/java/com/keuin/kbackupfabric/operation/BackupOperation.java index bceca40..53b05b8 100644 --- a/src/main/java/com/keuin/kbackupfabric/operation/BackupOperation.java +++ b/src/main/java/com/keuin/kbackupfabric/operation/BackupOperation.java @@ -1,10 +1,10 @@ package com.keuin.kbackupfabric.operation; +import com.keuin.kbackupfabric.exception.ZipUtilException; import com.keuin.kbackupfabric.metadata.BackupMetadata; import com.keuin.kbackupfabric.operation.abstracts.InvokableAsyncBlockingOperation; import com.keuin.kbackupfabric.util.PrintUtil; import com.keuin.kbackupfabric.util.ZipUtil; -import com.keuin.kbackupfabric.exception.ZipUtilException; import com.mojang.brigadier.context.CommandContext; import net.minecraft.server.MinecraftServer; import net.minecraft.server.command.ServerCommandSource; @@ -53,6 +53,7 @@ public class BackupOperation extends InvokableAsyncBlockingOperation { String levelPath = getLevelPath(server); String backupFileName = getBackupFileName(backupName); PrintUtil.info(String.format("zip(srcPath=%s, destPath=%s)", levelPath, backupSaveDirectoryFile.toString())); + msgInfo(context, "Compressing worlds ..."); PrintUtil.info("Compressing level ..."); ZipUtil.makeBackupZip(levelPath, backupSaveDirectoryFile.toString(), backupFileName, backupMetadata); File backupZipFile = new File(backupSaveDirectoryFile, backupFileName); diff --git a/src/main/java/com/keuin/kbackupfabric/operation/RestoreOperation.java b/src/main/java/com/keuin/kbackupfabric/operation/RestoreOperation.java index 42003db..6754393 100644 --- a/src/main/java/com/keuin/kbackupfabric/operation/RestoreOperation.java +++ b/src/main/java/com/keuin/kbackupfabric/operation/RestoreOperation.java @@ -1,9 +1,9 @@ package com.keuin.kbackupfabric.operation; +import com.keuin.kbackupfabric.exception.ZipUtilException; import com.keuin.kbackupfabric.operation.abstracts.InvokableBlockingOperation; import com.keuin.kbackupfabric.util.PrintUtil; import com.keuin.kbackupfabric.util.ZipUtil; -import com.keuin.kbackupfabric.exception.ZipUtilException; import com.mojang.brigadier.context.CommandContext; import net.minecraft.server.MinecraftServer; import net.minecraft.server.command.ServerCommandSource; @@ -121,6 +121,16 @@ public class RestoreOperation extends InvokableBlockingOperation { ZipUtil.unzip(backupFilePath, levelDirectory, false); long endTime = System.currentTimeMillis(); PrintUtil.info(String.format("Restore complete! (%.2fs) Please restart the server manually.", (endTime - startTime) / 1000.0)); + PrintUtil.info("If you want to restart automatically after restoring, please visit the project manual at: https://github.com/keuin/KBackup-Fabric/blob/master/README.md"); + + try { + Thread.sleep(1000); + } catch (InterruptedException ignored) { + } + + //ServerRestartUtil.forkAndRestart(); + System.exit(111); + } catch (SecurityException | IOException | ZipUtilException e) { PrintUtil.error("An exception occurred while restoring: " + e.getMessage()); } diff --git a/src/main/java/com/keuin/kbackupfabric/operation/abstracts/AbstractBlockingOperation.java b/src/main/java/com/keuin/kbackupfabric/operation/abstracts/AbstractBlockingOperation.java index de672cd..15b72c9 100644 --- a/src/main/java/com/keuin/kbackupfabric/operation/abstracts/AbstractBlockingOperation.java +++ b/src/main/java/com/keuin/kbackupfabric/operation/abstracts/AbstractBlockingOperation.java @@ -13,11 +13,11 @@ public abstract class AbstractBlockingOperation extends AbstractSerializedOperat protected final boolean operate() { synchronized (sync) { if (isBlocking) { - System.out.println("blocked."); +// System.out.println("blocked."); wasBlocked = true; return false; } else { - System.out.println("not blocked."); +// System.out.println("not blocked."); wasBlocked = false; isBlocking = true; } diff --git a/src/main/java/com/keuin/kbackupfabric/util/ServerRestartUtil.java b/src/main/java/com/keuin/kbackupfabric/util/ServerRestartUtil.java new file mode 100644 index 0000000..1642e5c --- /dev/null +++ b/src/main/java/com/keuin/kbackupfabric/util/ServerRestartUtil.java @@ -0,0 +1,58 @@ +package com.keuin.kbackupfabric.util; + +public class ServerRestartUtil { + + public static void forkAndRestart() { +// Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() { +// @Override +// public void run() { +// // Here we restart the minecraft server +// StringBuilder cmd = new StringBuilder(); +// cmd.append(System.getProperty("java.home")).append(File.separator).append("bin").append(File.separator).append("java "); +// for (String jvmArg : ManagementFactory.getRuntimeMXBean().getInputArguments()) { +// cmd.append(jvmArg + " "); +// } +// cmd.append("-cp ").append(ManagementFactory.getRuntimeMXBean().getClassPath()).append(" "); +// cmd.append(MinecraftServer.class.getName()).append(" "); +// for (String arg : args) { +// cmd.append(arg).append(" "); +// } +// Runtime.getRuntime().exec(cmd.toString()); +// System.exit(0); +// } +// })); + } + + private static void startRestartThread() { +// (new Thread(() -> { +// +// +// // kill threads +// Set threads = Thread.getAllStackTraces().keySet(); +// Thread currentThread = Thread.currentThread(); +// for (Thread t : threads) { +// if(t != currentThread && t.isAlive()) { +// t.setUncaughtExceptionHandler((t1, e) -> { +// // set empty handler +// }); +// //t.interrupt(); +// try { +// t.setDaemon(true); +// } catch (Exception ignored) { +// } +// +// t.stop(); +// } +// } +// +// try { +// Thread.sleep(5000); +// } catch (InterruptedException ignored) { +// } +// +// // restart Minecraft server +// String[] args = new String[]{}; +// MinecraftServer.main(args); +// })).start(); + } +} -- cgit v1.2.3