public String[] getCleanSql() {
if (cleanSql == null) {
// loop over all foreign key constraints
List dropForeignKeysSql = new ArrayList();
List createForeignKeysSql = new ArrayList();
Iterator iter = configuration.getTableMappings();
while (iter.hasNext()) {
Table table = (Table) iter.next();
if (table.isPhysicalTable()) {
Iterator subIter = table.getForeignKeyIterator();
while (subIter.hasNext()) {
ForeignKey fk = (ForeignKey) subIter.next();
if (fk.isPhysicalConstraint()) {
// collect the drop key constraint
dropForeignKeysSql.add(
fk.sqlDropString(
dialect,
properties.getProperty(Environment.DEFAULT_CATALOG),
properties.getProperty(Environment.DEFAULT_SCHEMA)));
createForeignKeysSql.add(
fk.sqlCreateString(
dialect,
mapping,
properties.getProperty(Environment.DEFAULT_CATALOG),
properties.getProperty(Environment.DEFAULT_SCHEMA)));
}
}
}
}
List deleteSql = new ArrayList();
iter = configuration.getTableMappings();
while (iter.hasNext()) {
Table table = (Table) iter.next();
deleteSql.add("delete from " + table.getName());
}
List cleanSqlList = new ArrayList();
cleanSqlList.addAll(dropForeignKeysSql);
cleanSqlList.addAll(deleteSql);
cleanSqlList.addAll(createForeignKeysSql);
cleanSql = (String[]) cleanSqlList.toArray(new String[cleanSqlList.size()]);
}
return cleanSql;
}
/**
* Returns a table's foreign keys and their columns as a Map from the key name to the ForeignKey
* object.
*
* <p>A foreign key may not have a name. On such a database, 2 foreign keys must reference 2
* different tables. Otherwise there's no way to tell them apart and the foreign key information
* reported by DatabaseMetaData becomes ill-formed.
*/
public static Map<String, ForeignKey> getForeignKeys(DatabaseMetaData metadata, String tableName)
throws Exception {
ResultSet keys =
metadata.getImportedKeys(
metadata.getConnection().getCatalog(), metadata.getUserName(), tableName);
Map<String, ForeignKey> map = new HashMap<String, ForeignKey>();
while (keys.next()) {
String table = keys.getString(IMPORTED_PK_TAB_NAME);
String name = keys.getString(IMPORTED_FK_KEY_NAME);
if (name == null || name.length() == 0) name = "UNNAMED_FK_" + table;
ForeignKey key = map.get(name);
if (key == null) {
map.put(name, key = new ForeignKey(table));
}
key.add(keys.getString(IMPORTED_FK_COL_NAME));
}
keys.close();
return map;
}
@Override
protected void readIndexes(
DatabaseSnapshot snapshot, String schema, DatabaseMetaData databaseMetaData)
throws DatabaseException, SQLException {
Database database = snapshot.getDatabase();
updateListeners("Reading indexes for " + database.toString() + " ...");
String query =
"select aic.index_name, 3 AS TYPE, aic.table_name, aic.column_name, aic.column_position AS ORDINAL_POSITION, null AS FILTER_CONDITION, ai.tablespace_name AS TABLESPACE, ai.uniqueness FROM all_ind_columns aic, all_indexes ai WHERE aic.table_owner='"
+ database.convertRequestedSchemaToSchema(schema)
+ "' and aic.index_name = ai.index_name ORDER BY INDEX_NAME, ORDINAL_POSITION";
Statement statement = null;
ResultSet rs = null;
Map<String, Index> indexMap = null;
try {
statement =
((JdbcConnection) database.getConnection()).getUnderlyingConnection().createStatement();
rs = statement.executeQuery(query);
indexMap = new HashMap<String, Index>();
while (rs.next()) {
String indexName = convertFromDatabaseName(rs.getString("INDEX_NAME"));
String tableName = rs.getString("TABLE_NAME");
String tableSpace = rs.getString("TABLESPACE");
String columnName = convertFromDatabaseName(rs.getString("COLUMN_NAME"));
if (columnName == null) {
// nothing to index, not sure why these come through sometimes
continue;
}
short type = rs.getShort("TYPE");
boolean nonUnique;
String uniqueness = rs.getString("UNIQUENESS");
if ("UNIQUE".equals(uniqueness)) {
nonUnique = false;
} else {
nonUnique = true;
}
short position = rs.getShort("ORDINAL_POSITION");
String filterCondition = rs.getString("FILTER_CONDITION");
if (type == DatabaseMetaData.tableIndexStatistic) {
continue;
}
Index index;
if (indexMap.containsKey(indexName)) {
index = indexMap.get(indexName);
} else {
index = new Index();
Table table = snapshot.getTable(tableName);
if (table == null) {
continue; // probably different schema
}
index.setTable(table);
index.setTablespace(tableSpace);
index.setName(indexName);
index.setUnique(!nonUnique);
index.setFilterCondition(filterCondition);
indexMap.put(indexName, index);
}
for (int i = index.getColumns().size(); i < position; i++) {
index.getColumns().add(null);
}
index.getColumns().set(position - 1, columnName);
}
} finally {
JdbcUtils.closeResultSet(rs);
JdbcUtils.closeStatement(statement);
}
for (Map.Entry<String, Index> entry : indexMap.entrySet()) {
snapshot.getIndexes().add(entry.getValue());
}
/*
* marks indexes as "associated with" instead of "remove it"
* Index should have associations with:
* foreignKey, primaryKey or uniqueConstraint
* */
for (Index index : snapshot.getIndexes()) {
for (PrimaryKey pk : snapshot.getPrimaryKeys()) {
if (index.getTable().getName().equalsIgnoreCase(pk.getTable().getName())
&& index.getColumnNames().equals(pk.getColumnNames())) {
index.addAssociatedWith(Index.MARK_PRIMARY_KEY);
}
}
for (ForeignKey fk : snapshot.getForeignKeys()) {
if (index.getTable().getName().equalsIgnoreCase(fk.getForeignKeyTable().getName())
&& index.getColumnNames().equals(fk.getForeignKeyColumns())) {
index.addAssociatedWith(Index.MARK_FOREIGN_KEY);
}
}
for (UniqueConstraint uc : snapshot.getUniqueConstraints()) {
if (index.getTable().getName().equalsIgnoreCase(uc.getTable().getName())
&& index.getColumnNames().equals(uc.getColumnNames())) {
index.addAssociatedWith(Index.MARK_UNIQUE_CONSTRAINT);
}
}
}
}
