/**
* Takes a list of SqlVisitors and returns a new list with only the SqlVisitors set to apply to
* rollbacks
*/
protected List<SqlVisitor> filterRollbackVisitors(final List<SqlVisitor> visitors) {
final List<SqlVisitor> rollbackVisitors = new ArrayList<SqlVisitor>();
if (visitors != null) {
for (SqlVisitor visitor : visitors) {
if (visitor.isApplyToRollback()) {
rollbackVisitors.add(visitor);
}
}
}
return rollbackVisitors;
}
public static List<SqlStatement> getAlterTableStatements(
AlterTableVisitor alterTableVisitor,
Database database,
String catalogName,
String schemaName,
String tableName)
throws DatabaseException {
DatabaseSnapshot snapshot = null; // todo
List<SqlStatement> statements = new ArrayList<SqlStatement>();
Table table = null;
try {
table =
SnapshotGeneratorFactory.getInstance()
.createSnapshot(
(Table)
new Table().setName(tableName).setSchema(new Schema(new Catalog(null), null)),
database);
} catch (InvalidExampleException e) {
throw new UnexpectedLiquibaseException(e);
}
List<ColumnConfig> createColumns = new Vector<ColumnConfig>();
List<ColumnConfig> copyColumns = new Vector<ColumnConfig>();
if (table != null) {
for (Column column : table.getColumns()) {
ColumnConfig new_column = new ColumnConfig(column);
if (alterTableVisitor.createThisColumn(new_column)) {
createColumns.add(new_column);
}
ColumnConfig copy_column = new ColumnConfig(column);
if (alterTableVisitor.copyThisColumn(copy_column)) {
copyColumns.add(copy_column);
}
}
}
for (ColumnConfig column : alterTableVisitor.getColumnsToAdd()) {
if (alterTableVisitor.createThisColumn(column)) {
createColumns.add(column);
}
if (alterTableVisitor.copyThisColumn(column)) {
copyColumns.add(column);
}
}
List<Index> newIndices = new Vector<Index>();
for (Index index :
new ArrayList<
Index>()) { // todo SnapshotGeneratorFactory.getInstance().getGenerator(Index.class,
// database).get(new Schema(new Catalog(null), schemaName), database)) {
if (index.getTable().getName().equalsIgnoreCase(tableName)) {
if (alterTableVisitor.createThisIndex(index)) {
newIndices.add(index);
}
}
}
// rename table
String temp_table_name = tableName + "_temporary";
statements.add(new RenameTableStatement(catalogName, schemaName, tableName, temp_table_name));
// create temporary table
CreateTableChange ct_change_tmp = new CreateTableChange();
ct_change_tmp.setSchemaName(schemaName);
ct_change_tmp.setTableName(tableName);
for (ColumnConfig column : createColumns) {
ct_change_tmp.addColumn(column);
}
statements.addAll(Arrays.asList(ct_change_tmp.generateStatements(database)));
// copy rows to temporary table
statements.add(new CopyRowsStatement(temp_table_name, tableName, copyColumns));
// delete original table
statements.add(new DropTableStatement(catalogName, schemaName, temp_table_name, false));
// validate indices
statements.add(new ReindexStatement(catalogName, schemaName, tableName));
// add remaining indices
for (Index index_config : newIndices) {
statements.add(
new CreateIndexStatement(
index_config.getName(),
catalogName,
schemaName,
tableName,
index_config.isUnique(),
index_config.getAssociatedWithAsString(),
index_config.getColumns().toArray(new String[index_config.getColumns().size()])));
}
return statements;
}
@Override
public boolean dataTypeIsNotModifiable(final String typeName) {
return unmodifiableDataTypes.contains(typeName.toLowerCase());
}
