X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=server%2Fruntime%2Forg.argeo.server.core%2Fsrc%2Fmain%2Fjava%2Forg%2Fargeo%2Fserver%2Fbackup%2FAbstractAtomicBackup.java;fp=server%2Fruntime%2Forg.argeo.server.core%2Fsrc%2Fmain%2Fjava%2Forg%2Fargeo%2Fserver%2Fbackup%2FAbstractAtomicBackup.java;h=dd47651a6be759d761dc41ed980abf7856d9d303;hb=9eb1585fe9b213d55398aedfdd48c998c30b46dc;hp=0000000000000000000000000000000000000000;hpb=2c6cfdbdc82a3ca09d7c8407386665f744002738;p=lgpl%2Fargeo-commons.git

diff --git a/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/backup/AbstractAtomicBackup.java b/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/backup/AbstractAtomicBackup.java
new file mode 100644
index 000000000..dd47651a6
--- /dev/null
+++ b/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/backup/AbstractAtomicBackup.java
@@ -0,0 +1,70 @@
+package org.argeo.server.backup;
+
+import org.apache.commons.vfs.FileObject;
+import org.apache.commons.vfs.FileSystemManager;
+import org.apache.commons.vfs.FileSystemOptions;
+import org.apache.commons.vfs.provider.sftp.SftpFileSystemConfigBuilder;
+import org.argeo.ArgeoException;
+
+/**
+ * Simplify atomic backups implementation, especially by managing VFS.
+ */
+public abstract class AbstractAtomicBackup implements AtomicBackup {
+	private String name;
+	private String compression = "bz2";
+
+	protected abstract void writeBackup(FileObject targetFo);
+
+	public AbstractAtomicBackup() {
+	}
+
+	public AbstractAtomicBackup(String name) {
+		this.name = name;
+	}
+
+	@Override
+	public String backup(FileSystemManager fileSystemManager,
+			String backupsBase, BackupContext backupContext,
+			FileSystemOptions opts) {
+		if (name == null)
+			throw new ArgeoException("Atomic backup name must be set");
+
+		FileObject targetFo = null;
+		try {
+			if (backupsBase.startsWith("sftp:"))
+				SftpFileSystemConfigBuilder.getInstance()
+						.setStrictHostKeyChecking(opts, "no");
+			if (compression == null || compression.equals("none"))
+				targetFo = fileSystemManager.resolveFile(backupsBase + '/'
+						+ backupContext.getRelativeFolder() + '/' + name, opts);
+			else if (compression.equals("bz2"))
+				targetFo = fileSystemManager.resolveFile("bz2:" + backupsBase
+						+ '/' + backupContext.getRelativeFolder() + '/' + name
+						+ ".bz2" + "!" + name, opts);
+			else if (compression.equals("gz"))
+				targetFo = fileSystemManager.resolveFile("gz:" + backupsBase
+						+ '/' + backupContext.getRelativeFolder() + '/' + name
+						+ ".gz" + "!" + name, opts);
+			else
+				throw new ArgeoException("Unsupported compression "
+						+ compression);
+
+			writeBackup(targetFo);
+
+			return targetFo.toString();
+		} catch (Exception e) {
+			throw new ArgeoException("Cannot backup " + name + " to "
+					+ targetFo, e);
+		} finally {
+			BackupUtils.closeFOQuietly(targetFo);
+		}
+	}
+
+	public void setName(String name) {
+		this.name = name;
+	}
+
+	public void setCompression(String compression) {
+		this.compression = compression;
+	}
+}