♻️ 重构代码

jiluhu · Aug 26, 2020 · 87e7e6b · 87e7e6b
1 parent 4152851
commit 87e7e6b
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/screw-extension/src/main/java/cn/smallbun/screw/extension/pojo/process/PojoModelProcess.java b/screw-extension/src/main/java/cn/smallbun/screw/extension/pojo/process/PojoModelProcess.java
@@ -84,37 +84,37 @@ public List<PojoModel> getPojoModel() {
             Set<String> importList = new HashSet<>();
             List<TypeModel> fieldList = new ArrayList<>();
 
-            /*for (ColumnModel column : model.getColumns()) {
+            for (ColumnModel column : model.getColumns()) {
                 TypeModel typeModel = new TypeModel();
                 typeModel.setFieldName(column.getColumnName());
                 typeModel.setFieldType(column.getTypeName());
                 typeModel.setRemarks(column.getRemarks());
-            
+
                 //先判断用户是否自定义
                 Class<?> classType = dialect.getTypeByMap(customType, column.getTypeName());
                 if (classType == null) {
                     classType = dialect.getClassTypeByFieldType(column.getTypeName());
                 }
-            
+
                 //如果对象不在java.lang包下，需要import
                 if (!classType.getTypeName().startsWith("java.lang")) {
                     importList.add(classType.getTypeName());
                 }
-            
+
                 typeModel
                     .setClassName(nameStrategy.transFieldName(column.getColumnName(), classType));
                 typeModel.setClassType(classType.getSimpleName());
-            
+
                 //如果不使用lombok，需要生成get，set方法
                 if (!pojoConfiguration.isUseLombok()) {
                     typeModel
                         .setGetName(nameStrategy.transGetName(column.getColumnName(), classType));
                     typeModel
                         .setSetName(nameStrategy.transSetName(column.getColumnName(), classType));
                 }
-            
+
                 fieldList.add(typeModel);
-            }*/
+            }
             pojoModel.setImportList(importList);
             pojoModel.setFieldList(fieldList);
             pojoModels.add(pojoModel);