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

import com.centit.support.database.DBConnect;

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

	@Override
	public void setDBConfig(DBConnect 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) {
			return null;
		}
	}

	@Override
	public void setDBSchema(String schema) {
		
	}
}