drange 导出

src/main/java/com/huiju/estateagents/drainage/controller/TaDrainageController.java

@@ -47,13 +47,18 @@ public class TaDrainageController extends BaseController {
      */
     @RequestMapping(value="/admin/taDrainage",method= RequestMethod.GET)
     public ResponseBean taDrainageList(@RequestParam(value ="pageNum",defaultValue = "1") Integer pageNum,
-                                       @RequestParam(value ="pageSize",defaultValue = "10") Integer pageSize, HttpServletRequest request){
+                                       @RequestParam(value ="pageSize",defaultValue = "10") Integer pageSize,
+                                       @RequestParam(value ="drainageId",defaultValue = "10") Integer drainageId,
+                                       @RequestParam(value ="name",defaultValue = "10") String name,
+                                       HttpServletRequest request){
         ResponseBean responseBean = new ResponseBean();
         try {
             //使用分页插件
 		    IPage<TaDrainage> pg = new Page<>(pageNum, pageSize);
             QueryWrapper<TaDrainage> queryWrapper = new QueryWrapper<>();
             queryWrapper.eq("org_id",getOrgId(request));
+            queryWrapper.eq("drainage_id",drainageId);
+            queryWrapper.like("name",name);
             queryWrapper.orderByDesc("create_date");
 
             IPage<TaDrainage> result = iTaDrainageService.page(pg, queryWrapper);

src/main/java/com/huiju/estateagents/drainage/controller/TaDrainageRecordController.java

@@ -1,13 +1,21 @@
 package com.huiju.estateagents.drainage.controller;
 
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.ExcelWriter;
+import com.alibaba.excel.write.metadata.WriteSheet;
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.google.gson.JsonObject;
 import com.huiju.estateagents.base.BaseController;
 import com.huiju.estateagents.base.ResponseBean;
 import com.huiju.estateagents.common.StringUtils;
 import com.huiju.estateagents.drainage.entity.TaDrainageRecord;
 import com.huiju.estateagents.drainage.service.ITaDrainageRecordService;
+import com.huiju.estateagents.excel.ReporRecommendCustomer;
+import com.huiju.estateagents.excel.handler.CustomCellWriteHandler;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -20,6 +28,12 @@ import org.springframework.web.bind.annotation.ResponseBody;
 import org.springframework.web.bind.annotation.RestController;
 
 import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.time.format.DateTimeFormatter;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
 
 /**
  * <p>
@@ -72,6 +86,68 @@ public class TaDrainageRecordController extends BaseController {
         }
         return responseBean;
     }
+    
+    
+    /**
+     * 导出数据
+     * @return
+     */
+    @RequestMapping(value="/admin/taDrainageRecord/export",method= RequestMethod.GET)
+    public void taDrainageRecordList(@RequestParam(value = "drainageId") Integer drainageId,
+                                             @RequestParam(value = "startTime",required = false) String startTime,
+                                             @RequestParam(value = "endTime",required = false) String endTime,
+                                             HttpServletRequest request, HttpServletResponse response) throws IOException {
+            //使用分页插件
+            QueryWrapper<TaDrainageRecord> queryWrapper = new QueryWrapper<>();
+            queryWrapper.eq("drainage_id",drainageId);
+            queryWrapper.eq("org_id",getOrgId(request));
+            queryWrapper.gt(!StringUtils.isEmpty(startTime),"create_date",startTime);
+            queryWrapper.lt(!StringUtils.isEmpty(endTime),"create_date",endTime);
+            queryWrapper.orderByDesc("create_date");
+    
+            List<TaDrainageRecord> list = iTaDrainageRecordService.list(queryWrapper);
+            List<List<String>> resList = new ArrayList<List<String>>();
+            
+            //构造表头
+            if (list.size() > 0) {
+                List<String> headerList = new ArrayList<>();
+                TaDrainageRecord taDrainageRecord = list.get(0);
+                headerList.add("提交时间");
+                JSONArray arr= JSONArray.parseArray(taDrainageRecord.getContent());
+                arr.forEach(e -> {
+                    JSONObject jsonObject = JSONObject.parseObject(e.toString());
+                    headerList.add(jsonObject.getString("label"));
+                });
+                resList.add(headerList);
+            }
+            
+            //构造表数据
+            DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm");
+            list.forEach(e -> {
+                List<String> bodyList = new ArrayList<>();
+                bodyList.add(e.getCreateDate().format(formatter));
+                JSONArray arr= JSONArray.parseArray(e.getContent());
+                arr.forEach(json -> {
+                    JSONObject jsonObject = JSONObject.parseObject(json.toString());
+                    bodyList.add(jsonObject.getString("value"));
+                });
+                resList.add(bodyList);
+            });
+    
+            response.setContentType("application/octet-stream");
+            response.setCharacterEncoding("utf-8");
+            response.setHeader("Content-disposition", "attachment;filename=引流详情.xlsx");
+            // 设置读取的class
+            ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()).registerWriteHandler(new CustomCellWriteHandler()).build();
+            // 设置 sheet, 同一个sheet只需要设置一次
+            WriteSheet writeSheet = EasyExcel.writerSheet("引流详情").build();
+            
+            
+            excelWriter.write(resList, writeSheet);
+    
+            /// finish 会帮忙关闭流
+            excelWriter.finish();
+    }
 
     /**
      * 保存对象