Compare commits

15 Commits

13 changed files with 433 additions and 157 deletions

22
.idea/artifacts/SysBackup_jar.xml generated Normal file
View File

@@ -0,0 +1,22 @@
<component name="ArtifactManager">
<artifact type="jar" name="SysBackup:jar">
<output-path>$PROJECT_DIR$/out/artifacts/SysBackup_jar</output-path>
<root id="archive" name="SysBackup.jar">
<element id="module-output" name="SysBackup" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/com/google/j2objc/j2objc-annotations/1.3/j2objc-annotations-1.3.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/org/antlr/antlr4-runtime/4.7.2/antlr4-runtime-4.7.2.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/org/tomlj/tomlj/1.0.0/tomlj-1.0.0.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/com/google/guava/guava/31.1-jre/guava-31.1-jre.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/net/lingala/zip4j/zip4j/2.11.1/zip4j-2.11.1.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/com/google/code/findbugs/jsr305/3.0.2/jsr305-3.0.2.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/com/google/guava/failureaccess/1.0.1/failureaccess-1.0.1.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/commons-io/commons-io/2.11.0/commons-io-2.11.0.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/me/tongfei/progressbar/0.9.3/progressbar-0.9.3.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/org/jline/jline/3.21.0/jline-3.21.0.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/com/google/errorprone/error_prone_annotations/2.11.0/error_prone_annotations-2.11.0.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/org/checkerframework/checker-qual/3.12.0/checker-qual-3.12.0.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/org/apache/commons/commons-lang3/3.12.0/commons-lang3-3.12.0.jar" path-in-jar="/" />
<element id="extracted-dir" path="$MAVEN_REPOSITORY$/com/google/guava/listenablefuture/9999.0-empty-to-avoid-conflict-with-guava/listenablefuture-9999.0-empty-to-avoid-conflict-with-guava.jar" path-in-jar="/" />
</root>
</artifact>
</component>

61
pom.xml
View File

@@ -7,6 +7,7 @@
<groupId>net.horizoncode</groupId>
<artifactId>SysBackup</artifactId>
<version>1.0-SNAPSHOT</version>
<packaging>jar</packaging>
<name>SysBackup</name>
@@ -58,6 +59,11 @@
<artifactId>commons-lang3</artifactId>
<version>3.12.0</version>
</dependency>
<dependency>
<groupId>com.diogonunes</groupId>
<artifactId>JColor</artifactId>
<version>5.5.1</version>
</dependency>
<!-- testing dependencies -->
<dependency>
<groupId>junit</groupId>
@@ -70,44 +76,25 @@
<build>
<pluginManagement><!-- lock down plugins versions to avoid using Maven defaults (may be moved to parent pom) -->
<plugins>
<!-- clean lifecycle, see https://maven.apache.org/ref/current/maven-core/lifecycles.html#clean_Lifecycle -->
<plugin>
<artifactId>maven-clean-plugin</artifactId>
<version>3.1.0</version>
</plugin>
<!-- default lifecycle, jar packaging: see https://maven.apache.org/ref/current/maven-core/default-bindings.html#Plugin_bindings_for_jar_packaging -->
<plugin>
<artifactId>maven-resources-plugin</artifactId>
<version>3.0.2</version>
</plugin>
<plugin>
<artifactId>maven-compiler-plugin</artifactId>
<version>3.8.0</version>
</plugin>
<plugin>
<artifactId>maven-surefire-plugin</artifactId>
<version>2.22.1</version>
</plugin>
<plugin>
<artifactId>maven-jar-plugin</artifactId>
<version>3.0.2</version>
</plugin>
<plugin>
<artifactId>maven-install-plugin</artifactId>
<version>2.5.2</version>
</plugin>
<plugin>
<artifactId>maven-deploy-plugin</artifactId>
<version>2.8.2</version>
</plugin>
<!-- site lifecycle, see https://maven.apache.org/ref/current/maven-core/lifecycles.html#site_Lifecycle -->
<plugin>
<artifactId>maven-site-plugin</artifactId>
<version>3.7.1</version>
</plugin>
<plugin>
<artifactId>maven-project-info-reports-plugin</artifactId>
<version>3.0.0</version>
<groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-shade-plugin</artifactId>
<version>3.3.0</version>
<executions>
<execution>
<goals>
<goal>shade</goal>
</goals>
<configuration>
<shadedArtifactAttached>true</shadedArtifactAttached>
<transformers>
<transformer implementation="org.apache.maven.plugins.shade.resource.ManifestResourceTransformer">
<mainClass>Bootstrapper</mainClass>
</transformer>
</transformers>
</configuration>
</execution>
</executions>
</plugin>
</plugins>
</pluginManagement>

View File

@@ -1,9 +1,10 @@
import net.horizoncode.sysbackup.cli.CLIProcessor;
import net.horizoncode.sysbackup.SysBackup;
import java.net.URISyntaxException;
public class Bootstrapper {
public static void main(String[] args) {
CLIProcessor cliProcessor = new CLIProcessor();
cliProcessor.startCLI(args);
public static void main(String[] args) throws URISyntaxException {
new SysBackup().start(args);
}
}

View File

@@ -0,0 +1,29 @@
package net.horizoncode.sysbackup;
import lombok.Getter;
import net.horizoncode.sysbackup.cli.CLIProcessor;
import net.horizoncode.sysbackup.logging.Logger;
import java.io.File;
import java.net.URISyntaxException;
public class SysBackup {
@Getter private static final Logger logger = Logger.builder().logFile(new File("log")).build();
public void start(String[] args) throws URISyntaxException {
File jarFile =
new File(SysBackup.class.getProtectionDomain().getCodeSource().getLocation().toURI());
File executionPath =
new File(
new File(SysBackup.class.getProtectionDomain().getCodeSource().getLocation().toURI())
.getParent());
if (!jarFile.isFile()) getLogger().log(Logger.LogLevel.INFO, "Dev environment detected!");
CLIProcessor cliProcessor = new CLIProcessor();
cliProcessor.startCLI(
args, jarFile.isFile() ? executionPath : new File(System.getProperty("user.dir")));
}
}

View File

@@ -1,13 +1,17 @@
package net.horizoncode.sysbackup.cli;
import net.horizoncode.sysbackup.SysBackup;
import net.horizoncode.sysbackup.config.Config;
import net.horizoncode.sysbackup.logging.Logger;
import net.horizoncode.sysbackup.tasks.TaskBuilder;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.tomlj.Toml;
import org.tomlj.TomlParseResult;
import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.Locale;
import java.util.Objects;
@@ -25,86 +29,102 @@ public class CLIProcessor {
" java -jar sysbackup.jar generateTaskConf magento",
" java -jar sysbackup.jar backup magento"
};
for (String u : usage) {
System.out.println(u);
}
Arrays.stream(usage).forEach(System.out::println);
}
public void startCLI(String[] args) {
public void startCLI(String[] args, File executionPath) {
try {
if ((args == null) || (args.length == 0)) {
usage();
return;
}
Logger logger = SysBackup.getLogger();
for (int index = 0; index < args.length; index++) {
switch (args[index].toLowerCase(Locale.ROOT)) {
case "backup":
{
if (args.length <= 1) {
System.err.println("Please specify a output task config name!");
logger.log(Logger.LogLevel.WARN, "Please specify a output task config name!");
return;
}
String fileName = args[1];
File tasksFolder = new File("tasks");
File tasksFolder = new File(executionPath, "tasks");
if (!tasksFolder.exists())
if (!tasksFolder.mkdir()) System.err.println("Failed to create tasks folder!");
if (!tasksFolder.mkdir())
logger.log(Logger.LogLevel.ERROR, "Failed to create tasks folder!");
File taskFile = new File(tasksFolder, fileName + ".toml");
if (!taskFile.exists()) {
System.err.println("TaskFile " + fileName + ".toml does not exist!");
logger.log(Logger.LogLevel.ERROR, "TaskFile %s.toml does not exist!", fileName);
return;
}
logger.log(Logger.LogLevel.INFO, "setupping TaskBuilder...");
Config taskConfig = new Config(taskFile);
TaskBuilder taskBuilder = TaskBuilder.builder().taskConfig(taskConfig).build();
TaskBuilder taskBuilder =
TaskBuilder.builder()
.executionPath(executionPath)
.taskName(FilenameUtils.removeExtension(taskFile.getName()))
.taskConfig(taskConfig)
.build();
taskBuilder.start();
break;
}
case "generatetaskconf":
{
if (args.length <= 1) {
System.err.println("Please specify a output task config name!");
logger.log(Logger.LogLevel.ERROR, "Please specify a output task config name!");
return;
}
String fileName = args[1];
File tasksFolder = new File("tasks");
File tasksFolder = new File(executionPath, "tasks");
if (!tasksFolder.exists())
if (!tasksFolder.mkdir()) System.err.println("Failed to create tasks folder!");
System.out.println("Saving task config " + fileName + ".toml...");
if (!tasksFolder.mkdir())
logger.log(Logger.LogLevel.ERROR, "Failed to create tasks folder!");
logger.log(Logger.LogLevel.INFO, "Saving task config %s.toml...", fileName);
try {
FileUtils.copyInputStreamToFile(
Objects.requireNonNull(getClass().getResourceAsStream("/" + "exampletask.toml")),
Objects.requireNonNull(
getClass().getResourceAsStream("/" + "exampletask.toml")),
new File(tasksFolder, fileName + ".toml"));
System.out.println(fileName + ".toml saved!");
} catch (IOException exception) {
logger.log(Logger.LogLevel.ERROR, "Failed to save task config.");
}
logger.log(Logger.LogLevel.INFO, "%s.toml saved!", fileName);
break;
}
case "checktaskconf":
{
if (args.length <= 1) {
System.err.println("Please specify a output task config name!");
logger.log(Logger.LogLevel.ERROR, "Please specify a output task config name!");
return;
}
String fileName = args[1];
File tasksFolder = new File("tasks");
File tasksFolder = new File(executionPath, "tasks");
if (!tasksFolder.exists())
if (!tasksFolder.mkdir()) System.err.println("Failed to create tasks folder!");
if (!tasksFolder.mkdir())
logger.log(Logger.LogLevel.ERROR, "Failed to create tasks folder!");
File taskFile = new File(tasksFolder, fileName + ".toml");
if (!taskFile.exists()) {
System.err.println("TaskFile " + fileName + ".toml does not exist!");
logger.log(Logger.LogLevel.ERROR, "TaskFile %s.toml does not exist!", fileName);
return;
}
TomlParseResult toml;
try {
toml = Toml.parse(taskFile.toPath());
} catch (IOException e) {
System.err.println("failed to read TaskFile.");
logger.log(Logger.LogLevel.ERROR, "failed to read TaskFile.");
throw new RuntimeException(e);
}
if (toml.hasErrors()) {
System.err.printf(
"TaskFile checked: found %d issues!:\n", (long) toml.errors().size());
toml.errors().forEach(error -> System.err.println(error.toString()));
logger.log(
Logger.LogLevel.ERROR,
"TaskFile checked: found %d issues!:\n",
(long) toml.errors().size());
toml.errors().forEach(error -> logger.log(Logger.LogLevel.ERROR, error.toString()));
} else {
System.out.println("TaskFile checked successfully: no issues found!");
logger.log(Logger.LogLevel.INFO, "TaskFile checked successfully: no issues found!");
}
break;
}
@@ -116,6 +136,7 @@ public class CLIProcessor {
}
}
} catch (Throwable t) {
SysBackup.getLogger().log(Logger.LogLevel.ERROR, t.getMessage());
t.printStackTrace();
}
}

View File

@@ -0,0 +1,48 @@
package net.horizoncode.sysbackup.logging;
import com.diogonunes.jcolor.Ansi;
import com.diogonunes.jcolor.Attribute;
import lombok.Builder;
import lombok.Getter;
import org.apache.commons.io.FileUtils;
import java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.text.SimpleDateFormat;
import java.util.Date;
@Builder
public class Logger {
private File logFile;
public void log(LogLevel logLevel, String message, Object... args) {
SimpleDateFormat sdf = new SimpleDateFormat("HH:mm:ss");
String prefix =
String.format(
"[%s - %s] ",
sdf.format(new Date()), Ansi.colorize(logLevel.name(), logLevel.getColor()));
String line = prefix + message;
System.out.printf(line + "\r\n", args);
// append to logfile
try {
FileUtils.writeStringToFile(
logFile, line.replaceAll("\u001B\\[[;\\d]*m", "") + "\r\n", StandardCharsets.UTF_8, true);
} catch (IOException ex) {
ex.printStackTrace();
}
}
public enum LogLevel {
INFO(Attribute.CYAN_TEXT()),
WARN(Attribute.YELLOW_TEXT()),
ERROR(Attribute.RED_TEXT());
@Getter private final Attribute color;
LogLevel(Attribute color) {
this.color = color;
}
}
}

View File

@@ -2,13 +2,19 @@ package net.horizoncode.sysbackup.tasks;
import lombok.Builder;
import lombok.Getter;
import net.horizoncode.sysbackup.SysBackup;
import net.horizoncode.sysbackup.config.Config;
import net.horizoncode.sysbackup.logging.Logger;
import net.horizoncode.sysbackup.tasks.impl.DatabaseTask;
import net.horizoncode.sysbackup.tasks.impl.FileSystemTask;
import net.horizoncode.sysbackup.tasks.impl.VacuumTask;
import net.horizoncode.sysbackup.threading.ThreadPool;
import org.apache.commons.io.FilenameUtils;
import org.tomlj.TomlArray;
import java.io.File;
import java.text.SimpleDateFormat;
import java.time.temporal.ChronoUnit;
import java.util.Date;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.stream.IntStream;
@@ -19,14 +25,28 @@ public class TaskBuilder {
private final Config taskConfig;
private final String taskName;
@Builder.Default private final LinkedBlockingQueue<Task> taskList = new LinkedBlockingQueue<>();
@Builder.Default @Getter private final ThreadPool threadPool = new ThreadPool(3, 10);
private final File executionPath;
public void start() {
File backupDir = new File("backups");
Logger logger = SysBackup.getLogger();
File rootBackupDir = new File(executionPath, "backups");
if (!rootBackupDir.exists())
if (!rootBackupDir.mkdir()) {
logger.log(Logger.LogLevel.ERROR, "Failed to create root backup directory!");
System.exit(2);
}
File backupDir = new File(rootBackupDir, getTaskName());
if (!backupDir.exists())
if (!backupDir.mkdir()) {
System.err.println("Failed to create backups directory!");
logger.log(Logger.LogLevel.ERROR, "Failed to create backup directory!");
System.exit(2);
}
@@ -34,7 +54,10 @@ public class TaskBuilder {
new SimpleDateFormat(
getTaskConfig().getStringOrDefault("general.dateFormat", "yyyy-MM-dd HH-mm-ss"));
String fileName =
getTaskConfig().getStringOrDefault("filesystem.fileName", "{date} - {taskName}") + ".zip";
getTaskConfig().getStringOrDefault("general.outputFile", "{date} - {taskName}") + ".zip";
boolean doVAC = getTaskConfig().getBooleanOrDefault("vacuum.enabled", false);
boolean doFS = getTaskConfig().getBooleanOrDefault("filesystem.enabled", false);
boolean doDB = getTaskConfig().getBooleanOrDefault("mysql.enabled", false);
fileName =
fileName
@@ -45,32 +68,79 @@ public class TaskBuilder {
File outputFile = new File(backupDir, fileName);
if (getTaskConfig().getToml().contains("filesystem.targets")) {
if (doVAC) {
ChronoUnit unit =
ChronoUnit.valueOf(
getTaskConfig().getStringOrDefault("vacuum.unit", ChronoUnit.DAYS.name()));
int value = getTaskConfig().getIntOrDefault("vacuum.time", 5);
logger.log(
Logger.LogLevel.INFO, "Adding VacuumTask with lifetime of %d %s%n", value, unit.name());
taskList.add(
new VacuumTask(backupDir, unit, value) {
@Override
public void onDone() {
getThreadPool().getPool().submit(() -> executeNextTask());
}
});
}
if (doFS && getTaskConfig().getToml().contains("filesystem.targets")) {
logger.log(Logger.LogLevel.INFO, "Adding FileSystemTask...");
TomlArray filesArray = getTaskConfig().getArray("filesystem.targets");
IntStream.range(0, filesArray.size())
.forEach(
value -> {
String target = filesArray.getString(value);
logger.log(Logger.LogLevel.INFO, "Adding \"%s\"", target);
taskList.add(
new FileSystemTask(target, outputFile) {
@Override
public void onDone() {
executeNextTask();
getThreadPool().getPool().submit(() -> executeNextTask());
}
});
});
}
executeNextTask();
if (doDB) {
String database = getTaskConfig().getStringOrDefault("mysql.database", "");
String user = getTaskConfig().getStringOrDefault("mysql.user", "");
String password = getTaskConfig().getStringOrDefault("mysql.password", "");
if (!database.isEmpty() && !user.isEmpty() && !password.isEmpty()) {
DatabaseTask.DatabaseCredentials databaseCredentials =
DatabaseTask.DatabaseCredentials.builder()
.database(database)
.username(user)
.password(password.toCharArray())
.build();
logger.log(Logger.LogLevel.INFO, "Adding DatabaseTask for database \"%s\"", database);
taskList.add(
new DatabaseTask(databaseCredentials, outputFile) {
@Override
public void onDone() {
getThreadPool().getPool().submit(() -> executeNextTask());
}
});
} else {
logger.log(Logger.LogLevel.ERROR, "username, password or database is empty.");
}
}
getThreadPool().getPool().submit(this::executeNextTask);
}
private void executeNextTask() {
Task nextTask = taskList.poll();
if (nextTask != null) nextTask.start();
else {
System.out.println("Backup completed!");
System.out.println(0);
SysBackup.getLogger().log(Logger.LogLevel.INFO, "Backup completed!");
System.exit(0);
}
}
}

View File

@@ -2,17 +2,19 @@ package net.horizoncode.sysbackup.tasks.impl;
import lombok.Builder;
import lombok.Getter;
import me.tongfei.progressbar.ProgressBar;
import me.tongfei.progressbar.ProgressBarBuilder;
import me.tongfei.progressbar.ProgressBarStyle;
import net.horizoncode.sysbackup.SysBackup;
import net.horizoncode.sysbackup.logging.Logger;
import net.horizoncode.sysbackup.tasks.Task;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import net.lingala.zip4j.ZipFile;
import net.lingala.zip4j.progress.ProgressMonitor;
import org.apache.commons.lang3.RandomStringUtils;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.Charset;
import java.io.*;
import java.nio.charset.StandardCharsets;
import java.util.stream.Collectors;
@Getter
public class DatabaseTask extends Task {
@@ -29,32 +31,104 @@ public class DatabaseTask extends Task {
@Override
public void start() {
try {
String[] commandArgs =
new String[] {
"mysqldump",
"-u " + getDatabaseCredentials().username,
"-p" + new String(getDatabaseCredentials().password),
getDatabaseCredentials().database
};
Logger logger = SysBackup.getLogger();
String commandArgs =
"mysqldump -u "
+ getDatabaseCredentials().username
+ " -p"
+ new String(getDatabaseCredentials().password)
+ " "
+ getDatabaseCredentials().database;
Runtime runtime = Runtime.getRuntime();
Process process = runtime.exec(commandArgs);
InputStream inputStream = process.getInputStream();
FileUtils.copyInputStreamToFile(
inputStream,
String databaseContent = "";
while (process.isAlive()) {
try {
Thread.sleep(1000);
databaseContent =
new BufferedReader(
new InputStreamReader(process.getInputStream(), StandardCharsets.UTF_8))
.lines()
.collect(Collectors.joining("\n"));
} catch (InterruptedException e) {
logger.log(Logger.LogLevel.ERROR, e.getMessage());
throw new RuntimeException(e);
}
}
int exitValue = process.exitValue();
if (exitValue != 0) {
String text =
new BufferedReader(
new InputStreamReader(process.getErrorStream(), StandardCharsets.UTF_8))
.lines()
.collect(Collectors.joining("\n"));
logger.log(Logger.LogLevel.ERROR, text);
onDone();
return;
}
if (databaseContent.isEmpty()) {
logger.log(Logger.LogLevel.ERROR, "database content is empty");
onDone();
return;
}
File outputSQLFile =
new File(
outputFile.getParent(),
String.format(
"%s-%s.sql",
getDatabaseCredentials().database, RandomStringUtils.random(16, true, true))));
getDatabaseCredentials().database, RandomStringUtils.random(16, true, true)));
BufferedWriter writer = new BufferedWriter(new FileWriter(outputSQLFile));
writer.write(databaseContent);
writer.close();
logger.log(Logger.LogLevel.INFO, "Adding database to backup zip...");
try (ZipFile zipFile = new ZipFile(getOutputFile())) {
ProgressMonitor progressMonitor = zipFile.getProgressMonitor();
zipFile.setRunInThread(true);
zipFile.addFile(outputSQLFile);
ProgressBarBuilder pbb =
new ProgressBarBuilder()
.setStyle(ProgressBarStyle.ASCII)
.setInitialMax(progressMonitor.getTotalWork())
.setTaskName("Adding DB File...")
.setUnit("MiB", 1048576);
try (ProgressBar pb = pbb.build()) {
while (!progressMonitor.getState().equals(ProgressMonitor.State.READY)) {
pb.stepTo(progressMonitor.getWorkCompleted());
Thread.sleep(100);
}
pb.stepTo(progressMonitor.getTotalWork());
} catch (Exception exception) {
logger.log(Logger.LogLevel.ERROR, exception.getMessage());
exception.printStackTrace();
outputSQLFile.deleteOnExit();
onDone();
}
progressMonitor.endProgressMonitor();
outputSQLFile.deleteOnExit();
onDone();
} catch (Exception ex) {
logger.log(Logger.LogLevel.ERROR, ex.getMessage());
ex.printStackTrace();
onDone();
}
} catch (IOException e) {
SysBackup.getLogger().log(Logger.LogLevel.ERROR, e.getMessage());
throw new RuntimeException(e);
}
}
@Builder
private static class DatabaseCredentials {
public static class DatabaseCredentials {
private final String username;
private final char[] password;
private final String database;

View File

@@ -3,16 +3,13 @@ package net.horizoncode.sysbackup.tasks.impl;
import me.tongfei.progressbar.ProgressBar;
import me.tongfei.progressbar.ProgressBarBuilder;
import me.tongfei.progressbar.ProgressBarStyle;
import me.tongfei.progressbar.TerminalUtils;
import net.horizoncode.sysbackup.SysBackup;
import net.horizoncode.sysbackup.logging.Logger;
import net.horizoncode.sysbackup.tasks.Task;
import net.horizoncode.sysbackup.threading.ThreadPool;
import net.lingala.zip4j.ZipFile;
import net.lingala.zip4j.progress.ProgressMonitor;
import org.jline.terminal.Terminal;
import org.jline.terminal.TerminalBuilder;
import java.io.File;
import java.io.IOException;
import java.nio.file.Paths;
public class FileSystemTask extends Task {
@@ -20,14 +17,13 @@ public class FileSystemTask extends Task {
private final File target;
private File outputZipFile;
private final ThreadPool threadPool;
public FileSystemTask(String folderOrFilePath, File outputZipFile) {
this.threadPool = new ThreadPool(3, 10);
this.target = Paths.get(folderOrFilePath).toFile();
Logger logger = SysBackup.getLogger();
if (!target.exists()) {
onDone();
System.err.println("File or folder named \"" + folderOrFilePath + "\" does not exist.");
logger.log(
Logger.LogLevel.ERROR, "File or folder named \"%s\" does not exist.", folderOrFilePath);
System.exit(2);
return;
}
@@ -37,22 +33,9 @@ public class FileSystemTask extends Task {
@Override
public void start() {
threadPool
.getPool()
.submit(
() -> {
int terminalWidth = -1;
try {
Terminal terminal = TerminalBuilder.terminal();
try (terminal) {
terminalWidth = terminal.getWidth();
}
} catch (IOException e) {
throw new RuntimeException(e);
}
Logger logger = SysBackup.getLogger();
try (ZipFile zipFile = new ZipFile(outputZipFile)) {
System.out.println("Indexing files...");
logger.log(Logger.LogLevel.INFO, "Indexing files...");
ProgressMonitor progressMonitor = zipFile.getProgressMonitor();
zipFile.setRunInThread(true);
zipFile.addFolder(target);
@@ -60,9 +43,8 @@ public class FileSystemTask extends Task {
new ProgressBarBuilder()
.setStyle(ProgressBarStyle.ASCII)
.setInitialMax(progressMonitor.getTotalWork())
.setTaskName("Adding Files...");
if (terminalWidth != -1) pbb.setMaxRenderedLength(terminalWidth);
.setTaskName("Adding Files...")
.setUnit("MiB", 1048576);
try (ProgressBar pb = pbb.build()) {
while (!progressMonitor.getState().equals(ProgressMonitor.State.READY)) {
@@ -71,15 +53,16 @@ public class FileSystemTask extends Task {
}
pb.stepTo(progressMonitor.getTotalWork());
} catch (Exception exception) {
logger.log(Logger.LogLevel.ERROR, exception.getMessage());
exception.printStackTrace();
onDone();
}
progressMonitor.endProgressMonitor();
onDone();
} catch (Exception ex) {
logger.log(Logger.LogLevel.ERROR, ex.getMessage());
ex.printStackTrace();
onDone();
}
});
}
}

View File

@@ -0,0 +1,38 @@
package net.horizoncode.sysbackup.tasks.impl;
import lombok.Getter;
import net.horizoncode.sysbackup.tasks.Task;
import java.io.File;
import java.time.Duration;
import java.time.temporal.ChronoUnit;
import java.util.Arrays;
import java.util.Objects;
@Getter
public class VacuumTask extends Task {
private final File backupDir;
private final ChronoUnit unit;
private final int value;
public VacuumTask(File backupDir, ChronoUnit unit, int value) {
this.backupDir = backupDir;
this.unit = unit;
this.value = value;
}
@Override
public void start() {
if (backupDir.listFiles() != null) {
Arrays.stream(Objects.requireNonNull(backupDir.listFiles()))
.filter(
file ->
file.lastModified() + Duration.of(value, unit).toMillis()
<= System.currentTimeMillis())
.forEachOrdered(File::deleteOnExit);
}
onDone();
}
}

View File

@@ -3,7 +3,6 @@ package net.horizoncode.sysbackup.threading;
import lombok.Getter;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledThreadPoolExecutor;
public class ThreadPool {
@@ -11,13 +10,10 @@ public class ThreadPool {
@Getter private final ExecutorService pool;
public ThreadPool(int corePoolSize, int maxPoolSize) {
this.pool = scheduledExecutorService(corePoolSize, maxPoolSize);
}
private ScheduledExecutorService scheduledExecutorService(int corePoolSize, int maxPoolSize) {
ScheduledThreadPoolExecutor scheduledThreadPoolExecutor =
new ScheduledThreadPoolExecutor(corePoolSize);
scheduledThreadPoolExecutor.setMaximumPoolSize(maxPoolSize);
return scheduledThreadPoolExecutor;
this.pool = scheduledThreadPoolExecutor;
}
}

View File

@@ -0,0 +1,3 @@
Manifest-Version: 1.0
Main-Class: Bootstrapper

View File

@@ -1,16 +1,20 @@
[general]
dateFormat = "yyyy-MM-dd"
dateFormat = "yyyy-MM-dd HH-mm-ss"
outputFile = "{date} - {taskName}"
[vacuum]
enabled = true
time = 5
unit = "DAYS" # See java.time.temporal.ChronoUnit
[mysql]
enabled = true
host = ""
port = 3306
database = "magento"
user = ""
password = ""
fileName = "{date} - {taskName}"
[filesystem]
enabled = true
targets = ["/home/magento/", "/home/test/test.ini"]
fileName = "{date} - {taskName}"