/** * Generates the {@link Index} example (taken from {@link IndexExistsPrecondition}). * * @param database the database instance. * @param schema the schema instance. * @param tableName the table name of the index. * @return the index example. */ protected Index getIndexExample( final Database database, final Schema schema, final String tableName) { final Index example = new Index(); if (tableName != null) { example.setTable( (Table) new Table() .setName(database.correctObjectName(getTableName(), Table.class)) .setSchema(schema)); } example.setName(database.correctObjectName(getIndexName(), Index.class)); if (StringUtils.trimToNull(getColumnNames()) != null) { for (final String columnName : getColumnNames().split("\\s*,\\s*")) { final Column column = new Column(database.correctObjectName(columnName, Column.class)); example.getColumns().add(column); } } return example; }
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; }