package com.iplatform.generator.util;
|
|
import com.iplatform.generator.GenConstants;
|
import com.iplatform.model.po.S_gen_column;
|
import com.iplatform.model.po.S_gen_table;
|
import com.walker.db.TableInfo;
|
import com.walker.dbmeta.FieldInfo;
|
import com.walker.infrastructure.utils.DateUtils;
|
import com.walker.infrastructure.utils.NumberGenerator;
|
import com.walker.infrastructure.utils.StringUtils;
|
import com.walker.jdbc.generator.util.GenUtils;
|
|
import java.util.Arrays;
|
|
/**
|
* 代码生成工具类，目前参考了若依的vue前端生成，后台数据库则使用自己原有框架，<p></p>
|
* 但目前按照管理界面，仅能一次选择一个表生成。
|
* @date 2022-11-26
|
*/
|
public class CodeGenUtils {
|
|
public static final S_gen_column createGenColumn(FieldInfo fieldInfo, String userName, long tableId){
|
boolean isPk = false;
|
String dataType = getDbType(fieldInfo.getDataType());
|
S_gen_column s_gen_column = new S_gen_column();
|
s_gen_column.setColumn_id(NumberGenerator.getLongSequenceNumber());
|
s_gen_column.setCreate_time(DateUtils.getDateTimeNumber(System.currentTimeMillis()));
|
s_gen_column.setCreate_by(userName);
|
s_gen_column.setColumn_name(fieldInfo.getFieldName());
|
s_gen_column.setTable_id(String.valueOf(tableId));
|
s_gen_column.setColumn_comment(fieldInfo.getSummary());
|
s_gen_column.setColumn_type(fieldInfo.getDataType());
|
s_gen_column.setSort(fieldInfo.getColumnId());
|
if(StringUtils.isNotEmpty(fieldInfo.getColumnKey()) && fieldInfo.getColumnKey().equalsIgnoreCase("PRI")){
|
s_gen_column.setIs_pk("1");
|
isPk = true;
|
} else {
|
s_gen_column.setIs_pk("0");
|
}
|
|
//
|
s_gen_column.setJava_field(fieldInfo.getFieldName());
|
s_gen_column.setQuery_type(GenConstants.QUERY_EQ);
|
|
if (arraysContains(GenConstants.COLUMNTYPE_STR, dataType)
|
|| arraysContains(GenConstants.COLUMNTYPE_TEXT, dataType)) {
|
// 字符串长度超过500设置为文本域
|
// Integer columnLength = getColumnLength(column.getColumnType());
|
// String htmlType = columnLength >= 500 || arraysContains(GenConstants.COLUMNTYPE_TEXT, dataType) ? GenConstants.HTML_TEXTAREA : GenConstants.HTML_INPUT;
|
String htmlType = arraysContains(GenConstants.COLUMNTYPE_TEXT, dataType) ? GenConstants.HTML_TEXTAREA : GenConstants.HTML_INPUT;
|
s_gen_column.setHtml_type(htmlType);
|
|
} else if (arraysContains(GenConstants.COLUMNTYPE_TIME, dataType)) {
|
s_gen_column.setJava_field(GenConstants.TYPE_DATE);
|
s_gen_column.setHtml_type(GenConstants.HTML_DATETIME);
|
|
} else if (arraysContains(GenConstants.COLUMNTYPE_NUMBER, dataType)){
|
s_gen_column.setHtml_type(GenConstants.HTML_INPUT);
|
|
// 如果是浮点型 统一用double
|
if(fieldInfo.getScale() > 0){
|
s_gen_column.setJava_type(GenConstants.TYPE_DOUBLE);
|
} else if (fieldInfo.getPrecision() > 0 && fieldInfo.getPrecision() <= 10) {
|
// 如果是整形
|
s_gen_column.setJava_type(GenConstants.TYPE_INTEGER);
|
} else {
|
// 长整型
|
s_gen_column.setJava_type(GenConstants.TYPE_LONG);
|
}
|
}
|
|
// 插入字段（默认所有字段都需要插入）
|
s_gen_column.setIs_insert(GenConstants.REQUIRE);
|
// 编辑字段
|
if (!arraysContains(GenConstants.COLUMNNAME_NOT_EDIT, fieldInfo.getFieldName()) && !isPk) {
|
s_gen_column.setIs_edit(GenConstants.REQUIRE);
|
}
|
// 列表字段
|
if (!arraysContains(GenConstants.COLUMNNAME_NOT_LIST, fieldInfo.getFieldName()) && !isPk) {
|
s_gen_column.setIs_list(GenConstants.REQUIRE);
|
}
|
// 查询字段
|
if (!arraysContains(GenConstants.COLUMNNAME_NOT_QUERY, fieldInfo.getFieldName()) && !isPk) {
|
s_gen_column.setIs_query(GenConstants.REQUIRE);
|
}
|
// 查询字段类型
|
if (fieldInfo.getFieldName().endsWith("name") || fieldInfo.getFieldName().endsWith("title")) {
|
s_gen_column.setQuery_type(GenConstants.QUERY_LIKE);
|
}
|
// 状态字段设置单选框
|
if (fieldInfo.getFieldName().endsWith("status")) {
|
s_gen_column.setHtml_type(GenConstants.HTML_RADIO);
|
}
|
// 类型&性别字段设置下拉框
|
else if (fieldInfo.getFieldName().endsWith("type")
|
|| fieldInfo.getFieldName().endsWith("sex")) {
|
s_gen_column.setHtml_type(GenConstants.HTML_SELECT);
|
}
|
// 图片字段设置图片上传控件
|
else if (fieldInfo.getFieldName().endsWith("image")) {
|
s_gen_column.setHtml_type(GenConstants.HTML_IMAGE_UPLOAD);
|
}
|
// 文件字段设置文件上传控件
|
else if (fieldInfo.getFieldName().endsWith("file")) {
|
s_gen_column.setHtml_type(GenConstants.HTML_FILE_UPLOAD);
|
}
|
// 内容字段设置富文本控件
|
else if (fieldInfo.getFieldName().endsWith("content")) {
|
s_gen_column.setHtml_type(GenConstants.HTML_EDITOR);
|
}
|
return s_gen_column;
|
}
|
|
public static final S_gen_table createGenTable(TableInfo tableInfo, String userName, String packageName){
|
S_gen_table s_gen_table = new S_gen_table();
|
s_gen_table.setTable_id(NumberGenerator.getSequenceNumber());
|
s_gen_table.setCreate_time(DateUtils.getDateTimeNumber(System.currentTimeMillis()));
|
s_gen_table.setCreate_by(userName);
|
s_gen_table.setTable_name(tableInfo.getName());
|
s_gen_table.setTable_comment(tableInfo.getSummary());
|
// s_gen_table.setClass_name(convertClassName(genTable.getTableName()));
|
s_gen_table.setClass_name(GenUtils.toUpperFirst(tableInfo.getName()));
|
s_gen_table.setPackage_name(packageName);
|
s_gen_table.setModule_name(getModuleName(packageName));
|
s_gen_table.setBusiness_name(getBusinessName(tableInfo.getName()));
|
s_gen_table.setFunction_name(tableInfo.getSummary());
|
s_gen_table.setFunction_author(userName);
|
return s_gen_table;
|
}
|
|
/**
|
* 获取业务名
|
*
|
* @param tableName 表名
|
* @return 业务名
|
*/
|
public static String getBusinessName(String tableName) {
|
int lastIndex = tableName.lastIndexOf("_");
|
int nameLength = tableName.length();
|
return substring(tableName, lastIndex + 1, nameLength);
|
}
|
|
/**
|
* 获取模块名
|
*
|
* @param packageName 包名
|
* @return 模块名
|
*/
|
public static String getModuleName(String packageName) {
|
int lastIndex = packageName.lastIndexOf(".");
|
int nameLength = packageName.length();
|
return substring(packageName, lastIndex + 1, nameLength);
|
}
|
|
/**
|
* 获取数据库类型字段
|
*
|
* @param columnType 列类型
|
* @return 截取后的列类型
|
*/
|
public static String getDbType(String columnType) {
|
if (columnType.indexOf("(") > 0) {
|
return substringBefore(columnType, "(");
|
} else {
|
return columnType;
|
}
|
}
|
|
/**
|
* 校验数组是否包含指定值
|
*
|
* @param arr 数组
|
* @param targetValue 值
|
* @return 是否包含
|
*/
|
public static boolean arraysContains(String[] arr, String targetValue) {
|
return Arrays.asList(arr).contains(targetValue);
|
}
|
|
public static String substringBefore(final String str, final String separator) {
|
if (StringUtils.isEmpty(str) || separator == null) {
|
return str;
|
}
|
if (separator.isEmpty()) {
|
return StringUtils.EMPTY_STRING;
|
}
|
final int pos = str.indexOf(separator);
|
if (pos == -1) {
|
return str;
|
}
|
return str.substring(0, pos);
|
}
|
|
/**
|
* 截取字符串
|
*
|
* @param str 字符串
|
* @param start 开始
|
* @param end 结束
|
* @return 结果
|
*/
|
public static String substring(final String str, int start, int end) {
|
if (str == null) {
|
return StringUtils.EMPTY_STRING;
|
}
|
|
if (end < 0) {
|
end = str.length() + end;
|
}
|
if (start < 0) {
|
start = str.length() + start;
|
}
|
|
if (end > str.length()) {
|
end = str.length();
|
}
|
|
if (start > end) {
|
return StringUtils.EMPTY_STRING;
|
}
|
|
if (start < 0) {
|
start = 0;
|
}
|
if (end < 0) {
|
end = 0;
|
}
|
return str.substring(start, end);
|
}
|
}
|
