Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.baomidou.mybatisplus.generator.jdbc.DatabaseMetaDataWrapper Maven / Gradle / Ivy
/*
* Copyright (c) 2011-2023, 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
*
* http://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.baomidou.mybatisplus.generator.jdbc;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
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 Connection connection;
private final DatabaseMetaData databaseMetaData;
//TODO 暂时只支持一种
private final String catalog;
//TODO 暂时只支持一种
private final String schema;
public DatabaseMetaDataWrapper(Connection connection, String schemaName) {
try {
if (null == connection) {
throw new RuntimeException("数据库连接不能为空");
}
this.connection = connection;
this.databaseMetaData = connection.getMetaData();
this.catalog = connection.getCatalog();
this.schema = schemaName;
} catch (SQLException e) {
throw new RuntimeException("获取元数据错误:", e);
}
}
public Connection getConnection() {
return connection;
}
public void closeConnection() {
Optional.ofNullable(connection).ifPresent((con) -> {
try {
con.close();
} catch (SQLException sqlException) {
sqlException.printStackTrace();
}
});
}
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.typeName = resultSet.getString("TYPE_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) {
//TODO 目前测试在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();
//TODO 后面要根据表是否为视图来查询，后面重构表查询策略。
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;
private String typeName;
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;
}
public String getTypeName() {
return typeName;
}
public void setTypeName(String typeName) {
this.typeName = typeName;
}
}
}
