/** * 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()); }