Browse Source

:white_check_mark: fix 代码生成问题

Aron 6 years ago
parent
commit
031977e83e
1 changed files with 13 additions and 4 deletions
  1. 13 4
      src/main/java/com/ifast/common/utils/GenUtils.java

+ 13 - 4
src/main/java/com/ifast/common/utils/GenUtils.java

@@ -29,10 +29,12 @@ import java.util.zip.ZipOutputStream;
  */
  */
 public class GenUtils {
 public class GenUtils {
 
 
+    public static String STR_DELIMITER = "_";
+
     private static Logger log = LoggerFactory.getLogger(GenUtils.class);
     private static Logger log = LoggerFactory.getLogger(GenUtils.class);
 
 
     public static List<String> getTemplates() {
     public static List<String> getTemplates() {
-        List<String> templates = new ArrayList<String>();
+        List<String> templates = new ArrayList<>();
         templates.add("templates/common/generator/domain.java.vm");
         templates.add("templates/common/generator/domain.java.vm");
         templates.add("templates/common/generator/Dao.java.vm");
         templates.add("templates/common/generator/Dao.java.vm");
         templates.add("templates/common/generator/Mapper.xml.vm");
         templates.add("templates/common/generator/Mapper.xml.vm");
@@ -72,7 +74,9 @@ public class GenUtils {
         List<String> baseColumnNames = Arrays.asList("deleted", "version", "createAt", "createBy", "updateAt", "updateBy");
         List<String> baseColumnNames = Arrays.asList("deleted", "version", "createAt", "createBy", "updateAt", "updateBy");
         for (Map<String, String> column : columns) {
         for (Map<String, String> column : columns) {
         	columnNames.add(column.get("columnName"));
         	columnNames.add(column.get("columnName"));
-        	if(baseColumnNames.contains(column.get("columnName"))) continue;
+        	if(baseColumnNames.contains(column.get("columnName"))) {
+        	    continue;
+            }
         	
         	
             ColumnDO columnDO = new ColumnDO();
             ColumnDO columnDO = new ColumnDO();
             columnDO.setColumnName(column.get("columnName"));
             columnDO.setColumnName(column.get("columnName"));
@@ -164,15 +168,20 @@ public class GenUtils {
      * 列名转换成Java属性名
      * 列名转换成Java属性名
      */
      */
     public static String columnToJava(String columnName) {
     public static String columnToJava(String columnName) {
-        return WordUtils.capitalizeFully(columnName, new char[] { '_' }).replace("_", "");
+        if(columnName.contains(STR_DELIMITER)){
+            return WordUtils.capitalizeFully(columnName, new char[] { '_' }).replace(STR_DELIMITER, "");
+        }else {
+            return WordUtils.uncapitalize(columnName);
+        }
     }
     }
 
 
     /**
     /**
      * 表名转换成Java类名
      * 表名转换成Java类名
      */
      */
     public static String tableToJava(String tableName, String tablePrefix, String autoRemovePre) {
     public static String tableToJava(String tableName, String tablePrefix, String autoRemovePre) {
+
         if (Constant.Generator.AUTO_REOMVE_PRE.equals(autoRemovePre)) {
         if (Constant.Generator.AUTO_REOMVE_PRE.equals(autoRemovePre)) {
-            tableName = tableName.substring(tableName.indexOf("_") + 1);
+            tableName = tableName.substring(tableName.indexOf(STR_DELIMITER) + 1);
         }
         }
         if (StringUtils.isNotBlank(tablePrefix)) {
         if (StringUtils.isNotBlank(tablePrefix)) {
             tableName = tableName.replace(tablePrefix, "");
             tableName = tableName.replace(tablePrefix, "");