根据数据库名生成数据库结构说明
根据数据库名生成数据库结构说明
根据数据库名生成数据库结构说明:
- package com.cjm.common;
- import java.io.File;
- import java.io.FileOutputStream;
- import java.sql.Connection;
- import java.sql.DatabaseMetaData;
- import java.sql.DriverManager;
- import java.sql.PreparedStatement;
- import java.sql.ResultSet;
- import java.sql.ResultSetMetaData;
- import java.sql.Statement;
- import java.util.ArrayList;
- import java.util.HashMap;
- import java.util.List;
- import jxl.Workbook;
- import jxl.write.Label;
- import jxl.write.WritableSheet;
- import jxl.write.WritableWorkbook;
- public class CreateMetadata {
- private Connection cn = null;
- private List fieldTypes = null;
- private List executedTables = null;
- //private String[] types = {"TABLE", "VIEW"}; //只生成表和视图的数据字典
- private String[] types = {"TABLE"}; //只生成表和视图的数据字典
- private String driver;
- private String url;
- private String uid;
- private String pwd;
- private String catalog;
- private String schema;
- public CreateMetadata()throws Exception{
- initData();
- //initSqlServerDBParams();
- initOracleDBParams();
- Class.forName(driver);
- this.cn = DriverManager.getConnection(url, uid, pwd);
- }
- private void initData(){
- //字符串类型
- fieldTypes = new ArrayList();
- fieldTypes.add("CHAR");
- fieldTypes.add("NCHAR");
- fieldTypes.add("VARCHAR");
- fieldTypes.add("NVARCHAR");
- fieldTypes.add("VARCHAR2");
- fieldTypes.add("NVARCHAR2");
- //排除以下表
- executedTables = new ArrayList();
- executedTables.add("dtproperties");
- executedTables.add("sysconstraints");
- executedTables.add("syssegments");
- }
- /**
- * SqlServer数据库连接参数
- */
- private void initSqlServerDBParams(){
- catalog = "test"; //SqlServer的数据库名
- schema = null;
- driver = "net.sourceforge.jtds.jdbc.Driver";
- url = "jdbc:jtds:sqlserver://localhost:1433;DatabaseName=test";
- uid = "test";
- pwd = "test";
- }
- /**
- * Oracle数据库连接参数
- */
- private void initOracleDBParams(){
- catalog = null;
- schema = "GISAP"; //Oracle的用户名
- driver = "oracle.jdbc.driver.OracleDriver";
- url = "jdbc:oracle:thin:@localhost:1521:ORCL";
- uid = "gisap";
- pwd = "1";
- }
- /**
- * 取得一个表的所有主键字段
- */
- private String getTablePrimaryKeys(String tableName){
- try{
- DatabaseMetaData dbmd = cn.getMetaData();
- ResultSet rs = dbmd.getPrimaryKeys(catalog, schema, tableName);
- StringBuffer sb = new StringBuffer(",");
- while(rs.next()){
- sb.append(rs.getString("COLUMN_NAME") + ",");
- }
- rs.close();
- return sb.toString();
- }catch(Exception ex){
- return "";
- }
- }
- /**
- * 取得一个表的所有主键字段
- */
- private String getSqlStr(String tableName){
- StringBuffer sql = new StringBuffer();
- sql.append(" SELECT A.COLUMN_NAME 字段名, ");
- sql.append(" A.DATA_TYPE 数据类型, ");
- sql.append(" A.DATA_LENGTH 长度, ");
- sql.append(" A.DATA_PRECISION 整数位, ");
- sql.append(" A.DATA_SCALE 小数位, ");
- sql.append(" A.NULLABLE 允许空值, ");
- sql.append(" A.DATA_DEFAULT 缺省值, ");
- sql.append(" B.COMMENTS 备注, ");
- sql.append(" C.INDEXCOUNT 索引次数 ");
- sql.append(" FROM USER_TAB_COLUMNS A, ");
- sql.append(" USER_COL_COMMENTS B, ");
- sql.append(" (SELECT COUNT(*) INDEXCOUNT, COLUMN_NAME ");
- sql.append(" FROM USER_IND_COLUMNS ");
- sql.append(" WHERE TABLE_NAME = '"+tableName+"' ");
- sql.append(" GROUP BY COLUMN_NAME) C ");
- sql.append(" WHERE A.TABLE_NAME = B.TABLE_NAME ");
- sql.append(" AND A.COLUMN_NAME = B.COLUMN_NAME ");
- sql.append(" AND A.COLUMN_NAME = C.COLUMN_NAME(+) ");
- sql.append(" AND A.TABLE_NAME = '"+tableName+"' ");
- return sql.toString();
- }
- /**
- * 生成数据字典
- */
- public void createTableMetadata(String fileName){
- try{
- if(fileName == null || fileName.length() == 0){
- throw new IllegalArgumentException("fileName is null");
- }
- System.out.println("fileName:"+fileName);
- File file = new File(fileName);
- //delete old file
- if(file.exists() && file.isFile()) file.delete();
- //create sheet
- WritableWorkbook book = Workbook.createWorkbook(new FileOutputStream(file));
- WritableSheet sheet = book.createSheet("数据字典",0);
- DatabaseMetaData dbmd = cn.getMetaData();
- ResultSet rs = dbmd.getTables(catalog ,schema,"%", types);
- int rowIndex = 0;
- int tableCount = 0;
- while(rs.next()){
- try{
- String tableName = rs.getString("TABLE_NAME");
- System.out.println("tableName:"+tableName);
- //排除表
- if(executedTables.contains(tableName.toLowerCase())) continue;
- tableCount++;
- System.out.println(tableCount + "、" + tableName + " doing...");
- //表名
- sheet.mergeCells(0, rowIndex, 9, rowIndex); //合并单元格,5数字要与表头的cell个数一致
- sheet.addCell(new Label(0, rowIndex, tableCount + "、" + tableName));
- rowIndex++;
- //表头
- sheet.addCell(new Label(0,rowIndex,"序号"));
- sheet.addCell(new Label(1,rowIndex,"字段名"));
- sheet.addCell(new Label(2,rowIndex,"字段类型"));
- sheet.addCell(new Label(3,rowIndex,"长度"));
- sheet.addCell(new Label(4,rowIndex,"整数位"));
- sheet.addCell(new Label(5,rowIndex,"小数位"));
- sheet.addCell(new Label(6,rowIndex,"允许空值"));
- sheet.addCell(new Label(7,rowIndex,"缺省值"));
- sheet.addCell(new Label(8,rowIndex,"备注说明"));
- sheet.addCell(new Label(9,rowIndex,"索引次数"));
- rowIndex++;
- PreparedStatement ps = null;
- ps = cn.prepareStatement(this.getSqlStr(tableName));
- ResultSet res = ps.executeQuery();
- int colCnt = res.getMetaData().getColumnCount();
- int recordIndex = 1;
- while (res.next()) {
- sheet.addCell(new Label(0,rowIndex,String.valueOf(recordIndex)));
- for (int i = 1; i <= colCnt; i++) {
- sheet.addCell(new Label(i,rowIndex,res.getString(i)));
- }
- recordIndex++;
- rowIndex++;
- }
- rowIndex += 2;
- res.close();
- ps.close();
- }catch(Exception e){
- e.printStackTrace();
- }
- }
- rs.close();
- System.out.println("DONE");
- book.write();
- book.close();
- }catch(Exception ex){
- ex.printStackTrace();
- }finally{
- try{
- if(cn != null) cn.close();
- }catch(Exception e){
- e.printStackTrace();
- }
- }
- }
- public static void main(String[] args) {
- try{
- CreateMetadata md = new CreateMetadata();
- md.createTableMetadata("C:\\temp\\md.xls");
- }catch(Exception ex){
- ex.printStackTrace();
- }
- }
- }
评论暂时关闭