X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;ds=sidebyside;f=org.argeo.slc.runtime%2Fsrc%2Forg%2Fargeo%2Fslc%2Fbackup%2Fvfs%2FAbstractAtomicBackup.java;fp=org.argeo.slc.runtime%2Fsrc%2Forg%2Fargeo%2Fslc%2Fbackup%2Fvfs%2FAbstractAtomicBackup.java;h=6d55374244d2863970396ee3418531f7c0eb4526;hb=fbe96810dae8085b3d6935d085d21008fb809fc3;hp=0000000000000000000000000000000000000000;hpb=04e7ba0710e74d9f18322dcc2c730f972b7ea10c;p=gpl%2Fargeo-slc.git

diff --git a/org.argeo.slc.runtime/src/org/argeo/slc/backup/vfs/AbstractAtomicBackup.java b/org.argeo.slc.runtime/src/org/argeo/slc/backup/vfs/AbstractAtomicBackup.java
new file mode 100644
index 000000000..6d5537424
--- /dev/null
+++ b/org.argeo.slc.runtime/src/org/argeo/slc/backup/vfs/AbstractAtomicBackup.java
@@ -0,0 +1,82 @@
+package org.argeo.slc.backup.vfs;
+
+import org.apache.commons.vfs2.FileObject;
+import org.apache.commons.vfs2.FileSystemManager;
+import org.apache.commons.vfs2.FileSystemOptions;
+import org.apache.commons.vfs2.provider.sftp.SftpFileSystemConfigBuilder;
+
+/**
+ * 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;
+	}
+
+	public void init() {
+		if (name == null)
+			throw new MaintenanceException("Atomic backup name must be set");
+	}
+
+	public void destroy() {
+
+	}
+
+	@Override
+	public String backup(FileSystemManager fileSystemManager,
+			String backupsBase, BackupContext backupContext,
+			FileSystemOptions opts) {
+		if (name == null)
+			throw new MaintenanceException("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 MaintenanceException("Unsupported compression "
+						+ compression);
+
+			writeBackup(targetFo);
+
+			return targetFo.toString();
+		} catch (Exception e) {
+			throw new MaintenanceException("Cannot backup " + name + " to "
+					+ targetFo, e);
+		} finally {
+			BackupUtils.closeFOQuietly(targetFo);
+		}
+	}
+
+	public void setName(String name) {
+		this.name = name;
+	}
+
+	public String getName() {
+		return name;
+	}
+
+	public void setCompression(String compression) {
+		this.compression = compression;
+	}
+}