updates

src/main/java/org/yangliu/codegenerate/Main.java

@@ -38,8 +38,8 @@ public class Main {
 
     //实体类模板
     private static String bussi_entity_template = "XXXEntity.javai";
-    private static String bussi_entity_dto_template = "XXXEntityDto.javai";
-    private static String bussi_entity_vo_template = "XXXEntityVo.javai";
+    private static String bussi_entity_dto_template = "XXXEntityDTO.javai";
+    private static String bussi_entity_vo_template = "XXXEntityVO.javai";
 
     //mapper模板
     private static String bussi_mapper_template = "XXXMapper.javai";
@@ -107,17 +107,18 @@ public class Main {
         String entityParsePath = parsePath + (String) dataMap.get("entityPackage") + File.separator;
         MarkerGenerate.generate(templatepath, bussi_entity_template, entityParsePath, entityName + ".java", dataMap);
         String dtoParsePath = parsePath + "dto" + File.separator;
-        MarkerGenerate.generate(templatepath, bussi_entity_dto_template, dtoParsePath, entityName + "Dto.java", dataMap);
+        MarkerGenerate.generate(templatepath, bussi_entity_dto_template, dtoParsePath, entityName + "DTO.java", dataMap);
         String voParsePath = parsePath + "vo" + File.separator;
-        MarkerGenerate.generate(templatepath, bussi_entity_vo_template, voParsePath, entityName + "Vo.java", dataMap);
+        MarkerGenerate.generate(templatepath, bussi_entity_vo_template, voParsePath, entityName + "VO.java", dataMap);
 
         //生成Mapper
         String mapperParsePath = parsePath + "mapper" + File.separator;
         MarkerGenerate.generate(templatepath, bussi_mapper_template, mapperParsePath, entityName + "Mapper.java", dataMap);
+
         //生成Mapper.xml
         String mapperXmlParsePath = project_resource_mapper_path;
         MarkerGenerate.generate(templatepath, bussi_mapperxml_template, mapperXmlParsePath, entityName + "Mapper.xml", dataMap);
-        //生成Service
+        /*//生成Service
         String serviceParsePath = parsePath + "service" + File.separator;
         MarkerGenerate.generate(templatepath,bussi_service_template,serviceParsePath,String.format("I%sService.java",entityName),dataMap);
         //生成ServiceImpl
@@ -125,7 +126,7 @@ public class Main {
         MarkerGenerate.generate(templatepath,bussi_serviceimpl_template,serviceImplParsePath,String.format("%sServiceImpl.java",entityName),dataMap);
         //生成Controller
         String controllerParsePath = parsePath + "controller" + File.separator;
-        MarkerGenerate.generate(templatepath,bussi_controller_template,controllerParsePath,String.format("%sController.java",entityName),dataMap);
+        MarkerGenerate.generate(templatepath,bussi_controller_template,controllerParsePath,String.format("%sController.java",entityName),dataMap);*/
 
     }
 
@@ -169,10 +170,11 @@ public class Main {
                 System.out.println(columnName + " " + columnType + " " + datasize + " " + digits + " " + nullable + " " + remarks);
                 //System.out.println("1"+TableFieldUtil.fieldTypeParseJava(dbType,columnType));
                 String entityField = CaseFormat.LOWER_UNDERSCORE.to(CaseFormat.LOWER_CAMEL, columnName);
-                tableEntityVoList.add(new TableEntityVo(entityField, TableFieldUtil.fieldTypeParseJava(dbType, columnType), columnType, remarks));
+                tableEntityVoList.add(new TableEntityVo(columnName,entityField,TableFieldUtil.fieldTypeParseJava(dbType, columnType), columnType, remarks));
             }
             System.out.println("=================================");
         }
+        connection.close();
         return tableEntityVoList;
     }