1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
package com.keuin.kbackupfabric.operation.backup.method;
import com.keuin.kbackupfabric.backup.incremental.ObjectCollection;
import com.keuin.kbackupfabric.backup.incremental.ObjectCollectionFactory;
import com.keuin.kbackupfabric.backup.incremental.ObjectCollectionSerializer;
import com.keuin.kbackupfabric.backup.incremental.identifier.Sha256Identifier;
import com.keuin.kbackupfabric.backup.incremental.manager.IncrementalBackupStorageManager;
import com.keuin.kbackupfabric.operation.backup.feedback.IncrementalBackupFeedback;
import com.keuin.kbackupfabric.util.FilesystemUtil;
import com.keuin.kbackupfabric.util.PrintUtil;
import com.keuin.kbackupfabric.util.ThreadingUtil;
import java.io.File;
import java.io.IOException;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.HashSet;
import java.util.logging.Logger;
public class ConfiguredIncrementalBackupMethod implements ConfiguredBackupMethod {
private final String backupIndexFileName;
private final String levelPath;
private final String backupIndexFileSaveDirectory;
private final String backupBaseDirectory;
private static final Logger LOGGER = Logger.getLogger(ConfiguredIncrementalBackupMethod.class.getName());
public ConfiguredIncrementalBackupMethod(String backupIndexFileName, String levelPath, String backupIndexFileSaveDirectory, String backupBaseDirectory) {
this.backupIndexFileName = backupIndexFileName;
this.levelPath = levelPath;
this.backupIndexFileSaveDirectory = backupIndexFileSaveDirectory;
this.backupBaseDirectory = backupBaseDirectory;
}
@Override
public IncrementalBackupFeedback backup() {
final int hashFactoryThreads = ThreadingUtil.getRecommendedThreadCount(); // how many threads do we use to generate the hash tree
LOGGER.info("Threads: " + hashFactoryThreads);
IncrementalBackupFeedback feedback;
try {
File levelPathFile = new File(levelPath);
// construct incremental backup index
PrintUtil.info("Hashing files...");
ObjectCollection collection = new ObjectCollectionFactory<>(Sha256Identifier.getFactory(), hashFactoryThreads, 16)
.fromDirectory(levelPathFile, new HashSet<>(Arrays.asList("session.lock", "kbackup_metadata")));
// update storage
PrintUtil.info("Copying files...");
IncrementalBackupStorageManager storageManager = new IncrementalBackupStorageManager(Paths.get(backupBaseDirectory));
int filesAdded = storageManager.addObjectCollection(collection, levelPathFile);
// save index file
PrintUtil.info("Saving index file...");
ObjectCollectionSerializer.toFile(collection, new File(backupIndexFileSaveDirectory, backupIndexFileName));
// return result
PrintUtil.info("Incremental backup finished.");
feedback = new IncrementalBackupFeedback(filesAdded >= 0, filesAdded);
} catch (IOException e) {
e.printStackTrace(); // at least we should print it out if we discard the exception... Better than doing nothing.
feedback = new IncrementalBackupFeedback(false, 0);
}
if (!feedback.isSuccess()) {
LOGGER.severe("Failed to backup.");
// do clean-up if failed
File backupIndexFile = new File(backupIndexFileSaveDirectory, backupIndexFileName);
if (backupIndexFile.exists()) {
if (!backupIndexFile.delete()) {
LOGGER.warning("Failed to clean up: cannot delete file " + backupIndexFile.getName());
}
}
//TODO: do more deep clean for object files
}
return feedback;
}
@Override
public boolean restore() throws IOException {
// load collection
PrintUtil.info("Loading file list...");
ObjectCollection collection = ObjectCollectionSerializer.fromFile(
new File(backupIndexFileSaveDirectory, backupIndexFileName)
);
// delete old level
File levelPathFile = new File(levelPath);
PrintUtil.info("Deleting old level...");
if (!FilesystemUtil.forceDeleteDirectory(levelPathFile)) {
PrintUtil.info("Failed to delete old level!");
return false;
}
// restore file
PrintUtil.info("Copying files...");
IncrementalBackupStorageManager storageManager = new IncrementalBackupStorageManager(Paths.get(backupBaseDirectory));
int restoreObjectCount = storageManager.restoreObjectCollection(collection, levelPathFile);
PrintUtil.info(String.format("%d file(s) restored.", restoreObjectCount));
return true;
}
@Override
public boolean touch() {
File baseDirectoryFile = new File(backupBaseDirectory);
return baseDirectoryFile.isDirectory() || baseDirectoryFile.mkdir();
}
@Override
public String getBackupFileName() {
return backupIndexFileName;
}
}
|
