联建活动导出

3 years ago · cc9a49883e
6 changed files with 70 additions and 35 deletions
--- a/epmet-module/epmet-heart/epmet-heart-client/src/main/java/com/epmet/dto/IcPartyActivityDTO.java
+++ b/epmet-module/epmet-heart/epmet-heart-client/src/main/java/com/epmet/dto/IcPartyActivityDTO.java
@ -179,5 +179,9 @@ public class IcPartyActivityDTO implements Serializable {
     */
    @JsonIgnore
 	private Date updatedTime;
-
+    /**
+     * actId不为空小程序
+     * 为空管理平台
+     */
+    private String originName;
 }
--- a/epmet-module/epmet-heart/epmet-heart-client/src/main/java/com/epmet/dto/form/PartyActivityFormDTO.java
+++ b/epmet-module/epmet-heart/epmet-heart-client/src/main/java/com/epmet/dto/form/PartyActivityFormDTO.java
@ -1,5 +1,6 @@
 package com.epmet.dto.form;

+import com.epmet.commons.tools.dto.form.PageFormDTO;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 import org.springframework.format.annotation.DateTimeFormat;
@ -15,7 +16,7 @@ import java.util.Date;
 */
@NoArgsConstructor
@Data
-public class PartyActivityFormDTO implements Serializable {
+public class PartyActivityFormDTO extends PageFormDTO implements Serializable {
    private static final long serialVersionUID = -2510068555703677L;
    @NotBlank(message = "组织Id不能为空" )
    private String agencyId;
@ -27,7 +28,7 @@ public class PartyActivityFormDTO implements Serializable {
    @DateTimeFormat(pattern="yyyy-MM-dd HH:mm:ss")
    private Date endTime;
    private String serviceMatter;
-    private Integer pageNo;
-    private Integer pageSize;
+    // private Integer pageNo;
+    // private Integer pageSize;
    private String customerId;
 }
--- a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/controller/IcPartyActivityController.java
+++ b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/controller/IcPartyActivityController.java
@ -17,15 +17,21 @@

 package com.epmet.controller;

+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.ExcelWriter;
+import com.alibaba.excel.write.metadata.WriteSheet;
 import com.epmet.commons.tools.annotation.LoginUser;
 import com.epmet.commons.tools.aop.NoRepeatSubmit;
+import com.epmet.commons.tools.constant.NumConstant;
 import com.epmet.commons.tools.exception.ExceptionUtils;
 import com.epmet.commons.tools.exception.RenException;
 import com.epmet.commons.tools.page.PageData;
 import com.epmet.commons.tools.security.dto.TokenDto;
+import com.epmet.commons.tools.utils.ConvertUtils;
 import com.epmet.commons.tools.utils.DateUtils;
 import com.epmet.commons.tools.utils.ExcelUtils;
 import com.epmet.commons.tools.utils.Result;
+import com.epmet.commons.tools.utils.poi.excel.handler.FreezeAndFilter;
 import com.epmet.commons.tools.validator.ValidatorUtils;
 import com.epmet.commons.tools.validator.group.AddGroup;
 import com.epmet.commons.tools.validator.group.DefaultGroup;
@ -53,11 +59,9 @@ import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.URLEncoder;
-import java.util.ArrayList;
+import java.util.Date;
 import java.util.List;
 import java.util.concurrent.CompletableFuture;
-import java.util.concurrent.atomic.AtomicInteger;
-import java.util.stream.Collectors;


 /**
@ -116,23 +120,36 @@ public class IcPartyActivityController {
    }

    @PostMapping("export")
-    public void export(@RequestBody PartyActivityFormDTO formDTO, HttpServletResponse response) throws Exception {
-        List<IcPartyActivityDTO> list = icPartyActivityService.list(formDTO);
-        List<IcPartyActivityExcel> excelList = new ArrayList<>();
-        AtomicInteger i = new AtomicInteger(1);
-        if (CollectionUtils.isNotEmpty(list)) {
-            excelList = list.stream().map(item -> {
-                IcPartyActivityExcel excel = new IcPartyActivityExcel();
-                excel.setIndex(i.getAndIncrement());
-                excel.setUnitName(item.getUnitName());
-                excel.setTitle(item.getTitle());
-                excel.setAddress(item.getAddress());
-                excel.setPeopleCount(item.getPeopleCount());
-                excel.setActivityTime(DateUtils.format(item.getActivityTime(), DateUtils.DATE_TIME_PATTERN));
-                return excel;
-            }).collect(Collectors.toList());
+    public void export(@LoginUser TokenDto tokenDto, @RequestBody PartyActivityFormDTO formDTO, HttpServletResponse response) throws Exception {
+        formDTO.setCustomerId(tokenDto.getCustomerId());
+        formDTO.setIsPage(false);
+        ExcelWriter excelWriter = null;
+        formDTO.setPageSize(NumConstant.TEN_THOUSAND);
+        int pageNo = formDTO.getPageNo();
+        try {
+            // 这里 需要指定写用哪个class去写
+            String today= DateUtils.format(new Date(),DateUtils.DATE_PATTERN_MMDD);
+            String fileName = "联建活动".concat(today);
+            excelWriter = EasyExcel.write(ExcelUtils.getOutputStreamForExcel(fileName, response), IcPartyActivityExcel.class).build();
+            WriteSheet writeSheet = EasyExcel.writerSheet("Sheet1").registerWriteHandler(new FreezeAndFilter()).build();
+            PageData<IcPartyActivityDTO> data = null;
+            List<IcPartyActivityExcel> list = null;
+            do {
+                data = icPartyActivityService.search(formDTO);
+                list = ConvertUtils.sourceToTarget(data.getList(), IcPartyActivityExcel.class);
+                formDTO.setPageNo(++pageNo);
+                excelWriter.write(list, writeSheet);
+            } while (CollectionUtils.isNotEmpty(data.getList()) && data.getList().size() == formDTO.getPageSize());
+
+        } catch (Exception e) {
+            log.error("export exception", e);
+        } finally {
+            // 千万别忘记finish 会帮忙关闭流
+            if (excelWriter != null) {
+                excelWriter.finish();
+            }
        }
-        ExcelUtils.exportExcelToTarget(response, null, excelList, IcPartyActivityExcel.class);
+
    }

    /**
--- a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/controller/IcPartyUnitController.java
+++ b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/controller/IcPartyUnitController.java
@ -132,7 +132,7 @@ public class IcPartyUnitController {
        try {
            // 这里 需要指定写用哪个class去写
            String today= DateUtils.format(new Date(),DateUtils.DATE_PATTERN_MMDD);
-            String fileName = "行程上报信息".concat(today);
+            String fileName = "联建单位".concat(today);
            excelWriter = EasyExcel.write(ExcelUtils.getOutputStreamForExcel(fileName, response), IcPartyUnitExcel.class).build();
            WriteSheet writeSheet = EasyExcel.writerSheet("Sheet1").registerWriteHandler(new FreezeAndFilter()).build();
            PageData<IcPartyUnitDTO> data = null;
--- a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/excel/IcPartyActivityExcel.java
+++ b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/excel/IcPartyActivityExcel.java
@ -17,7 +17,8 @@

 package com.epmet.excel;

-import cn.afterturn.easypoi.excel.annotation.Excel;
+import com.alibaba.excel.annotation.ExcelProperty;
+import com.alibaba.excel.annotation.write.style.ColumnWidth;
 import lombok.Data;

 /**
@ -28,23 +29,30 @@ import lombok.Data;
 */
@Data
 public class IcPartyActivityExcel {
-
-    @Excel(name = "序号")
-    private Integer index;
-
-    @Excel(name = "单位名称")
+    @ColumnWidth(20)
+    @ExcelProperty(value = "单位名称")
    private String unitName;

-    @Excel(name = "活动标题")
+    @ColumnWidth(25)
+    @ExcelProperty(value = "活动标题")
    private String title;

-    @Excel(name = "活动地址")
+    @ColumnWidth(35)
+    @ExcelProperty(value = "活动地址")
    private String address;

-    @Excel(name = "服务人数")
+    @ColumnWidth(15)
+    @ExcelProperty(value = "服务人数")
    private Integer peopleCount;

-    @Excel(name = "活动时间")
+    @ColumnWidth(20)
+    @ExcelProperty(value = "活动时间")
    private String activityTime;
-
+    /**
+     * actId不为空小程序
+     * 为空管理平台
+     */
+    @ColumnWidth(20)
+    @ExcelProperty(value = "来源")
+    private String originName;
 }
--- a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/service/impl/IcPartyActivityServiceImpl.java
+++ b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/service/impl/IcPartyActivityServiceImpl.java
@ -151,6 +151,11 @@ public class IcPartyActivityServiceImpl extends BaseServiceImpl<IcPartyActivityD
                    dto.setServiceMatterNameList(serviceMatterNameList);
                    dto.setServiceMatterName(StringUtils.join(serviceMatterNameList, StrConstant.COMMA));
                }
+                if(StringUtils.isNotBlank(dto.getActId())){
+                    dto.setOriginName("小程序");
+                }else{
+                    dto.setOriginName("管理平台");
+                }
            });
        }