Mybatis源码研究6：元数据(metadata)

原理分析之六：元数据(metadata)

一、依赖关系

本包对mybatis的其它包没有任何依赖，mybatis的其它包也不依赖于本包，

即本包可以作为单独的组件或者工具类提出来。

二、类功能概述

包含了3个元数据实体类和一个工厂类。

Table是对表的简单封装，包括name,columns,primaryKey,catalog，schema。

Column是对列的简单封装，包括name和type。

Database是对数据库的简单封装，包括catalog，schema，tables。

DatabaseFactory用来创建Database对象。

三、类的方法和属性

//数据库-表格-列

public class Column {

// 列名

private String name;

// 列的类型

private int type;

//构造方法

//getter/setter方法

｝

//数据库-表格

public class Table {

//表名

private final String name;

//类别名

private String catalog;

//模式名

private String schema;

//列的集合

private final Map<String, Column> columns = new HashMap<String, Column>();

//主键列

private Column primaryKey;

//构造方法

//getter/setter方法

｝

//数据库

public class Database {

// 类别名称；它必须与存储在数据库中的类别名称匹配；该参数为 "" 表示获取没有类别的那些描述；为 null 则表示该类别名称不应该用于缩小搜索范围

private final String catalog;

// schemaPattern - 模式名称的模式；它必须与存储在数据库中的模式名称匹配；该参数为 "" 表示获取没有模式的那些描述；为 null

// 则表示该模式名称不应该用于缩小搜索范围

private final String schema;

// 表格的集合

private final Map<String, Table> tables = new HashMap<String, Table>();

//构造方法

//getter/setter方法

}

本包除了3个实体类封装了 属性和数据结构，只有一个核心方法。

public class DatabaseFactory {

// 根据数据库连接-类别-模式，生成Database对象。

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);

// 遍历结果集，取出表的信息，构建Table对象，加入到Databse的Table集合中

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();

// 遍历所有的Table，取出列名和主键信息

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;

}

｝

四、应用实例

public static void main(String[] args) throws ClassNotFoundException, SQLException {

Connection conn = getConnection();

String catalogFilter = null;

String schemaFilter = null;

Database db = DatabaseFactory.newDatabase(conn, catalogFilter,

schemaFilter);

pn("catalog:" + db.getCatalog());

pn("schema:" + db.getSchema());

String[] tableNames = db.getTableNames();

for (String name : tableNames) {

Table table = db.getTable(name);

p("catalog:"+table.getCatalog()+",");

p("schema:"+table.getSchema()+",");

String[] columnNames = table.getColumnNames();

for (String columnName : columnNames) {

Column column = table.getColumn(columnName);

p(column.getName()+",");

p(column.getType()+",");

}

pn("");

}

if (conn != null) {

conn.close();

}

}

// 简化打印

public static void pn(Object str) {

System.out.println(str);

}

public static void p(Object str) {

System.out.print(str);

}

// 获取数据库连接

public static Connection getConnection() throws ClassNotFoundException,

SQLException {

String driver = "com.mysql.jdbc.Driver";

String url = "jdbc:mysql://localhost:3306/mysql?userUnicode=true&characterEncoding=UTF8";

String user = "root";

String password = "123456";

Class.forName(driver);

return DriverManager.getConnection(url, user, password);

}

打印结果：

catalog:null

schema:null

catalog:mysql,schema:null,Host,1,Table_name,1,Table_priv,1,Column_priv,1,Grantor,1,Db,1,Timestamp,93,User,1,

catalog:mysql,schema:null,Use_leap_seconds,1,Time_zone_id,4

......