代码生成器，增加 excel 导出功能

src/main/resources/codegen/java/controller/controller.vm

@@ -8,12 +8,16 @@ import io.swagger.annotations.*;
 
 import javax.validation.constraints.*;
 import javax.validation.*;
+import javax.servlet.http.*;
 import java.util.*;
+import java.io.IOException;
 
 import ${PageResultClassName};
 import ${CommonResultClassName};
 import static ${CommonResultClassName}.success;
 
+import ${ExcelUtilsClassName};
+
 import ${basePackage}.${table.moduleName}.controller.${table.businessName}.vo.*;
 import ${basePackage}.${table.moduleName}.dal.dataobject.${table.businessName}.${table.className}DO;
 import ${basePackage}.${table.moduleName}.convert.${table.businessName}.${table.className}Convert;
@@ -73,6 +77,14 @@ public class ${table.className}Controller {
         return success(${table.className}Convert.INSTANCE.convertPage(pageResult));
     }
 
-
+    @GetMapping("/export-excel")
+    @ApiOperation("导出${table.classComment} Excel")
+    public void export${simpleClassName}Excel(@Valid ${table.className}ExportReqVO exportReqVO,
+              HttpServletResponse response) throws IOException {
+        List<${table.className}DO> list = ${classNameVar}Service.get${simpleClassName}List(exportReqVO);
+        // 导出 Excel
+        List<${table.className}ExcelVO> datas = ${table.className}Convert.INSTANCE.convertList02(list);
+        ExcelUtils.write(response, "${table.classComment}.xls", "数据", ${table.className}ExcelVO.class, datas);
+    }
 
 }

src/main/resources/codegen/java/controller/vo/excelReqVO.vm

@@ -1,42 +0,0 @@
-package ${basePackage}.${table.moduleName}.controller.${table.businessName}.vo;
-
-import lombok.*;
-import java.util.*;
-import io.swagger.annotations.*;
-import ${PageParamClassName};
-## 处理 Date 字段的引入
-#foreach ($column in $columns)
-    #if (${column.listOperation} && ${column.javaType} == "Date")## 时间类型
-		import org.springframework.format.annotation.DateTimeFormat;
-
-		import static ${DateUtilsClassName}.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND;
-        #break
-    #end
-#end
-## 字段模板
-#macro(columnTpl $prefix $prefixStr)
-    #if (${column.javaType} == "Date")## 时间类型
-		@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND)
-    #end
-@ApiModelProperty(value = "${prefixStr}${column.columnComment}"#if ("$!column.example" != ""), example = "${column.example}"#end)
-private ${column.javaType}#if ("$!prefix" != "") ${prefix}${JavaField}#else ${column.javaField}#end;
-#end
-
-@ApiModel(value = "${table.classComment} Excel 导出 Request VO", description = "参数和 ${table.className}PageReqVO 是一致的")
-@Data
-public class ${table.className}ExcelReqVO {
-
-#foreach ($column in $columns)
-    #set ($JavaField = $column.javaField.substring(0,1).toUpperCase() + ${column.javaField.substring(1)})##首字母大写
-    #if (${column.listOperation})##查询操作
-        #if (${column.listOperationCondition} == "BETWEEN")## 情况一，Between 的时候
-            #columnTpl('begin', '开始')
-
-            #columnTpl('end', '结束')
-        #else##情况二，非 Between 的时间
-            #columnTpl('', '')
-        #end
-
-    #end
-#end
-}

src/main/resources/codegen/java/controller/vo/excelVO.vm

@@ -0,0 +1,35 @@
+package ${basePackage}.${table.moduleName}.controller.${table.businessName}.vo;
+
+import lombok.*;
+import java.util.*;
+import io.swagger.annotations.*;
+
+import com.alibaba.excel.annotation.ExcelProperty;
+#foreach ($column in $columns)
+#if ("$!column.dictType" != "")## 有设置数据字典
+import ${DictFormatClassName};
+
+import static ${SysDictTypeEnumClassName}.*;
+#break
+#end
+#end
+
+/**
+ * ${table.classComment} Excel VO
+ *
+ * @author ${table.author}
+ */
+@Data
+public class ${table.className}ExcelVO {
+
+#foreach ($column in $columns)
+    #if (${column.listOperationResult})##返回字段
+    @ExcelProperty("${column.columnComment}")
+    #if ("$!column.dictType" != "")##处理枚举值
+    @DictFormat(${column.dictType.toUpperCase()})
+    #end
+    private ${column.javaType} ${column.javaField};
+
+    #end
+#end
+}

src/main/resources/codegen/java/controller/vo/exportReqVO.vm

@@ -0,0 +1,42 @@
+package ${basePackage}.${table.moduleName}.controller.${table.businessName}.vo;
+
+import lombok.*;
+import java.util.*;
+import io.swagger.annotations.*;
+import ${PageParamClassName};
+## 处理 Date 字段的引入
+#foreach ($column in $columns)
+#if (${column.listOperation} && ${column.javaType} == "Date")## 时间类型
+import org.springframework.format.annotation.DateTimeFormat;
+
+import static ${DateUtilsClassName}.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND;
+#break
+#end
+#end
+## 字段模板
+#macro(columnTpl $prefix $prefixStr)
+#if (${column.javaType} == "Date")## 时间类型
+    @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND)
+#end
+    @ApiModelProperty(value = "${prefixStr}${column.columnComment}"#if ("$!column.example" != ""), example = "${column.example}"#end)
+    private ${column.javaType}#if ("$!prefix" != "") ${prefix}${JavaField}#else ${column.javaField}#end;
+#end
+
+@ApiModel(value = "${table.classComment} Excel 导出 Request VO", description = "参数和 ${table.className}PageReqVO 是一致的")
+@Data
+public class ${table.className}ExportReqVO {
+
+#foreach ($column in $columns)
+#set ($JavaField = $column.javaField.substring(0,1).toUpperCase() + ${column.javaField.substring(1)})##首字母大写
+#if (${column.listOperation})##查询操作
+#if (${column.listOperationCondition} == "BETWEEN")## 情况一，Between 的时候
+    #columnTpl('begin', '开始')
+
+    #columnTpl('end', '结束')
+#else##情况二，非 Between 的时间
+    #columnTpl('', '')
+#end
+
+#end
+#end
+}

src/main/resources/codegen/java/convert/convert.vm

@@ -10,10 +10,10 @@ import ${basePackage}.${table.moduleName}.controller.${table.businessName}.vo.*;
 import ${basePackage}.${table.moduleName}.dal.dataobject.${table.businessName}.${table.className}DO;
 
 /**
-* ${table.classComment} Convert
-*
-* @author ${table.author}
-*/
+ * ${table.classComment} Convert
+ *
+ * @author ${table.author}
+ */
 @Mapper
 public interface ${table.className}Convert {
 
@@ -29,4 +29,6 @@ public interface ${table.className}Convert {
 
     PageResult<${table.className}RespVO> convertPage(PageResult<${table.className}DO> page);
 
+    List<${table.className}ExcelVO> convertList02(List<${table.className}DO> list);
+
 }

src/main/resources/codegen/java/dal/do.vm

@@ -6,10 +6,10 @@ import com.baomidou.mybatisplus.annotation.*;
 import ${BaseDOClassName};
 
 /**
-* ${table.classComment} DO
-*
-* @author ${table.author}
-*/
+ * ${table.classComment} DO
+ *
+ * @author ${table.author}
+ */
 @TableName("${table.tableName}")
 @Data
 @EqualsAndHashCode(callSuper = true)

src/main/resources/codegen/java/dal/mapper.vm

@@ -1,48 +1,60 @@
 package ${basePackage}.${table.moduleName}.dal.mysql.${table.businessName};
 
+import java.util.*;
+
 import ${PageResultClassName};
 import ${QueryWrapperClassName};
 import ${BaseMapperClassName};
 import ${basePackage}.${table.moduleName}.dal.dataobject.${table.businessName}.${table.className}DO;
 import org.apache.ibatis.annotations.Mapper;
-import ${basePackage}.${table.moduleName}.controller.${table.businessName}.vo.${table.className}PageReqVO;
+import ${basePackage}.${table.moduleName}.controller.${table.businessName}.vo.*;
 
+## 字段模板
+#macro(listCondition)
+#foreach ($column in $columns)
+#if (${column.listOperation})
+#set ($JavaField = $column.javaField.substring(0,1).toUpperCase() + ${column.javaField.substring(1)})##首字母大写
+#if (${column.listOperationCondition} == "=")##情况一，= 的时候
+                .eqIfPresent("${column.columnName}", reqVO.get${JavaField}())
+#end
+#if (${column.listOperationCondition} == "!=")##情况二，!= 的时候
+                .neIfPresent("${column.columnName}", reqVO.get${JavaField}())
+#end
+#if (${column.listOperationCondition} == ">")##情况三，> 的时候
+                .gtIfPresent("${column.columnName}", reqVO.get${JavaField}())
+#end
+#if (${column.listOperationCondition} == ">=")##情况四，>= 的时候
+                .geIfPresent("${column.columnName}", reqVO.get${JavaField}())
+#end
+#if (${column.listOperationCondition} == "<")##情况五，< 的时候
+                .gtIfPresent("${column.columnName}", reqVO.get${JavaField}())
+#end
+#if (${column.listOperationCondition} == "LIKE")##情况七，Like 的时候
+                .likeIfPresent("${column.columnName}", reqVO.get${JavaField}())
+#end
+#if (${column.listOperationCondition} == "BETWEEN")##情况八，Between 的时候
+                .betweenIfPresent("${column.columnName}", reqVO.getBegin${JavaField}(), reqVO.getEnd${JavaField}())
+#end
+#end
+#end
+#end
 /**
-* ${table.classComment} Mapper
-*
-* @author ${table.author}
-*/
+ * ${table.classComment} Mapper
+ *
+ * @author ${table.author}
+ */
 @Mapper
 public interface ${table.className}Mapper extends BaseMapperX<${table.className}DO> {
 
     default PageResult<${table.className}DO> selectPage(${table.className}PageReqVO reqVO) {
         return selectPage(reqVO, new QueryWrapperX<${table.className}DO>()
-#foreach ($column in $columns)
-#if (${column.listOperation})
-#set ($JavaField = $column.javaField.substring(0,1).toUpperCase() + ${column.javaField.substring(1)})##首字母大写
-#if (${column.listOperationCondition} == "=")##情况一，= 的时候
-            .eqIfPresent("${column.columnName}", reqVO.get${JavaField}())
-#end
-#if (${column.listOperationCondition} == "!=")##情况二，!= 的时候
-            .neIfPresent("${column.columnName}", reqVO.get${JavaField}())
-#end
-#if (${column.listOperationCondition} == ">")##情况三，> 的时候
-            .gtIfPresent("${column.columnName}", reqVO.get${JavaField}())
-#end
-#if (${column.listOperationCondition} == ">=")##情况四，>= 的时候
-            .geIfPresent("${column.columnName}", reqVO.get${JavaField}())
-#end
-#if (${column.listOperationCondition} == "<")##情况五，< 的时候
-            .gtIfPresent("${column.columnName}", reqVO.get${JavaField}())
-#end
-#if (${column.listOperationCondition} == "LIKE")##情况七，Like 的时候
-            .likeIfPresent("${column.columnName}", reqVO.get${JavaField}())
-#end
-#if (${column.listOperationCondition} == "BETWEEN")##情况八，Between 的时候
-            .betweenIfPresent("${column.columnName}", reqVO.getBegin${JavaField}(), reqVO.getEnd${JavaField}())
-#end
-#end
-#end
+			#listCondition()
+        );
+    }
+
+    default List<${table.className}DO> selectList(${table.className}ExportReqVO reqVO) {
+        return selectList(new QueryWrapperX<${table.className}DO>()
+			#listCondition()
         );
     }
 

src/main/resources/codegen/java/enums/errorcode.vm

@@ -1,2 +1,2 @@
 // ========== ${table.classComment} TODO 补充编号 ==========
-ErrorCode ${simpleClassName_underlineCase.toUpperCase()}_NOT_EXISTS = new ErrorCode(TODO 补充编号, "${table.classComment}不存在}");
+ErrorCode ${simpleClassName_underlineCase.toUpperCase()}_NOT_EXISTS = new ErrorCode(TODO 补充编号, "${table.classComment}不存在");

src/main/resources/codegen/java/service/service.vm

@@ -7,10 +7,10 @@ import ${basePackage}.${table.moduleName}.dal.dataobject.${table.businessName}.$
 import ${PageResultClassName};
 
 /**
-* ${table.classComment} Service 接口
-*
-* @author ${table.author}
-*/
+ * ${table.classComment} Service 接口
+ *
+ * @author ${table.author}
+ */
 public interface ${table.className}Service {
 
     /**
@@ -59,4 +59,12 @@ public interface ${table.className}Service {
      */
     PageResult<${table.className}DO> get${simpleClassName}Page(${table.className}PageReqVO pageReqVO);
 
+    /**
+     * 获得${table.classComment}列表, 用于 Excel 导出
+     *
+     * @param exportReqVO 查询条件
+     * @return ${table.classComment}分页
+     */
+    List<${table.className}DO> get${simpleClassName}List(${table.className}ExportReqVO exportReqVO);
+
 }

src/main/resources/codegen/java/service/serviceImpl.vm

@@ -19,10 +19,10 @@ import ${ServiceExceptionUtilClassName};
 import static ${basePackage}.${table.moduleName}.enums.${simpleModuleName_upperFirst}ErrorCodeConstants.*;
 
 /**
-* ${table.classComment} Service 实现类
-*
-* @author ${table.author}
-*/
+ * ${table.classComment} Service 实现类
+ *
+ * @author ${table.author}
+ */
 @Service
 @Validated
 public class ${table.className}ServiceImpl implements ${table.className}Service {
@@ -77,4 +77,9 @@ public class ${table.className}ServiceImpl implements ${table.className}Service
 		return ${classNameVar}Mapper.selectPage(pageReqVO);
     }
 
+    @Override
+    public List<${table.className}DO> get${simpleClassName}List(${table.className}ExportReqVO exportReqVO) {
+		return ${classNameVar}Mapper.selectList(exportReqVO);
+    }
+
 }