楚江ERP-客户开发异步导出功能开发

 package com.lframework.xingyun.sc.bo.customer.develop;
 
 import com.fasterxml.jackson.annotation.JsonFormat;
+import com.lframework.starter.web.core.dto.BaseDto;
 import com.lframework.xingyun.basedata.entity.*;
 import com.lframework.xingyun.sc.entity.CustomerDevelopPlan;
 import com.lframework.starter.common.constants.StringPool;
@@ -18,7 +19,7 @@ import lombok.Data;
  *
  */
 @Data
-public class GetCustomerDevelopPlanBo extends BaseBo<CustomerDevelopPlan> {
+public class GetCustomerDevelopPlanBo extends BaseBo<CustomerDevelopPlan> implements BaseDto {
 
     /**
      * ID

@@ -3,6 +3,7 @@ package com.lframework.xingyun.sc.controller.customer;
 import com.lframework.starter.bpm.dto.FlowTaskDto;
 import com.lframework.starter.bpm.mappers.FlowTaskWrapperMapper;
 import com.lframework.starter.bpm.vo.flow.task.QueryTodoTaskListVo;
+import com.lframework.starter.mq.core.utils.ExportTaskUtil;
 import com.lframework.starter.web.core.annotations.security.HasPermission;
 import com.lframework.starter.web.core.components.security.SecurityUtil;
 import com.lframework.starter.web.core.controller.DefaultBaseController;
@@ -12,9 +13,9 @@ import com.lframework.xingyun.basedata.entity.*;
 import com.lframework.xingyun.basedata.service.customer.CustomerService;
 import com.lframework.xingyun.basedata.service.office.OfficeService;
 import com.lframework.xingyun.basedata.service.product.ProductVarietyService;
-import com.lframework.xingyun.basedata.service.type.TypeInfoService;
 import com.lframework.xingyun.basedata.service.workshop.WorkshopService;
 import com.lframework.xingyun.sc.bo.customer.develop.GetCustomerDevelopPlanBo;
+import com.lframework.xingyun.sc.excel.customer.CustomerDevelopExportTaskWorker;
 import com.lframework.xingyun.sc.vo.customer.develop.QueryCustomerDevelopPlanVo;
 import com.lframework.xingyun.sc.service.customer.CustomerDevelopPlanService;
 import com.lframework.xingyun.sc.vo.customer.develop.CreateCustomerDevelopPlanVo;
@@ -152,6 +153,17 @@ public class CustomerDevelopPlanController extends DefaultBaseController {
         return InvokeResultBuilder.success();
     }
 
+    /**
+     * 导出
+     */
+    @ApiOperation("导出")
+    @HasPermission({"customer:develop:export"})
+    @PostMapping("/export")
+    public InvokeResult<Void> export(@Valid QueryCustomerDevelopPlanVo vo) {
+        ExportTaskUtil.exportTask("客户开发信息", CustomerDevelopExportTaskWorker.class, vo);
+        return InvokeResultBuilder.success();
+    }
+
 
     /**
      * 封装客户开发数据

@@ -26,4 +26,19 @@ public enum CustomerDevelopStatus implements BaseEnum<Integer> {
     public String getDesc() {
         return this.desc;
     }
+
+
+    /**
+     * 通过code获取desc
+     * @param code 状态码
+     * @return 描述信息
+     */
+    public static String getDescByCode(Integer code) {
+        for (CustomerDevelopStatus status : CustomerDevelopStatus.values()) {
+            if (status.getCode().equals(code)) {
+                return status.getDesc();
+            }
+        }
+        return null;
+    }
 }

+package com.lframework.xingyun.sc.excel.customer;
+
+import com.alibaba.excel.annotation.ExcelProperty;
+import com.alibaba.excel.annotation.format.DateTimeFormat;
+import com.lframework.starter.common.constants.StringPool;
+import com.lframework.starter.web.core.bo.BaseBo;
+import com.lframework.starter.web.core.components.excel.ExcelModel;
+import com.lframework.xingyun.sc.bo.customer.develop.GetCustomerDevelopPlanBo;
+import com.lframework.xingyun.sc.enums.CustomerDevelopStatus;
+import lombok.Data;
+
+
+@Data
+public class CustomerDevelopExportModel extends BaseBo<GetCustomerDevelopPlanBo> implements ExcelModel {
+
+
+    /**
+     * 客户基本信息
+     */
+    @ExcelProperty("客户名称")
+    private String customerName;
+
+    /**
+     * 生产厂
+     */
+    @ExcelProperty("生产厂")
+    private String workshopName;
+
+    /**
+     * 科办
+     */
+    @ExcelProperty("科办")
+    private String officeName;
+
+    /**
+     * 客户类型
+     */
+    @ExcelProperty("客户类型")
+    private String customerTypeName;
+
+    /**
+     * 产品品种
+     */
+    @ExcelProperty("产品品种")
+    private String productVarietyName;
+
+    /**
+     * 月用量
+     */
+    @ExcelProperty("月用量")
+    private Double monthlyUsage;
+
+    /**
+     * 目标量
+     */
+    @ExcelProperty("目标量")
+    private Double targetQuantity;
+
+    /**
+     * 行业
+     */
+    @ExcelProperty("行业")
+    private String industry;
+
+    /**
+     * 牌号
+     */
+    @ExcelProperty("牌号")
+    private String mark;
+
+    /**
+     * 厚度
+     */
+    @ExcelProperty("厚度")
+    private Double thickness;
+
+    /**
+     * 宽度
+     */
+    @ExcelProperty("宽度")
+    private Double width;
+
+    /**
+     * 材质要求
+     */
+    @ExcelProperty("材质要求")
+    private String materialRequire;
+
+    /**
+     * 品质要求
+     */
+    @ExcelProperty("品质要求")
+    private String qualityRequire;
+
+    /**
+     * 同行
+     */
+    @ExcelProperty("同行")
+    private String peer;
+
+    /**
+     * 核价模式
+     */
+    @ExcelProperty("核价模式")
+    private String pricingMode;
+
+    /**
+     * 结算天数
+     */
+    @ExcelProperty("结算天数")
+    private Double settleDays;
+
+    /**
+     * 责任人
+     */
+    @ExcelProperty("责任人")
+    private String chargeUserName;
+
+    /**
+     * 状态
+     */
+    @ExcelProperty("状态")
+    private String statusName;
+
+    /**
+     * 创建人
+     */
+    @ExcelProperty("创建人")
+    private String createBy;
+
+    /**
+     * 更新人
+     */
+    @ExcelProperty("更新人")
+    private String updateBy;
+
+    /**
+     * 创建时间
+     */
+    @ExcelProperty("创建时间")
+    @DateTimeFormat(StringPool.DATE_TIME_PATTERN)
+    private String createTime;
+
+    /**
+     * 更新时间
+     */
+    @ExcelProperty("更新时间")
+    @DateTimeFormat(StringPool.DATE_TIME_PATTERN)
+    private String updateTime;
+
+
+    public CustomerDevelopExportModel() {
+
+    }
+
+    public CustomerDevelopExportModel(GetCustomerDevelopPlanBo dto) {
+        super(dto);
+    }
+
+    @Override
+    public BaseBo<GetCustomerDevelopPlanBo> convert(GetCustomerDevelopPlanBo dto) {
+        return super.convert(dto);
+    }
+
+    @Override
+    protected void afterInit(GetCustomerDevelopPlanBo dto) {
+        if (dto.getCustomer() != null) {
+            this.customerName = dto.getCustomer().getName();
+        }
+        if (dto.getWorkshop() != null) {
+            this.workshopName = dto.getWorkshop().getName();
+        }
+        if (dto.getOffice() != null) {
+            this.officeName = dto.getOffice().getName();
+        }
+        if (dto.getProductVariety() != null) {
+            this.productVarietyName = dto.getProductVariety().getName();
+        }
+        if (dto.getStatus() != null) {
+            this.statusName = CustomerDevelopStatus.getDescByCode(dto.getStatus());
+        }
+        if (dto.getCustomerType() != null) {
+            switch (dto.getCustomerType()) {
+                case "DEALER":
+                    this.customerTypeName = "经销商";
+                    break;
+                case "TERMINAL":
+                    this.customerTypeName = "终端";
+                    break;
+                default:
+                    break;
+            }
+        }
+    }
+}

+package com.lframework.xingyun.sc.excel.customer;
+
+import com.lframework.starter.common.utils.CollectionUtil;
+import com.lframework.starter.web.core.components.resp.PageResult;
+import com.lframework.starter.web.core.utils.ApplicationUtil;
+import com.lframework.starter.web.core.utils.JsonUtil;
+import com.lframework.starter.mq.core.components.export.ExportTaskWorker;
+import com.lframework.starter.web.core.utils.PageResultUtil;
+import com.lframework.starter.web.inner.entity.SysUser;
+import com.lframework.starter.web.inner.service.system.SysUserService;
+import com.lframework.xingyun.basedata.entity.Customer;
+import com.lframework.xingyun.basedata.entity.Office;
+import com.lframework.xingyun.basedata.entity.ProductVariety;
+import com.lframework.xingyun.basedata.entity.Workshop;
+import com.lframework.xingyun.basedata.service.customer.CustomerService;
+import com.lframework.xingyun.basedata.service.office.OfficeService;
+import com.lframework.xingyun.basedata.service.product.ProductVarietyService;
+import com.lframework.xingyun.basedata.service.workshop.WorkshopService;
+import com.lframework.xingyun.sc.bo.customer.develop.GetCustomerDevelopPlanBo;
+import com.lframework.xingyun.sc.entity.CustomerDevelopPlan;
+import com.lframework.xingyun.sc.service.customer.CustomerDevelopPlanService;
+import com.lframework.xingyun.sc.vo.customer.develop.QueryCustomerDevelopPlanVo;
+import org.apache.commons.collections4.CollectionUtils;
+import org.apache.commons.lang3.StringUtils;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.function.Function;
+import java.util.stream.Collectors;
+
+public class CustomerDevelopExportTaskWorker implements
+        ExportTaskWorker<QueryCustomerDevelopPlanVo, GetCustomerDevelopPlanBo, CustomerDevelopExportModel> {
+
+
+    @Override
+    public QueryCustomerDevelopPlanVo parseParams(String json) {
+        return JsonUtil.parseObject(json, QueryCustomerDevelopPlanVo.class);
+    }
+
+    @Override
+    public PageResult<GetCustomerDevelopPlanBo> getDataList(int pageIndex, int pageSize, QueryCustomerDevelopPlanVo params) {
+        CustomerDevelopPlanService customerDevelopPlanService = ApplicationUtil.getBean(CustomerDevelopPlanService.class);
+        PageResult<CustomerDevelopPlan> pageResult = customerDevelopPlanService.query(pageIndex, pageSize, params);
+        List<CustomerDevelopPlan> dataList = pageResult.getDatas();
+        List<GetCustomerDevelopPlanBo> results = null;
+        if (!CollectionUtil.isEmpty(dataList)) {
+            results = this.packCustomerDevelopData(dataList);
+        }
+        return PageResultUtil.rebuild(pageResult, results);
+    }
+
+    @Override
+    public CustomerDevelopExportModel exportData(GetCustomerDevelopPlanBo data) {
+        return new CustomerDevelopExportModel(data);
+    }
+
+    @Override
+    public Class<CustomerDevelopExportModel> getModelClass() {
+        return CustomerDevelopExportModel.class;
+    }
+
+
+    /**
+     * 封装客户开发数据
+     *
+     * @param dataList  数据集
+     * @return  List<GetCustomerDevelopPlanBo>
+     */
+    private List<GetCustomerDevelopPlanBo> packCustomerDevelopData(List<CustomerDevelopPlan> dataList) {
+        // 处理关联数据
+        List<String> customerIds = new ArrayList<>();
+        List<String> workshopIds = new ArrayList<>();
+        List<String> officeIds = new ArrayList<>();
+        List<String> productVarietyIds = new ArrayList<>();
+        List<String> chargeUserIds = new ArrayList<>();
+        for (CustomerDevelopPlan plan : dataList) {
+            String customerId = plan.getCustomerId();
+            String workshopId = plan.getWorkshopId();
+            String officeId = plan.getOfficeId();
+            String productVarietyId = plan.getProductVarietyId();
+            String chargeUserId = plan.getChargeUserId();
+            if (StringUtils.isNotBlank(customerId) && !customerIds.contains(customerId)) {
+                customerIds.add(customerId);
+            }
+            if (StringUtils.isNotBlank(workshopId) && !workshopIds.contains(workshopId)) {
+                workshopIds.add(workshopId);
+            }
+            if (StringUtils.isNotBlank(officeId) && !officeIds.contains(officeId)) {
+                officeIds.add(officeId);
+            }
+            if (StringUtils.isNotBlank(productVarietyId) && !productVarietyIds.contains(productVarietyId)) {
+                productVarietyIds.add(productVarietyId);
+            }
+            if (StringUtils.isNotBlank(chargeUserId) && !chargeUserIds.contains(chargeUserId)) {
+                chargeUserIds.add(chargeUserId);
+            }
+        }
+        // 获取客户基本信息
+        CustomerService customerService = ApplicationUtil.getBean(CustomerService.class);
+        List<Customer> customerList = customerService.listByIds(customerIds);
+        Map<String, Customer> customerMap = new HashMap<>();
+        if (CollectionUtils.isNotEmpty(customerList)) {
+            customerMap = customerList.stream().collect(Collectors.toMap(Customer::getId, Function.identity()));
+        }
+        // 获取厂房信息
+        WorkshopService workshopService = ApplicationUtil.getBean(WorkshopService.class);
+        List<Workshop> workshopList = workshopService.listByIds(workshopIds);
+        Map<String, Workshop> workshopMap = new HashMap<>();
+        if (CollectionUtils.isNotEmpty(workshopList)) {
+            workshopMap = workshopList.stream().collect(Collectors.toMap(Workshop::getId, Function.identity()));
+        }
+        // 获取办事处/科办信息
+        OfficeService officeService = ApplicationUtil.getBean(OfficeService.class);
+        List<Office> officeList = officeService.listByIds(officeIds);
+        Map<String, Office> officeMap = new HashMap<>();
+        if (CollectionUtils.isNotEmpty(officeList)) {
+            officeMap = officeList.stream().collect(Collectors.toMap(Office::getId, Function.identity()));
+        }
+        // 获取产品品种信息
+        ProductVarietyService productVarietyService = ApplicationUtil.getBean(ProductVarietyService.class);
+        List<ProductVariety> productVarietyList = productVarietyService.listByIds(productVarietyIds);
+        Map<String, ProductVariety> productVarietyMap = new HashMap<>();
+        if (CollectionUtils.isNotEmpty(productVarietyList)) {
+            productVarietyMap = productVarietyList.stream().collect(Collectors.toMap(ProductVariety::getId, Function.identity()));
+        }
+        // 获取责任人数据
+        SysUserService sysUserService = ApplicationUtil.getBean(SysUserService.class);
+        List<SysUser> userList = sysUserService.listByIds(chargeUserIds);
+        Map<String, SysUser> userMap = new HashMap<>();
+        if (CollectionUtils.isNotEmpty(userList)) {
+            userMap = userList.stream().collect(Collectors.toMap(SysUser::getId, Function.identity()));
+        }
+        List<GetCustomerDevelopPlanBo> result = new ArrayList<>();
+        for (CustomerDevelopPlan plan : dataList) {
+            GetCustomerDevelopPlanBo data = new GetCustomerDevelopPlanBo(plan);
+            // 客户信息
+            Customer customer = customerMap.get(plan.getCustomerId());
+            data.setCustomer(customer);
+            // 厂房信息
+            Workshop workshop = workshopMap.get(plan.getWorkshopId());
+            data.setWorkshop(workshop);
+            // 办事处/科办
+            Office office = officeMap.get(plan.getOfficeId());
+            data.setOffice(office);
+            // 产品品种
+            ProductVariety productVariety = productVarietyMap.get(plan.getProductVarietyId());
+            data.setProductVariety(productVariety);
+            // 责任人
+            SysUser user = userMap.get(plan.getChargeUserId());
+            data.setChargeUserName(user == null ? null : user.getName());
+
+            result.add(data);
+        }
+        return result;
+    }
+}

@@ -72,4 +72,10 @@ public class QueryCustomerDevelopPlanVo extends SortPageVo implements BaseVo, Se
      */
     @ApiModelProperty("待办类型")
     private String todoType;
+
+    /**
+     * 导出类型
+     */
+    @ApiModelProperty("导出类型")
+    private String exportType = "CUSTOMER_DEVELOP";
 }