@Override
protected void readColumns(
DatabaseSnapshot snapshot, String schema, DatabaseMetaData databaseMetaData)
throws SQLException, DatabaseException {
findIntegerColumns(snapshot, schema);
super.readColumns(snapshot, schema, databaseMetaData);
/*
* Code Description:
* Finding all 'tablespace' attributes of column's PKs
* */
Database database = snapshot.getDatabase();
Statement statement = null;
ResultSet rs = null;
try {
statement =
((JdbcConnection) database.getConnection()).getUnderlyingConnection().createStatement();
// Setting default schema name. Needed for correct statement generation
if (schema == null) schema = database.convertRequestedSchemaToSchema(schema);
String query =
"select ui.tablespace_name TABLESPACE, ucc.table_name TABLE_NAME, ucc.column_name COLUMN_NAME FROM all_indexes ui , all_constraints uc , all_cons_columns ucc where uc.constraint_type = 'P' and ucc.constraint_name = uc.constraint_name and uc.index_name = ui.index_name and uc.owner = '"
+ schema
+ "' and ui.table_owner = '"
+ schema
+ "' and ucc.owner = '"
+ schema
+ "'";
rs = statement.executeQuery(query);
while (rs.next()) {
Column column = snapshot.getColumn(rs.getString("TABLE_NAME"), rs.getString("COLUMN_NAME"));
// setting up tablespace property to column, to configure it's PK-index
if (column == null) {
continue; // probably a different schema
}
column.setTablespace(rs.getString("TABLESPACE"));
}
} finally {
if (rs != null) {
try {
rs.close();
} catch (SQLException ignore) {
}
}
if (statement != null) {
try {
statement.close();
} catch (SQLException ignore) {
}
}
}
}
/**
* Method finds all INTEGER columns in snapshot's database
*
* @param snapshot current database snapshot
* @return String list with names of all INTEGER columns
* @throws java.sql.SQLException execute statement error
*/
private List<String> findIntegerColumns(DatabaseSnapshot snapshot, String schema)
throws SQLException, DatabaseException {
Database database = snapshot.getDatabase();
// Setting default schema name. Needed for correct statement generation
if (schema == null) {
schema = database.convertRequestedSchemaToSchema(schema);
}
Statement statement =
((JdbcConnection) database.getConnection()).getUnderlyingConnection().createStatement();
ResultSet integerListRS = null;
// Finding all columns created as 'INTEGER'
try {
integerListRS =
statement.executeQuery(
"select TABLE_NAME, COLUMN_NAME from all_tab_columns where data_precision is null and data_scale = 0 and data_type = 'NUMBER' and owner = '"
+ schema
+ "'");
while (integerListRS.next()) {
integerList.add(
integerListRS.getString("TABLE_NAME") + "." + integerListRS.getString("COLUMN_NAME"));
}
} finally {
if (integerListRS != null) {
try {
integerListRS.close();
} catch (SQLException ignore) {
}
}
if (statement != null) {
try {
statement.close();
} catch (SQLException ignore) {
}
}
}
return integerList;
}