+package org.argeo.cms.internal.backup;
+
+import java.security.PrivilegedExceptionAction;
+import java.util.Map;
+
+import javax.jcr.Credentials;
+import javax.jcr.Repository;
+import javax.jcr.RepositoryException;
+import javax.jcr.Session;
+
+import org.argeo.cms.maintenance.DataMigration;
+import org.argeo.jcr.JcrUtils;
+
+/** Migrate data between two workspaces, at JCR level. */
+public class RepositoryMigration implements PrivilegedExceptionAction<Boolean> {
+ private final Repository sourceRepository;
+ private final Repository targetRepository;
+ private final DataMigration dataMigration;
+
+ private Credentials sourceCredentials = null;
+ private Credentials targetCredentials = null;
+
+ public RepositoryMigration(Repository sourceRepository,
+ Repository targetRepository, DataMigration dataMigration) {
+ this.sourceRepository = sourceRepository;
+ this.targetRepository = targetRepository;
+ this.dataMigration = dataMigration;
+ }
+
+ @Override
+ public Boolean run() throws Exception {
+ Map<String, String> wk = dataMigration.workspacesToMigrate();
+ if (wk == null)
+ return migrate(sourceRepository, null, targetRepository, null);
+ else {
+ for (String sourceWorkspace : wk.keySet()) {
+ String targetWorkspace = wk.get(sourceWorkspace);
+ boolean ok = migrate(sourceRepository, sourceWorkspace,
+ targetRepository, targetWorkspace);
+ if (!ok)
+ return false;
+ }
+ return true;
+ }
+ }
+
+ protected final boolean migrate(Repository sourceRepository,
+ String sourceWorkspace, Repository targetRepository,
+ String targetWorkspace) throws RepositoryException {
+ Session source = null, target = null;
+ try {
+ source = sourceRepository.login(sourceCredentials, sourceWorkspace);
+ target = targetRepository.login(targetCredentials, targetWorkspace);
+ return dataMigration.migrate(source, target);
+ } finally {
+ JcrUtils.logoutQuietly(source);
+ JcrUtils.logoutQuietly(target);
+ }
+ }
+
+ public void setSourceCredentials(Credentials sourceCredentials) {
+ this.sourceCredentials = sourceCredentials;
+ }
+
+ public void setTargetCredentials(Credentials targetCredentials) {
+ this.targetCredentials = targetCredentials;
+ }
+
+}