public class DatabaseFactory

package org.apache.ibatis.metadata;

import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;

public class DatabaseFactory {
private DatabaseFactory() {
}

public static Database newDatabase(Connection conn, String catalogFilter, String schemaFilter) throws SQLException {
Database database = new Database(catalogFilter, schemaFilter);
ResultSet rs = null;
try {
DatabaseMetaData dbmd = conn.getMetaData();

try {
rs = dbmd.getColumns(catalogFilter, schemaFilter, null, null);
while (rs.next()) {
String catalogName = rs.getString("TABLE_CAT");
String schemaName = rs.getString("TABLE_SCHEM");
String tableName = rs.getString("TABLE_NAME");
String columnName = rs.getString("COLUMN_NAME");
int dataType = Integer.parseInt(rs.getString("DATA_TYPE"));
Table table = database.getTable(tableName);
if (table == null) {
table = new Table(tableName);
table.setCatalog(catalogName);
table.setSchema(schemaName);
database.addTable(table);
}
table.addColumn(new Column(columnName, dataType));
}
} finally {
if (rs != null) rs.close();
}

try {
String[] tableNames = database.getTableNames();
for (int i = 0; i < tableNames.length; i++) {
Table table = database.getTable(tableNames[i]);
rs = dbmd.getPrimaryKeys(catalogFilter, schemaFilter, table.getName());
if (rs.next()) {
String columnName = rs.getString("COLUMN_NAME");
table.setPrimaryKey(table.getColumn(columnName));
}
}
} finally {
if (rs != null) rs.close();
}

} finally {
try {
conn.rollback();
} catch (Exception e) { /*ignore*/ }
}
return database;
}

}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值