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 index 733c13a..c2aa9f6 100644 --- 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,27 +84,27 @@ public class PojoModelProcess implements PojoProcess { Set importList = new HashSet<>(); List 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 @@ -112,9 +112,9 @@ public class PojoModelProcess implements PojoProcess { typeModel .setSetName(nameStrategy.transSetName(column.getColumnName(), classType)); } - + fieldList.add(typeModel); - }*/ + } pojoModel.setImportList(importList); pojoModel.setFieldList(fieldList); pojoModels.add(pojoModel);