com.centit.support.database.metadata.JdbcMetadata Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of centit-database Show documentation
数据库操作通用方法和函数，从以前的util包中分离出来，并且整合了部分sys-module中的函数
There is a newer version: 5.3.2302
package com.centit.support.database.metadata;

import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


public class JdbcMetadata implements DatabaseMetadata {
    protected static final Logger logger = LoggerFactory.getLogger(JdbcMetadata.class);
    private Connection dbc;

    @Override
    public void setDBConfig(Connection dbc) {
        this.dbc = dbc;
    }

    /**
     * 没有获取外键
     */
    @Override
    public SimpleTableInfo getTableMetadata(String tabName) {
        SimpleTableInfo tab = new SimpleTableInfo(tabName);
        try {
            tab.setSchema(dbc.getSchema().toUpperCase());
            DatabaseMetaData dbmd = dbc.getMetaData();

            ResultSet rs = dbmd.getTables(null, dbc.getSchema(), tabName, null);
            if(rs.next()) {
                tab.setTableLabelName(rs.getString("REMARKS"));
            }
            rs.close();

            rs = dbmd.getTables(dbc.getCatalog(), dbc.getSchema(), tabName, null);
            while (rs.next()) {
                SimpleTableField field = new SimpleTableField();
                field.setColumnName(rs.getString("COLUMN_NAME"));
                field.setColumnType(rs.getString("TYPE_NAME"));
                field.setMaxLength(rs.getInt("COLUMN_SIZE"));
                field.setPrecision(rs.getInt("DECIMAL_DIGITS"));
                field.setScale(rs.getInt("COLUMN_SIZE"));
                field.setNullEnable(rs.getString("NULLABLE"));
                field.setColumnComment( rs.getString("REMARKS"));
                field.mapToMetadata();
                tab.getColumns().add(field);
            }
            rs.close();
            rs = dbmd.getPrimaryKeys(dbc.getCatalog(),dbc.getSchema(), tabName);
            while (rs.next()) {
                tab.getPkColumns().add(rs.getString("COLUMN_NAME"));
                tab.setPkName(rs.getString("PK_NAME"));
            }
            rs.close();

            rs = dbmd.getExportedKeys(dbc.getCatalog(),dbc.getSchema(), tabName);
            Map refs = new HashMap();
            while (rs.next()) {
                String fkTableName = rs.getString("FKTABLE_NAME");
                SimpleTableReference ref= refs.get(fkTableName);
                if(ref==null){
                    ref = new SimpleTableReference();
                    ref.setTableName(fkTableName);
                    ref.setParentTableName(tabName);
                    ref.setReferenceCode(rs.getString("FK_NAME"));
                }
                SimpleTableField field = new SimpleTableField();
                field.setColumnName(rs.getString("FKCOLUMN_NAME"));
                ref.getReferenceColumns().put(rs.getString("PKCOLUMN_NAME"),
                        field.getColumnName());
                ref.getFkColumns().add(field);
            }
            rs.close();

            for(Map.Entry entry:refs.entrySet()){
                tab.getReferences().add(entry.getValue());
            }

        } catch (SQLException e) {
            logger.error(e.getMessage(),e);//e.printStackTrace();
        }
        return tab;
    }

    @Override
    public String getDBSchema() {
        try {
            return dbc.getSchema();
        } catch (SQLException e) {
            logger.error(e.getMessage(),e);//e.printStackTrace();
            return null;
        }
    }

    @Override
    public void setDBSchema(String schema) {

    }
}