diff --git a/gen/src/main/java/com/gitee/gen/gen/JavaColumnDefinition.java b/gen/src/main/java/com/gitee/gen/gen/JavaColumnDefinition.java index 28510777f41c74c843850ae9db6452ceb1a40b24..34803f3759e6d4bc305917663ff578ce1d092333 100644 --- a/gen/src/main/java/com/gitee/gen/gen/JavaColumnDefinition.java +++ b/gen/src/main/java/com/gitee/gen/gen/JavaColumnDefinition.java @@ -3,6 +3,7 @@ package com.gitee.gen.gen; import com.gitee.gen.gen.converter.ColumnTypeConverter; import com.gitee.gen.gen.converter.JavaColumnTypeConverter; import com.gitee.gen.util.FieldUtil; +import org.springframework.util.StringUtils; import java.util.HashMap; import java.util.Map; @@ -15,6 +16,7 @@ public class JavaColumnDefinition extends ColumnDefinition { private static final JavaColumnTypeConverter COLUMN_TYPE_CONVERTER = new JavaColumnTypeConverter(); private static final Map TYPE_MYBATIS_MAP = new HashMap<>(64); + static { TYPE_MYBATIS_MAP.put(TypeEnum.BIT.getType(), "BOOLEAN"); TYPE_MYBATIS_MAP.put(TypeEnum.BOOLEAN.getType(), "BOOLEAN"); @@ -50,7 +52,11 @@ public class JavaColumnDefinition extends ColumnDefinition { */ public String getJavaFieldName() { String fieldName = FieldUtil.underlineFilter(getColumnName()); - return fieldName.replaceAll("_", ""); + if(StringUtils.isEmpty(fieldName)){ + return fieldName; + } + fieldName = fieldName.replaceAll("_", ""); + return fieldName.substring(0, 1).toLowerCase() + fieldName.substring(1); } /** diff --git a/gen/src/main/java/com/gitee/gen/gen/TableDefinition.java b/gen/src/main/java/com/gitee/gen/gen/TableDefinition.java index 87000af8874b10fbda7d55aa5e1e27c9cd4db0c6..6e600ad5ff110debdc6d828f92461f8b741e5e3f 100644 --- a/gen/src/main/java/com/gitee/gen/gen/TableDefinition.java +++ b/gen/src/main/java/com/gitee/gen/gen/TableDefinition.java @@ -109,6 +109,20 @@ public class TableDefinition { return false; } + /** + * 是否含有Long字段 + * + * @return + */ + public boolean getHasLongField() { + for (ColumnDefinition definition : columnDefinitions) { + if ("Long".equals(((JavaColumnDefinition) definition).getJavaType())) { + return true; + } + } + return false; + } + /** * 获取主键信息 * diff --git a/gen/src/main/java/com/gitee/gen/gen/dm/DmColumnSelector.java b/gen/src/main/java/com/gitee/gen/gen/dm/DmColumnSelector.java index fa67450a9fd54765f522c2df7e780c79b838b350..7f35e4bc576a34c27c034706dbf2ad2da5fe8ab9 100644 --- a/gen/src/main/java/com/gitee/gen/gen/dm/DmColumnSelector.java +++ b/gen/src/main/java/com/gitee/gen/gen/dm/DmColumnSelector.java @@ -17,7 +17,7 @@ import java.util.Set; */ public class DmColumnSelector extends ColumnSelector { - private static final TypeFormatter TYPE_FORMATTER = new OracleTypeFormatter(); + private static final TypeFormatter TYPE_FORMATTER = new DmTypeFormatter(); private static final String COLUMN_SQL = " SELECT " + " atc.COLUMN_NAME FIELD, atc.DATA_TYPE TYPE, atc.DATA_SCALE SCALE, atc.DATA_LENGTH MAXLENGTH, " + diff --git a/gen/src/main/java/com/gitee/gen/gen/dm/DmTypeFormatter.java b/gen/src/main/java/com/gitee/gen/gen/dm/DmTypeFormatter.java index e35163f9104e6b2bd467812c48f3234064e00889..d70f1cd7b3f250c79230eb70bd2303b57db6548c 100644 --- a/gen/src/main/java/com/gitee/gen/gen/dm/DmTypeFormatter.java +++ b/gen/src/main/java/com/gitee/gen/gen/dm/DmTypeFormatter.java @@ -32,12 +32,12 @@ public class DmTypeFormatter implements TypeFormatter { @Override public boolean isInt(String columnType) { - return contains(Arrays.asList("int", "integer"), columnType); + return !isLong(columnType) && contains(Arrays.asList("int", "integer"), columnType); } @Override public boolean isLong(String columnType) { - return !isVarchar(columnType) && contains(Collections.singletonList("long"), columnType); + return !isVarchar(columnType) && contains(Collections.singletonList("bigint"), columnType); } @Override