jdbcConn = connFactory.getJdbcConnection();
jdbcConn.setAutoCommit(false);
}
catch (SQLException e)
{
throw new BackupException(e);
}
catch (RepositoryException e)
{
throw new BackupException(e);
}
}
else
{
jdbcConn = (Connection)context.getObject(DataRestoreContext.DB_CONNECTION);
}
backupInfo =
new ObjectReaderImpl(PrivilegedFileHelper.fileInputStream(new File(storageDir,
"JDBCWorkspaceDataContainer.info")));
String srcContainerName = backupInfo.readString();
boolean srcMultiDb = backupInfo.readBoolean();
Map<String, RestoreTableRule> tables = new LinkedHashMap<String, RestoreTableRule>();
// ITEM table
String dstTableName = "JCR_" + (multiDb ? "M" : "S") + "ITEM";
String srcTableName = "JCR_" + (srcMultiDb ? "M" : "S") + "ITEM";
RestoreTableRule restoreTableRule = new RestoreTableRule();
restoreTableRule.setSrcContainerName(srcContainerName);
restoreTableRule.setSrcMultiDb(srcMultiDb);
restoreTableRule.setDstContainerName(containerName);
restoreTableRule.setDstMultiDb(multiDb);
restoreTableRule.setSrcTableName(srcTableName);
if (multiDb)
{
if (!srcMultiDb)
{
// CONTAINER_NAME column index
restoreTableRule.setDeleteColumnIndex(4);
// ID and PARENT_ID column indexes
Set<Integer> convertColumnIndex = new HashSet<Integer>();
convertColumnIndex.add(0);
convertColumnIndex.add(1);
restoreTableRule.setConvertColumnIndex(convertColumnIndex);
}
}
else
{
if (srcMultiDb)
{
// CONTAINER_NAME column index
restoreTableRule.setNewColumnIndex(4);
restoreTableRule.setNewColumnName("CONTAINER_NAME");
restoreTableRule.setNewColumnType(Types.VARCHAR);
// ID and PARENT_ID column indexes
Set<Integer> convertColumnIndex = new HashSet<Integer>();
convertColumnIndex.add(0);
convertColumnIndex.add(1);
restoreTableRule.setConvertColumnIndex(convertColumnIndex);
}
else
{
// ID and PARENT_ID and CONTAINER_NAME column indexes
Set<Integer> convertColumnIndex = new HashSet<Integer>();
convertColumnIndex.add(0);
convertColumnIndex.add(1);
convertColumnIndex.add(4);
restoreTableRule.setConvertColumnIndex(convertColumnIndex);
}
}
tables.put(dstTableName, restoreTableRule);
// VALUE table
dstTableName = "JCR_" + (multiDb ? "M" : "S") + "VALUE";
srcTableName = "JCR_" + (srcMultiDb ? "M" : "S") + "VALUE";
restoreTableRule = new RestoreTableRule();
restoreTableRule.setSrcContainerName(srcContainerName);
restoreTableRule.setSrcMultiDb(srcMultiDb);
restoreTableRule.setDstContainerName(containerName);
restoreTableRule.setDstMultiDb(multiDb);
restoreTableRule.setSrcTableName(srcTableName);
// auto increment ID column
restoreTableRule.setSkipColumnIndex(0);
if (!multiDb || !srcMultiDb)
{
// PROPERTY_ID column index
Set<Integer> convertColumnIndex = new HashSet<Integer>();
convertColumnIndex.add(3);
restoreTableRule.setConvertColumnIndex(convertColumnIndex);
}
tables.put(dstTableName, restoreTableRule);
// REF tables
dstTableName = "JCR_" + (multiDb ? "M" : "S") + "REF";
srcTableName = "JCR_" + (srcMultiDb ? "M" : "S") + "REF";
restoreTableRule = new RestoreTableRule();
restoreTableRule.setSrcContainerName(srcContainerName);
restoreTableRule.setSrcMultiDb(srcMultiDb);
restoreTableRule.setDstContainerName(containerName);
restoreTableRule.setDstMultiDb(multiDb);
restoreTableRule.setSrcTableName(srcTableName);
if (!multiDb || !srcMultiDb)
{
// NODE_ID and PROPERTY_ID column indexes
Set<Integer> convertColumnIndex = new HashSet<Integer>();
convertColumnIndex.add(0);
convertColumnIndex.add(1);
restoreTableRule.setConvertColumnIndex(convertColumnIndex);
}
tables.put(dstTableName, restoreTableRule);
DBCleaner dbCleaner = null;
if (context.getObject(DataRestoreContext.DB_CLEANER) != null)
{
dbCleaner = (DBCleaner)context.getObject(DataRestoreContext.DB_CLEANER);
}
else
{
dbCleaner = DBCleanService.getWorkspaceDBCleaner(jdbcConn, wsConfig);
}
restorers.add(new DBRestore(storageDir, jdbcConn, tables, wsConfig, swapCleaner, dbCleaner));
// prepare value storage restorer
if (wsConfig.getContainer().getValueStorages() != null)
{
List<File> dataDirs = new ArrayList<File>();
List<File> backupDirs = new ArrayList<File>();
List<ValueStorageEntry> valueStorages = wsConfig.getContainer().getValueStorages();
for (ValueStorageEntry valueStorage : valueStorages)
{
File dataDir = new File(valueStorage.getParameterValue(FileValueStorage.PATH));
dataDirs.add(dataDir);
File zipFile = new File(storageDir, "values-" + valueStorage.getId() + ".zip");
if (PrivilegedFileHelper.exists(zipFile))
{
backupDirs.add(zipFile);
}
else
{
// try to check if we have deal with old backup format
zipFile = new File(storageDir, "values/" + valueStorage.getId());
if (PrivilegedFileHelper.exists(zipFile))
{
backupDirs.add(zipFile);
}
else
{
throw new RepositoryConfigurationException("There is no backup data for value storage with id "
+ valueStorage.getId());
}
}
}
restorers.add(new DirectoryRestore(dataDirs, backupDirs));
}
return new ComplexDataRestore(restorers);
}
catch (FileNotFoundException e)
{
throw new BackupException(e);
}
catch (IOException e)
{
throw new BackupException(e);
}
catch (NamingException e)
{
throw new BackupException(e);
}
catch (SQLException e)
{
throw new BackupException(e);
}
catch (RepositoryConfigurationException e)
{
throw new BackupException(e);
}
finally
{
if (backupInfo != null)
{