com.kg.component.generator.query.SQLQuery 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.query;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.kg.component.generator.config.IDbQuery;
import com.kg.component.generator.config.ITypeConvert;
import com.kg.component.generator.config.builder.ConfigBuilder;
import com.kg.component.generator.config.builder.Entity;
import com.kg.component.generator.config.po.TableField;
import com.kg.component.generator.config.po.TableInfo;
import com.kg.component.generator.config.querys.H2Query;
import com.kg.component.generator.config.rules.IColumnType;
import com.kg.component.generator.jdbc.DatabaseMetaDataWrapper;
import org.jetbrains.annotations.NotNull;
import java.sql.SQLException;
import java.util.*;
/**
* 这是兼容以前旧版本提供的查询方式，需要每个数据库对接适配。
*
* @author nieqiurong 2021/1/6.
* @see IDbQuery 数据库适配
* @see ITypeConvert 类型适配处理
* @since 3.5.0
*/
public class SQLQuery extends AbstractDatabaseQuery {
public SQLQuery(@NotNull ConfigBuilder configBuilder) {
super(configBuilder);
}
@NotNull
@Override
public List queryTables() {
boolean isInclude = strategyConfig.getInclude().size() > 0;
boolean isExclude = strategyConfig.getExclude().size() > 0;
//所有的表信息
List tableList = new ArrayList<>();
//需要反向生成或排除的表信息
List includeTableList = new ArrayList<>();
List excludeTableList = new ArrayList<>();
try {
dbQuery.execute(dbQuery.tablesSql(), result -> {
String tableName = result.getStringResult(dbQuery.tableName());
if (StringUtils.isNotBlank(tableName)) {
TableInfo tableInfo = new TableInfo(this.configBuilder, tableName);
String tableComment = result.getTableComment();
// 跳过视图
if (!(strategyConfig.isSkipView() && "VIEW".equals(tableComment))) {
tableInfo.setComment(tableComment);
if (isInclude && strategyConfig.matchIncludeTable(tableName)) {
includeTableList.add(tableInfo);
} else if (isExclude && strategyConfig.matchExcludeTable(tableName)) {
excludeTableList.add(tableInfo);
}
tableList.add(tableInfo);
}
}
});
filter(tableList, includeTableList, excludeTableList);
// 性能优化，只处理需执行表字段 https://github.com/baomidou/mybatis-plus/issues/219
tableList.forEach(this::convertTableFields);
return tableList;
} catch (SQLException e) {
throw new RuntimeException(e);
} finally {
// 数据库操作完成,释放连接对象
dbQuery.closeConnection();
}
}
protected void convertTableFields(@NotNull TableInfo tableInfo) {
DbType dbType = this.dataSourceConfig.getDbType();
String tableName = tableInfo.getName();
try {
Map columnsInfoMap = databaseMetaDataWrapper.getColumnsInfo(tableName, false);
String tableFieldsSql = dbQuery.tableFieldsSql(tableName);
Set h2PkColumns = new HashSet<>();
if (DbType.H2 == dbType) {
dbQuery.execute(String.format(H2Query.PK_QUERY_SQL, tableName), result -> {
String primaryKey = result.getStringResult(dbQuery.fieldKey());
if (Boolean.parseBoolean(primaryKey)) {
h2PkColumns.add(result.getStringResult(dbQuery.fieldName()));
}
});
}
Entity entity = strategyConfig.entity();
dbQuery.execute(tableFieldsSql, result -> {
String columnName = result.getStringResult(dbQuery.fieldName());
TableField field = new TableField(this.configBuilder, columnName);
DatabaseMetaDataWrapper.Column column = columnsInfoMap.get(columnName.toLowerCase());
TableField.MetaInfo metaInfo = new TableField.MetaInfo(column);
// 避免多重主键设置，目前只取第一个找到ID，并放到list中的索引为0的位置
boolean isId = DbType.H2 == dbType ? h2PkColumns.contains(columnName) : result.isPrimaryKey();
// 处理ID
if (isId) {
field.primaryKey(dbQuery.isKeyIdentity(result.getResultSet()));
tableInfo.setHavePrimaryKey(true);
if (field.isKeyIdentityFlag() && entity.getIdType() != null) {
LOGGER.warn("当前表[{}]的主键为自增主键，会导致全局主键的ID类型设置失效!", tableName);
}
}
// 处理ID
field.setColumnName(columnName)
.setType(result.getStringResult(dbQuery.fieldType()))
.setComment(result.getFiledComment())
.setCustomMap(dbQuery.getCustomFields(result.getResultSet()));
String propertyName = entity.getNameConvert().propertyNameConvert(field);
IColumnType columnType = dataSourceConfig.getTypeConvert().processTypeConvert(globalConfig, field);
field.setPropertyName(propertyName, columnType);
field.setMetaInfo(metaInfo);
tableInfo.addField(field);
});
} catch (SQLException e) {
throw new RuntimeException(e);
}
tableInfo.processTable();
}
}