package com.iplatform.mybatis.service;
|
|
import com.iplatform.mybatis.domain.GenTable;
|
import com.iplatform.mybatis.domain.GenTableColumn;
|
import com.iplatform.mybatis.util.DataTypeUtils;
|
import com.iplatform.mybatis.util.SqlUtils;
|
import com.walker.db.DatabaseType;
|
import com.walker.infrastructure.utils.DateUtils;
|
import com.walker.infrastructure.utils.StringUtils;
|
import com.walker.jdbc.JdbcInspector;
|
import com.walker.jdbc.service.BaseServiceImpl;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
@Service
|
public class MetaDataServiceImpl extends BaseServiceImpl {
|
|
/**
|
* 返回从元数据中查询的表结构信息
|
* @param tableName 表名称，可以是表前缀
|
* @param isPrecision 是否精确查找，如果是则仅返回和表名一样的表结构。
|
* @param packageName 包名称，必填
|
* @param moduleName 模块名称，选填
|
* @param businessName 业务名称，选填
|
* @param functionName 功能名称，选填
|
* @return
|
* @date 2024-02-20
|
*/
|
public List<GenTable> queryDatabaseTableInfo(String tableName, boolean isPrecision
|
, String packageName, String moduleName, String businessName, String functionName){
|
List<GenTable> genTableList = new ArrayList<>(8);
|
DatabaseType databaseType = JdbcInspector.getInstance().getPrimaryDatabaseType();
|
log.debug("databaseType = " + databaseType.toString());
|
if(DataTypeUtils.isOracle(databaseType)){
|
// oracle元数据中表名都是大写
|
tableName = tableName.toUpperCase();
|
} else {
|
tableName = tableName.toLowerCase();
|
}
|
|
Map<String, Object> sqlMap = new HashMap<>();
|
if(isPrecision){
|
sqlMap.put("tablename", "%" + tableName);
|
} else {
|
sqlMap.put("tablename", tableName + "%");
|
}
|
|
String sql_table = "";
|
// String sql_pk = "";
|
String sql_column = "";
|
|
if (DataTypeUtils.isOracle(databaseType)) {
|
sql_table = "select table_name from user_tables where table_name like :tablename order by table_name desc";
|
// sql_pk = "select col.table_name,col.column_name from user_constraints con,user_cons_columns col where con.constraint_name=col.constraint_name and con.constraint_type='P' and col.table_name like :tablename ";
|
sql_column = SqlUtils.ORACLE_QUERY_COLUMNS;
|
} else if(DataTypeUtils.isMysql(databaseType)) {
|
sql_table = "select table_name from information_schema.tables where table_schema=(SELECT DATABASE()) and table_name like :tablename order by table_name desc";
|
// sql_pk = "select table_name,column_name from INFORMATION_SCHEMA.Columns where column_key='PRI' AND table_schema=(SELECT DATABASE()) and table_name like :tablename";
|
sql_column = "select table_name,column_name,data_type,NUMERIC_PRECISION data_precision,NUMERIC_SCALE data_scale, column_key from INFORMATION_SCHEMA.Columns where table_schema=(SELECT DATABASE()) and table_name like :tablename order by table_name,ordinal_position\n";
|
|
} else if(databaseType == DatabaseType.POSTGRES){
|
sql_table = "SELECT c.relname table_name from pg_class c, pg_namespace n where c.relkind = 'r' and n.nspname = 'public' and c.relnamespace = n.oid and c.relname like :tablename";
|
sql_column = SqlUtils.SQL_PG_FIELDS;
|
log.debug("POSTGRES......");
|
|
} else {
|
throw new UnsupportedOperationException("（po生成）不支持的数据库类型：" + databaseType);
|
}
|
List<Map<String, Object>> table_list = this.dao.select(sql_table, sqlMap);
|
List<Map<String, Object>> column_list = this.dao.select(sql_column, sqlMap);
|
|
log.debug("生成表结构（PO） = {}", table_list);
|
|
for (Map<String, Object> table : table_list) {
|
String table_name = table.get("table_name").toString();
|
String pk_name = "id";
|
String pk_type = "NUMBER";
|
List<Map<String, Object>> column_temp = new ArrayList<>();
|
|
String columnKey = null;
|
for (Map<String, Object> column : column_list) {
|
if(!column.get("table_name").toString().equalsIgnoreCase(table.get("table_name").toString())){
|
// 只有当前表字段才处理，老代码遍历所有表字段有问题。2022-09-20
|
continue;
|
}
|
// 2022-09-07 修改（后续有oracle需求还需要改）
|
if(column.get("column_key") != null){
|
columnKey = column.get("column_key").toString();
|
if(DataTypeUtils.isPrimaryColumn(columnKey)){
|
// mysql主键字段是：PRI，oracle是：P，2023-03-03 修改
|
pk_name = column.get("column_name").toString().toLowerCase();
|
pk_type = column.get("data_type").toString().toLowerCase();
|
log.info(table_name + " 找到主键:" + pk_name + ", " + pk_type);
|
continue;
|
}
|
}
|
|
if ((table_name.equalsIgnoreCase(column.get("table_name").toString())) && (!column.get("column_name").toString().equalsIgnoreCase(pk_name))) {
|
String type = column.get("data_type").toString().toLowerCase();
|
column.put("name", column.get("column_name").toString().toLowerCase());
|
if ((type.equalsIgnoreCase("number")) || (type.equalsIgnoreCase("decimal"))) {
|
int precision = Integer.valueOf(column.get("data_precision") == null ? "0" : column.get("data_precision").toString()).intValue();
|
int scale = Integer.valueOf(column.get("data_scale") == null ? "0" : column.get("data_scale").toString()).intValue();
|
type = DataTypeUtils.getType(type, precision, scale);
|
}
|
column.put("type", type);
|
column_temp.add(column);
|
}
|
}
|
table.put("table_name", table_name.toLowerCase());
|
table.put("pk_name", pk_name);
|
// table.put("pk_type", pk_type);
|
table.put("pk_type", DataTypeUtils.getType(pk_type, 0, 0));
|
table.put("columns", column_temp);
|
|
GenTable genTable = this.acquireGenTable(table, packageName, moduleName, businessName, functionName, pk_name);
|
genTableList.add(genTable);
|
}
|
return genTableList;
|
}
|
|
private GenTable acquireGenTable(Map<String, Object> table
|
, String packageName, String moduleName, String businessName, String functionName, String pkName){
|
GenTable genTable = new GenTable();
|
genTable.setCreateTime(DateUtils.getDateTimeNumber());
|
genTable.setTableName(table.get("table_name").toString());
|
genTable.setTableComment(StringUtils.EMPTY_STRING);
|
genTable.setClassName(StringUtils.transferUnderlineName2Camel(genTable.getTableName(), true));
|
genTable.setTplCategory("crud");
|
genTable.setGenType("0");
|
genTable.setFunctionAuthor("shikeying");
|
genTable.setPackageName(packageName);
|
// genTable.setGenPath("/");
|
if(StringUtils.isNotEmpty(moduleName)){
|
genTable.setModuleName(moduleName);
|
}
|
if(StringUtils.isNotEmpty(businessName)){
|
genTable.setBusinessName(businessName);
|
}
|
if(StringUtils.isNotEmpty(functionName)){
|
genTable.setFunctionName(functionName);
|
}
|
|
List<GenTableColumn> genTableColumnList = new ArrayList<>();
|
List<Map<String, Object>> columns = (List<Map<String, Object>>)table.get("columns");
|
GenTableColumn genTableColumn = null;
|
boolean isPrimaryKey = false;
|
|
for(Map<String, Object> map : columns){
|
if(StringUtils.isNotEmpty(pkName) && map.get("column_name").toString().equals(pkName)){
|
isPrimaryKey = true;
|
}
|
genTableColumn = this.acquireGenColumn(map, isPrimaryKey);
|
if(isPrimaryKey){
|
genTable.setPkColumn(genTableColumn);
|
}
|
genTableColumnList.add(genTableColumn);
|
}
|
genTable.setColumns(genTableColumnList);
|
return genTable;
|
}
|
|
private GenTableColumn acquireGenColumn(Map<String, Object> map, boolean isPrimaryKey){
|
GenTableColumn genTableColumn = new GenTableColumn();
|
genTableColumn.setCreateTime(DateUtils.getDateTimeNumber());
|
genTableColumn.setColumnName(map.get("column_name").toString());
|
genTableColumn.setColumnType(map.get("type").toString());
|
genTableColumn.setColumnComment(StringUtils.EMPTY_STRING);
|
if(isPrimaryKey){
|
genTableColumn.setIsPk("1");
|
}
|
genTableColumn.setJavaField(StringUtils.transferUnderlineName2Camel(genTableColumn.getColumnName(), true));
|
return genTableColumn;
|
}
|
}
|
