@ -75,21 +75,38 @@ public class MetaDataLoaderImpl implements MetaDataLoader {
} }
@Override @Override
public List < Table > getTables ( DataSource dataSource , String catalogName , String schemaName ) throws MetaDataAccessException { public List < Table > getTables ( DataSource dataSource , String catalogName , String schemaName , String . . . tableNames ) throws MetaDataAccessException {
List < Table > result = new ArrayList < > ( ) ; List < Table > result = new ArrayList < > ( ) ;
Connection connection = null ; Connection connection = null ;
ResultSet rs = null ; ResultSet rs = null ;
try { try {
connection = dataSource . getConnection ( ) ; connection = dataSource . getConnection ( ) ;
DatabaseMetaData databaseMetaData = connection . getMetaData ( ) ; DatabaseMetaData databaseMetaData = connection . getMetaData ( ) ;
rs = databaseMetaData . getTables ( catalogName , schemaName , null , new String [ ] { "TABLE" } ) ; if ( tableNames = = null | | tableNames . length = = 0 ) {
while ( rs . next ( ) ) { rs = databaseMetaData . getTables ( catalogName , schemaName , null , new String [ ] { "TABLE" } ) ;
Table table = new Table ( ) ; while ( rs . next ( ) ) {
Table table = new Table ( ) ;
table . setName ( rs . getString ( "TABLE_NAME" ) ) ;
table . setRemarks ( rs . getString ( "REMARKS" ) ) ;
table . setColumns ( getColumns ( dataSource , catalogName , schemaName , table . getName ( ) ) ) ;
table . setForeignKeys ( getForeignKeys ( dataSource , catalogName , schemaName , table . getName ( ) ) ) ;
buildSelfReference ( table ) ;
result . add ( table ) ;
}
} else {
for ( String tableName : tableNames ) {
rs = databaseMetaData . getTables ( catalogName , schemaName , tableName , new String [ ] { "TABLE" } ) ;
while ( rs . next ( ) ) {
Table table = new Table ( ) ;
table . setName ( rs . getString ( "TABLE_NAME" ) ) ; table . setName ( rs . getString ( "TABLE_NAME" ) ) ;
table . setRemarks ( rs . getString ( "REMARKS" ) ) ; table . setRemarks ( rs . getString ( "REMARKS" ) ) ;
table . setColumns ( getColumns ( dataSource , catalogName , schemaName , table . getName ( ) ) ) ; table . setColumns ( getColumns ( dataSource , catalogName , schemaName , table . getName ( ) ) ) ;
table . setForeignKeys ( getForeignKeys ( dataSource , catalogName , schemaName , table . getName ( ) ) ) ;
buildSelfReference ( table ) ;
result . add ( table ) ; result . add ( table ) ;
} }
}
}
} catch ( SQLException e ) { } catch ( SQLException e ) {
throw new MetaDataAccessException ( e ) ; throw new MetaDataAccessException ( e ) ;
} finally { } finally {
@ -120,40 +137,6 @@ public class MetaDataLoaderImpl implements MetaDataLoader {
return result ; return result ;
} }
private Column getColumn ( ResultSet rs ) throws SQLException {
String TABLE_CAT = rs . getString ( "TABLE_CAT" ) ; //table catalog (may be null)
String TABLE_SCHEM = rs . getString ( "TABLE_SCHEM" ) ; //table schema (may be null)
String TABLE_NAME = rs . getString ( "TABLE_NAME" ) ; //table name (表名称)
String COLUMN_NAME = rs . getString ( "COLUMN_NAME" ) ; //column name(列名)
int DATA_TYPE = rs . getInt ( "DATA_TYPE" ) ; //SQL type from java.sql.Types(列的数据类型)
String TYPE_NAME = rs . getString ( "TYPE_NAME" ) ; //Data source dependent type name, for a UDT the type name is fully qualified
int COLUMN_SIZE = rs . getInt ( "COLUMN_SIZE" ) ; //column size
int BUFFER_LENGTH = rs . getInt ( "BUFFER_LENGTH" ) ; //is not used
int DECIMAL_DIGITS = rs . getInt ( "DECIMAL_DIGITS" ) ; //the number of fractional digits. Null is returned for data types where DECIMAL_DIGITS is not applicable.
int NUM_PREC_RADIX = rs . getInt ( "NUM_PREC_RADIX" ) ; //Radix (typically either 10 or 2)
int NULLABLE = rs . getInt ( "NULLABLE" ) ; //is NULL allowed.
String REMARKS = rs . getString ( "REMARKS" ) ; //comment describing column (may be null)
String COLUMN_DEF = rs . getString ( "COLUMN_DEF" ) ; //default value for the column, (may be null)
int SQL_DATA_TYPE = rs . getInt ( "SQL_DATA_TYPE" ) ; //unused
int SQL_DATETIME_SUB = rs . getInt ( "SQL_DATETIME_SUB" ) ; //unused
int CHAR_OCTET_LENGTH = rs . getInt ( "CHAR_OCTET_LENGTH" ) ; //for char types the maximum number of bytes in the column
int ORDINAL_POSITION = rs . getInt ( "ORDINAL_POSITION" ) ; //index of column in table (starting at 1)
String IS_NULLABLE = rs . getString ( "IS_NULLABLE" ) ; //ISO rules are used to determine the nullability for a column.
//String SCOPE_CATLOG =rs.getString("SCOPE_CATLOG"); //catalog of table that is the scope of a reference attribute (null if DATA_TYPE isn't REF)
//String SCOPE_SCHEMA =rs.getString("SCOPE_SCHEMA"); //schema of table that is the scope of a reference attribute (null if the DATA_TYPE isn't REF)
//String SCOPE_TABLE =rs.getString("SCOPE_TABLE"); //table name that this the scope of a reference attribure (null if the DATA_TYPE isn't REF)
String SOURCE_DATA_TYPE = rs . getString ( "SOURCE_DATA_TYPE" ) ; //source type of a distinct type or user-generated Ref type, SQL type from java.sql.Types
String IS_AUTOINCREMENT = rs . getString ( "IS_AUTOINCREMENT" ) ; //Indicates whether this column is auto incremented
Column column = new Column ( ) ;
column . setName ( COLUMN_NAME ) ;
column . setRemarks ( REMARKS ) ;
column . setSqlType ( SqlTypeUtil . getSqlTypeName ( DATA_TYPE ) ) ;
column . setJavaType ( SqlTypeUtil . getJavaType ( DATA_TYPE ) ) ;
return column ;
}
public List < TableSummary > getTableSummary ( DataSource dataSource , String catalogName , String schemaName , boolean isCount ) throws MetaDataAccessException { public List < TableSummary > getTableSummary ( DataSource dataSource , String catalogName , String schemaName , boolean isCount ) throws MetaDataAccessException {
List < Table > tables = this . getTables ( dataSource , catalogName , schemaName ) ; List < Table > tables = this . getTables ( dataSource , catalogName , schemaName ) ;
if ( tables = = null | | tables . isEmpty ( ) ) { if ( tables = = null | | tables . isEmpty ( ) ) {
@ -181,92 +164,77 @@ public class MetaDataLoaderImpl implements MetaDataLoader {
return result ; return result ;
} }
public boolean isSelfReference ( schemacrawler . schema . Table table ) { private void buildSelfReference ( Table table ) {
Collection < schemacrawler . schema . Table > parentTables = table . getRelatedTables ( TableRelationshipType . parent ) ; if ( table = = null ) { return ; }
if ( parentTables ! = null & & ! parentTables . isEmpty ( ) ) { List < ForeignKey > foreignKeys = table . getForeignKeys ( ) ;
for ( schemacrawler . schema . Table parentTable : parentTables ) { if ( foreignKeys = = null | | foreignKeys . isEmpty ( ) ) { return ; }
if ( parentTable . equals ( table ) ) { for ( ForeignKey foreignKey : foreignKeys ) {
return true ; if ( table . getName ( ) . equals ( foreignKey . getPrimaryKeyTableName ( ) ) ) {
} table . setSelfReference ( true ) ;
table . setSelfReferencePrimaryKeyColumnName ( foreignKey . getPrimaryKeyColumnName ( ) ) ;
table . setSelfReferenceForeignKeyColumnName ( foreignKey . getForeignKeyColumnName ( ) ) ;
return ;
} }
} }
return false ;
} }
private Table from ( DatabaseMetaData databaseMetaData , String tableName , String tableRemarks ) { private Column getColumn ( ResultSet rs ) throws SQLException {
// Table result =new Table();
String TABLE_CAT = rs . getString ( "TABLE_CAT" ) ; //table catalog (may be null)
// result.setName(tableName);
String TABLE_SCHEM = rs . getString ( "TABLE_SCHEM" ) ; //table schema (may be null)
// result.setRemarks(tableRemarks);
String TABLE_NAME = rs . getString ( "TABLE_NAME" ) ; //table name (表名称)
// // 处理列
String COLUMN_NAME = rs . getString ( "COLUMN_NAME" ) ; //column name(列名)
// for(schemacrawler.schema.Column column : table.getColumns()) {
int DATA_TYPE = rs . getInt ( "DATA_TYPE" ) ; //SQL type from java.sql.Types(列的数据类型)
// result.getColumns().add(from(column));
String TYPE_NAME = rs . getString ( "TYPE_NAME" ) ; //Data source dependent type name, for a UDT the type name is fully qualified
// }
int COLUMN_SIZE = rs . getInt ( "COLUMN_SIZE" ) ; //column size
// // 处理外键
int BUFFER_LENGTH = rs . getInt ( "BUFFER_LENGTH" ) ; //is not used
// Collection<schemacrawler.schema.ForeignKey> foreignKeys =table.getForeignKeys();
int DECIMAL_DIGITS = rs . getInt ( "DECIMAL_DIGITS" ) ; //the number of fractional digits. Null is returned for data types where DECIMAL_DIGITS is not applicable.
// for(schemacrawler.schema.ForeignKey foreignKey : foreignKeys){
int NUM_PREC_RADIX = rs . getInt ( "NUM_PREC_RADIX" ) ; //Radix (typically either 10 or 2)
// result.getForeignKeys().add(from(foreignKey));
int NULLABLE = rs . getInt ( "NULLABLE" ) ; //is NULL allowed.
// }
String REMARKS = rs . getString ( "REMARKS" ) ; //comment describing column (may be null)
// // 处理自引用
String COLUMN_DEF = rs . getString ( "COLUMN_DEF" ) ; //default value for the column, (may be null)
// for(ForeignKey foreignKey : result.getForeignKeys()){
int SQL_DATA_TYPE = rs . getInt ( "SQL_DATA_TYPE" ) ; //unused
// if(foreignKey.getForeignKeyTableName().equalsIgnoreCase(foreignKey.getPrimaryKeyTableName())){
int SQL_DATETIME_SUB = rs . getInt ( "SQL_DATETIME_SUB" ) ; //unused
// result.setSelfReferenceForeignKeyColumnName(foreignKey.getForeignKeyColumnName());
int CHAR_OCTET_LENGTH = rs . getInt ( "CHAR_OCTET_LENGTH" ) ; //for char types the maximum number of bytes in the column
// result.setSelfReferencePrimaryKeyColumnName(foreignKey.getPrimaryKeyColumnName());
int ORDINAL_POSITION = rs . getInt ( "ORDINAL_POSITION" ) ; //index of column in table (starting at 1)
// }
String IS_NULLABLE = rs . getString ( "IS_NULLABLE" ) ; //ISO rules are used to determine the nullability for a column.
// }
//String SCOPE_CATLOG =rs.getString("SCOPE_CATLOG"); //catalog of table that is the scope of a reference attribute (null if DATA_TYPE isn't REF)
// // 处理索引
//String SCOPE_SCHEMA =rs.getString("SCOPE_SCHEMA"); //schema of table that is the scope of a reference attribute (null if the DATA_TYPE isn't REF)
// Collection<schemacrawler.schema.Index> indexes =table.getIndexes();
//String SCOPE_TABLE =rs.getString("SCOPE_TABLE"); //table name that this the scope of a reference attribure (null if the DATA_TYPE isn't REF)
// for(schemacrawler.schema.Index index : indexes){
String SOURCE_DATA_TYPE = rs . getString ( "SOURCE_DATA_TYPE" ) ; //source type of a distinct type or user-generated Ref type, SQL type from java.sql.Types
// result.getIndexes().add(from(index));
String IS_AUTOINCREMENT = rs . getString ( "IS_AUTOINCREMENT" ) ; //Indicates whether this column is auto incremented
// }
// return result;
return null ;
}
private ForeignKey from ( schemacrawler . schema . ForeignKey foreignKey ) {
ForeignKey result = new ForeignKey ( ) ;
result . setForeignKeyTableName ( foreignKey . getForeignKeyTable ( ) . getName ( ) ) ;
result . setPrimaryKeyTableName ( foreignKey . getPrimaryKeyTable ( ) . getName ( ) ) ;
List < ColumnReference > columnReferences = foreignKey . getColumnReferences ( ) ; Column column = new Column ( ) ;
if ( columnReferences ! = null & & ! columnReferences . isEmpty ( ) ) { column . setName ( COLUMN_NAME ) ;
for ( ColumnReference columnReference : columnReferences ) { column . setRemarks ( REMARKS ) ;
schemacrawler . schema . Column foreignKeyColumn = columnReference . getForeignKeyColumn ( ) ; column . setSqlType ( SqlTypeUtil . getSqlTypeName ( DATA_TYPE ) ) ;
schemacrawler . schema . Column primaryKeyColumn = columnReference . getPrimaryKeyColumn ( ) ; column . setJavaType ( SqlTypeUtil . getJavaType ( DATA_TYPE ) ) ;
result . setForeignKeyColumnName ( foreignKeyColumn . getName ( ) ) ; column . setSize ( COLUMN_SIZE ) ;
result . setPrimaryKeyColumnName ( primaryKeyColumn . getName ( ) ) ; column . setNullable ( "YES" . equalsIgnoreCase ( IS_NULLABLE ) ? true : false ) ;
} column . setAutoIncremented ( "YES" . equalsIgnoreCase ( IS_AUTOINCREMENT ) ? true : false ) ;
} return column ;
return result ;
} }
private Column from ( schemacrawler . schema . Column column ) { private List < ForeignKey > getForeignKeys ( DataSource dataSource , String catalogName , String schemaName , String tableName ) throws MetaDataAccessException {
Column result = new Column ( ) ; List < ForeignKey > result = new ArrayList < > ( ) ;
result . setName ( column . getName ( ) ) ; Connection connection = null ;
result . setRemarks ( column . getRemarks ( ) ) ; ResultSet rs = null ;
result . setJavaType ( column . getType ( ) . getTypeMappedClass ( ) ) ; try {
result . setSqlType ( column . getColumnDataType ( ) . getJavaSqlType ( ) . getName ( ) ) ; connection = dataSource . getConnection ( ) ;
result . setVendorTypeNumber ( column . getColumnDataType ( ) . getJavaSqlType ( ) . getVendorTypeNumber ( ) ) ; DatabaseMetaData databaseMetaData = connection . getMetaData ( ) ;
result . setNullable ( column . isNullable ( ) ) ; rs = databaseMetaData . getImportedKeys ( catalogName , schemaName , tableName ) ;
result . setDefaultValue ( column . getDefaultValue ( ) ) ; while ( rs . next ( ) ) {
result . setGenerated ( column . isGenerated ( ) ) ; ForeignKey foreignKey = new ForeignKey ( ) ;
result . setHidden ( column . isHidden ( ) ) ; foreignKey . setPrimaryKeyTableName ( rs . getString ( "PKTABLE_NAME" ) ) ;
result . setAutoIncremented ( column . isAutoIncremented ( ) ) ; foreignKey . setPrimaryKeyColumnName ( rs . getString ( "PKCOLUMN_NAME" ) ) ;
result . setPartOfIndex ( column . isPartOfIndex ( ) ) ; foreignKey . setForeignKeyTableName ( rs . getString ( "FKTABLE_NAME" ) ) ;
result . setPartOfUniqueIndex ( column . isPartOfUniqueIndex ( ) ) ; foreignKey . setForeignKeyColumnName ( rs . getString ( "FKCOLUMN_NAME" ) ) ;
result . setPartOfPrimaryKey ( column . isPartOfPrimaryKey ( ) ) ; result . add ( foreignKey ) ;
result . setSize ( column . getSize ( ) ) ;
result . setWidth ( column . getWidth ( ) ) ;
return result ;
} }
} catch ( SQLException e ) {
private Index from ( schemacrawler . schema . Index index ) { throw new MetaDataAccessException ( e ) ;
Index result = new Index ( ) ; } finally {
result . setName ( index . getName ( ) ) ; try { rs . close ( ) ; } catch ( SQLException e ) { }
result . setUnique ( index . isUnique ( ) ) ; try { connection . close ( ) ; } catch ( SQLException e ) { }
result . setIndexType ( IndexType . from ( index . getIndexType ( ) . id ( ) ) ) ;
List < schemacrawler . schema . IndexColumn > columns = index . getColumns ( ) ;
for ( schemacrawler . schema . IndexColumn column : columns ) {
result . getColumns ( ) . add ( from ( column ) ) ;
} }
return result ; return result ;
} }
