com.kg.component.generator.jdbc.DatabaseMetaDataWrapper Maven / Gradle / Ivy
/*
* Copyright (c) 2011-2021, baomidou ([email protected]).
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
* use this file except in compliance with the License. You may obtain a copy of
* the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations under
* the License.
*/
package com.kg.component.generator.jdbc;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.kg.component.generator.config.DataSourceConfig;
import org.apache.ibatis.type.JdbcType;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.*;
/**
* 数据库数据元包装类
*
* @author nieqiurong 2021/2/8.
* @since 3.5.0
*/
public class DatabaseMetaDataWrapper {
private static final Logger logger = LoggerFactory.getLogger(DatabaseMetaDataWrapper.class);
private final DatabaseMetaData databaseMetaData;
// 暂时只支持一种
private final String catalog;
// 暂时只支持一种
private final String schema;
/**
* 3.5.4版本会删除此方法
*
* @param connection
* @see #DatabaseMetaDataWrapper(DataSourceConfig)
*/
@Deprecated
public DatabaseMetaDataWrapper(Connection connection) {
try {
this.databaseMetaData = connection.getMetaData();
this.catalog = connection.getCatalog();
this.schema = connection.getSchema();
} catch (SQLException e) {
throw new RuntimeException("获取元数据错误:", e);
}
}
public DatabaseMetaDataWrapper(DataSourceConfig dataSourceConfig) {
try {
Connection connection = dataSourceConfig.getConn();
this.databaseMetaData = connection.getMetaData();
this.catalog = connection.getCatalog();
this.schema = dataSourceConfig.getSchemaName();
} catch (SQLException e) {
throw new RuntimeException("获取元数据错误:", e);
}
}
public Map getColumnsInfo(String tableNamePattern, boolean queryPrimaryKey) {
return getColumnsInfo(this.catalog, this.schema, tableNamePattern, queryPrimaryKey);
}
/**
* 获取表字段信息
*
* @return 表字段信息 (小写字段名->字段信息)
*/
public Map getColumnsInfo(String catalog, String schema, String tableName, boolean queryPrimaryKey) {
Set primaryKeys = new HashSet<>();
if (queryPrimaryKey) {
try (ResultSet primaryKeysResultSet = databaseMetaData.getPrimaryKeys(catalog, schema, tableName)) {
while (primaryKeysResultSet.next()) {
String columnName = primaryKeysResultSet.getString("COLUMN_NAME");
primaryKeys.add(columnName);
}
if (primaryKeys.size() > 1) {
logger.warn("当前表:{}，存在多主键情况！", tableName);
}
} catch (SQLException e) {
throw new RuntimeException("读取表主键信息:" + tableName + "错误:", e);
}
}
Map columnsInfoMap = new LinkedHashMap<>();
try (ResultSet resultSet = databaseMetaData.getColumns(catalog, schema, tableName, "%")) {
while (resultSet.next()) {
Column column = new Column();
String name = resultSet.getString("COLUMN_NAME");
column.name = name;
column.primaryKey = primaryKeys.contains(name);
column.jdbcType = JdbcType.forCode(resultSet.getInt("DATA_TYPE"));
column.length = resultSet.getInt("COLUMN_SIZE");
column.scale = resultSet.getInt("DECIMAL_DIGITS");
column.remarks = formatComment(resultSet.getString("REMARKS"));
column.defaultValue = resultSet.getString("COLUMN_DEF");
column.nullable = resultSet.getInt("NULLABLE") == DatabaseMetaData.columnNullable;
try {
column.autoIncrement = "YES".equals(resultSet.getString("IS_AUTOINCREMENT"));
} catch (SQLException sqlException) {
logger.warn("获取IS_AUTOINCREMENT出现异常:", sqlException);
// 目前测试在oracle旧驱动下存在问题，降级成false.
}
columnsInfoMap.put(name.toLowerCase(), column);
}
return Collections.unmodifiableMap(columnsInfoMap);
} catch (SQLException e) {
throw new RuntimeException("读取表字段信息:" + tableName + "错误:", e);
}
}
public String formatComment(String comment) {
return StringUtils.isBlank(comment) ? StringPool.EMPTY : comment.replaceAll("\r\n", "\t");
}
public Table getTableInfo(String tableName) {
return getTableInfo(this.catalog, this.schema, tableName);
}
public List getTables(String tableNamePattern, String[] types) {
return getTables(this.catalog, this.schema, tableNamePattern, types);
}
public List getTables(String catalog, String schemaPattern, String tableNamePattern, String[] types) {
List tables = new ArrayList<>();
try (ResultSet resultSet = databaseMetaData.getTables(catalog, schemaPattern, tableNamePattern, types)) {
Table table;
while (resultSet.next()) {
table = new Table();
table.name = resultSet.getString("TABLE_NAME");
table.remarks = formatComment(resultSet.getString("REMARKS"));
table.tableType = resultSet.getString("TABLE_TYPE");
tables.add(table);
}
} catch (SQLException e) {
throw new RuntimeException("读取数据库表信息出现错误", e);
}
return tables;
}
public Table getTableInfo(String catalog, String schema, String tableName) {
Table table = new Table();
// 后面要根据表是否为试图来查询，后面重构表查询策略。
try (ResultSet resultSet = databaseMetaData.getTables(catalog, schema, tableName, new String[]{"TABLE", "VIEW"})) {
table.name = tableName;
while (resultSet.next()) {
table.remarks = formatComment(resultSet.getString("REMARKS"));
table.tableType = resultSet.getString("TABLE_TYPE");
}
} catch (SQLException e) {
throw new RuntimeException("读取表信息:" + tableName + "错误:", e);
}
return table;
}
public static class Table {
private String name;
private String remarks;
private String tableType;
public String getRemarks() {
return remarks;
}
public String getTableType() {
return tableType;
}
public String getName() {
return name;
}
public boolean isView() {
return "VIEW".equals(tableType);
}
}
public static class Column {
private boolean primaryKey;
private boolean autoIncrement;
private String name;
private int length;
private boolean nullable;
private String remarks;
private String defaultValue;
private int scale;
private JdbcType jdbcType;
public String getName() {
return name;
}
public int getLength() {
return length;
}
public boolean isNullable() {
return nullable;
}
public String getRemarks() {
return remarks;
}
public String getDefaultValue() {
return defaultValue;
}
public int getScale() {
return scale;
}
public JdbcType getJdbcType() {
return jdbcType;
}
public boolean isPrimaryKey() {
return primaryKey;
}
public boolean isAutoIncrement() {
return autoIncrement;
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy