我有兩個目錄結構,需要將它們同步,即將源文件夾中新增或更改的文件和文件夾複製到目標文件夾中,並刪除源中不存在的目標文件和文件夾。然而,這個過程需要有一個「取消」按鈕,這將停止執行並回滾所有的更改。我四處搜索,發現了這個代碼,但我對它的實際工作原理的理解充其量是朦朧的。如何在java中實現可停止和可取消的文件夾同步?
//$Id: FileHelper.java 15522 2008-11-05 20:06:43Z hardy.ferentschik $
//Revised from hibernate search util
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.channels.FileChannel;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
/**
* Utility class for synchronizing files/directories.
*
* @author Emmanuel Bernard
* @author Sanne Grinovero
* @author Hardy Ferentschik
*/
public abstract class FileHelper {
private static final int FAT_PRECISION = 2000;
public static final long DEFAULT_COPY_BUFFER_SIZE = 16 * 1024 * 1024; // 16 MB
public static boolean areInSync(File source, File destination) throws IOException {
if (source.isDirectory()) {
if (!destination.exists()) {
return false;
}
else if (!destination.isDirectory()) {
throw new IOException(
"Source and Destination not of the same type:"
+ source.getCanonicalPath() + " , " + destination.getCanonicalPath()
);
}
String[] sources = source.list();
Set<String> srcNames = new HashSet<String>(Arrays.asList(sources));
String[] dests = destination.list();
// check for files in destination and not in source
for (String fileName : dests) {
if (!srcNames.contains(fileName)) {
return false;
}
}
boolean inSync = true;
for (String fileName : sources) {
File srcFile = new File(source, fileName);
File destFile = new File(destination, fileName);
if (!areInSync(srcFile, destFile)) {
inSync = false;
break;
}
}
return inSync;
}
else {
if (destination.exists() && destination.isFile()) {
long sts = source.lastModified()/FAT_PRECISION;
long dts = destination.lastModified()/FAT_PRECISION;
return sts == dts;
}
else {
return false;
}
}
}
public static void synchronize(File source, File destination, boolean smart) throws IOException {
synchronize(source, destination, smart, DEFAULT_COPY_BUFFER_SIZE);
}
public static void synchronize(File source, File destination, boolean smart, long chunkSize) throws IOException {
if (chunkSize <= 0) {
System.out.println("Chunk size must be positive: using default value.");
chunkSize = DEFAULT_COPY_BUFFER_SIZE;
}
if (source.isDirectory()) {
if (!destination.exists()) {
if (!destination.mkdirs()) {
throw new IOException("Could not create path " + destination);
}
}
else if (!destination.isDirectory()) {
throw new IOException(
"Source and Destination not of the same type:"
+ source.getCanonicalPath() + " , " + destination.getCanonicalPath()
);
}
String[] sources = source.list();
Set<String> srcNames = new HashSet<String>(Arrays.asList(sources));
String[] dests = destination.list();
//delete files not present in source
for (String fileName : dests) {
if (!srcNames.contains(fileName)) {
delete(new File(destination, fileName));
}
}
//copy each file from source
for (String fileName : sources) {
File srcFile = new File(source, fileName);
File destFile = new File(destination, fileName);
synchronize(srcFile, destFile, smart, chunkSize);
}
}
else {
if (destination.exists() && destination.isDirectory()) {
delete(destination);
}
if (destination.exists()) {
long sts = source.lastModified()/FAT_PRECISION;
long dts = destination.lastModified()/FAT_PRECISION;
//do not copy if smart and same timestamp and same length
if (!smart || sts == 0 || sts != dts || source.length() != destination.length()) {
copyFile(source, destination, chunkSize);
}
}
else {
copyFile(source, destination, chunkSize);
}
}
}
private static void copyFile(File srcFile, File destFile, long chunkSize) throws IOException {
FileInputStream is = null;
FileOutputStream os = null;
try {
is = new FileInputStream(srcFile);
FileChannel iChannel = is.getChannel();
os = new FileOutputStream(destFile, false);
FileChannel oChannel = os.getChannel();
long doneBytes = 0L;
long todoBytes = srcFile.length();
while (todoBytes != 0L) {
long iterationBytes = Math.min(todoBytes, chunkSize);
long transferredLength = oChannel.transferFrom(iChannel, doneBytes, iterationBytes);
if (iterationBytes != transferredLength) {
throw new IOException(
"Error during file transfer: expected "
+ iterationBytes + " bytes, only " + transferredLength + " bytes copied."
);
}
doneBytes += transferredLength;
todoBytes -= transferredLength;
}
}
finally {
if (is != null) {
is.close();
}
if (os != null) {
os.close();
}
}
boolean successTimestampOp = destFile.setLastModified(srcFile.lastModified());
if (!successTimestampOp) {
System.out.println("Could not change timestamp for {}. Index synchronization may be slow. " + destFile);
}
}
public static void delete(File file) {
if (file.isDirectory()) {
for (File subFile : file.listFiles()) {
delete(subFile);
}
}
if (file.exists()) {
if (!file.delete()) {
System.out.println("Could not delete {}" + file);
}
}
}
}
代碼功能爲標榜,但我不能確定的是:
我將如何實現取消?我能想到的最好的方法是引入一個synchronizationCancelled布爾變量,在按下取消按鈕時將設置爲true,並將synchronized()與檢查相關聯,如果檢測結果爲false,則會停止執行。
我該如何實現回滾?我可以想到的最好的辦法是將更改/刪除/新文件複製到一邊,然後在取消後將它們複製回原來的位置。
都是我的猜測實際上做了正確的方式,還有一些注意事項給那些我不知道的,還是應該取消和回退可以在這種情況下完全不同實現的?