public void visit(Configuration cfg, Table table, IssueCollector pc) {
if (table.isPhysicalTable()) {
setSchemaSelection(table);
List list = reader.readDatabaseSchema(dbc, null, null);
if (list.isEmpty()) {
pc.reportIssue(
new Issue(
"SCHEMA_TABLE_MISSING",
Issue.HIGH_PRIORITY,
"Missing table "
+ Table.qualify(table.getCatalog(), table.getSchema(), table.getName())));
return;
} else if (list.size() > 1) {
pc.reportIssue(
new Issue(
"SCHEMA_TABLE_MISSING",
Issue.NORMAL_PRIORITY,
"Found "
+ list.size()
+ " tables for "
+ Table.qualify(table.getCatalog(), table.getSchema(), table.getName())));
return;
} else {
currentDbTable = (Table) list.get(0);
visitColumns(cfg, table, pc);
}
} else {
// log?
}
}
/**
* There are 2 solutions: 1. DatabaseCollector#addTable()/getTable() should be called for not
* quoted parameters - I think it is preferable way. 2. DatabaseCollector#addTable()/getTable()
* should be called for quoted parameters - here users should use the same quotes as JDBCReader.
* Because of this there are 2 opposite methods(and they are both failed as addTable uses quoted
* names but getTable uses non-quoted names )
*/
public void testQuotedNamesAndDefaultDatabaseCollector() {
StandardServiceRegistryBuilder builder = new StandardServiceRegistryBuilder();
ServiceRegistry serviceRegistry = builder.build();
MetaDataDialect realMetaData =
MetaDataDialectFactory.createMetaDataDialect(
serviceRegistry.getService(JdbcServices.class).getDialect(), cfg.getProperties());
JDBCReader reader =
JDBCReaderFactory.newJDBCReader(
cfg.getProperties(),
new DefaultReverseEngineeringStrategy(),
realMetaData,
serviceRegistry);
DatabaseCollector dc = new DefaultDatabaseCollector(reader.getMetaDataDialect());
reader.readDatabaseSchema(dc, null, SCHEMA);
assertNotNull("The table should be found", dc.getTable(SCHEMA, "PUBLIC", TABLE1));
assertNotNull("The table should be found", dc.getTable(SCHEMA, "PUBLIC", TABLE2));
assertNull(
"Quoted names should not return the table", dc.getTable(quote(SCHEMA), "PUBLIC", QTABLE1));
assertNull(
"Quoted names should not return the table", dc.getTable(quote(SCHEMA), "PUBLIC", QTABLE2));
assertEquals("Foreign key 'masterref' was filtered!", 1, dc.getOneToManyCandidates().size());
}
/**
* Read JDBC Metadata from the database. Does not create any classes or other ORM releated
* structures.
*
* @param catalog
* @param schema
* @return
* @throws SQLException
*/
public DatabaseCollector readDatabaseSchema(String catalog, String schema) throws SQLException {
// use default from settings if nothing else specified.
catalog = catalog != null ? catalog : settings.getDefaultCatalogName();
schema = schema != null ? schema : settings.getDefaultSchemaName();
JDBCReader reader =
JDBCReaderFactory.newJDBCReader(cfg.getProperties(), settings, revengStrategy);
DatabaseCollector dbs = new MappingsDatabaseCollector(mappings, reader.getMetaDataDialect());
reader.readDatabaseSchema(dbs, catalog, schema);
return dbs;
}
private boolean isTable(Object key) throws HibernateException {
// BIG HACK - should probably utilize the table cache before going to the jdbcreader :(
if (key instanceof String) {
String[] strings = StringHelper.split(".", (String) key);
if (strings.length == 1) {
tableSelector.clearSchemaSelections();
tableSelector.addSchemaSelection(new SchemaSelection(null, null, strings[0]));
List list = reader.readDatabaseSchema(dbc, null, null);
return !list.isEmpty();
} else if (strings.length == 3) {
tableSelector.clearSchemaSelections();
tableSelector.addSchemaSelection(new SchemaSelection(strings[0], strings[1], strings[2]));
List list = reader.readDatabaseSchema(dbc, null, null);
return !list.isEmpty();
} else if (strings.length == 2) {
tableSelector.clearSchemaSelections();
tableSelector.addSchemaSelection(new SchemaSelection(null, strings[0], strings[1]));
List list = reader.readDatabaseSchema(dbc, null, null);
return !list.isEmpty();
}
}
return false;
}