楚江erp：客户简称导入相关接口开发

@@ -2,8 +2,11 @@ package com.lframework.xingyun.basedata.controller;
 
 import com.lframework.starter.web.core.annotations.security.HasPermission;
 import com.lframework.starter.web.core.controller.DefaultBaseController;
+import com.lframework.starter.web.core.utils.ExcelUtil;
 import com.lframework.xingyun.basedata.bo.customer.GetCustomerShortBo;
 import com.lframework.xingyun.basedata.bo.customer.QueryCustomerShortBo;
+import com.lframework.xingyun.basedata.excel.customer.CustomerShortImportListener;
+import com.lframework.xingyun.basedata.excel.customer.CustomerShortModel;
 import com.lframework.xingyun.basedata.vo.customer.QueryCustomerShortVo;
 import com.lframework.xingyun.basedata.service.customer.CustomerShortService;
 import com.lframework.xingyun.basedata.vo.customer.CreateCustomerShortVo;
@@ -24,8 +27,10 @@ import io.swagger.annotations.Api;
 import org.springframework.web.bind.annotation.DeleteMapping;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
 
 import javax.validation.Valid;
+import javax.validation.constraints.NotNull;
 import java.util.List;
 import java.util.stream.Collectors;
 
@@ -50,7 +55,7 @@ public class CustomerShortController extends DefaultBaseController {
      * 查询列表
      */
     @ApiOperation("查询列表")
-    @HasPermission({"customer:customer:query"})
+    @HasPermission({"base-data:custome-abbreviation:query"})
     @GetMapping("/query")
     public InvokeResult<PageResult<QueryCustomerShortBo>> query(@Valid QueryCustomerShortVo vo) {
         PageResult<CustomerShort> pageResult = customerShortService.query(getPageIndex(vo), getPageSize(vo), vo);
@@ -67,7 +72,7 @@ public class CustomerShortController extends DefaultBaseController {
      */
     @ApiOperation("根据ID查询")
     @ApiImplicitParam(value = "id", name = "id", paramType = "query", required = true)
-    @HasPermission({"customer:customer:query"})
+    @HasPermission({"base-data:custome-abbreviation:query"})
     @GetMapping
     public InvokeResult<GetCustomerShortBo> get(@NotBlank(message = "id不能为空！") String id) {
         CustomerShort data = customerShortService.findById(id);
@@ -83,7 +88,7 @@ public class CustomerShortController extends DefaultBaseController {
      * 新增
      */
     @ApiOperation("新增")
-    @HasPermission({"customer:customer:add"})
+    @HasPermission({"base-data:custome-abbreviation:add"})
     @PostMapping
     public InvokeResult<Void> create(@Valid CreateCustomerShortVo vo) {
         customerShortService.create(vo);
@@ -94,7 +99,7 @@ public class CustomerShortController extends DefaultBaseController {
      * 修改
      */
     @ApiOperation("修改")
-    @HasPermission({"customer:customer:modify"})
+    @HasPermission({"base-data:custome-abbreviation:modify"})
     @PutMapping
     public InvokeResult<Void> update(@Valid UpdateCustomerShortVo vo) {
         customerShortService.update(vo);
@@ -106,10 +111,38 @@ public class CustomerShortController extends DefaultBaseController {
      */
     @ApiOperation("根据ID删除")
     @ApiImplicitParam(value = "id", name = "id", paramType = "query", required = true)
-    @HasPermission({"customer:customer:delete"})
+    @HasPermission({"base-data:custome-abbreviation:delete"})
     @DeleteMapping
     public InvokeResult<Void> deleteById(@NotBlank(message = "id不能为空！") String id) {
         customerShortService.deleteById(id);
         return InvokeResultBuilder.success();
     }
+
+    /**
+     * 下载导入模板
+     */
+    @ApiOperation("下载导入模板")
+    @HasPermission({"base-data:custome-abbreviation:import"})
+    @GetMapping("/import/template")
+    public void downloadImportTemplate() {
+        ExcelUtil.exportXls("客户简称导入模板", CustomerShortModel.class);
+    }
+
+    /**
+     * 导入
+     *
+     * @param id        任务ID
+     * @param file      文件
+     */
+    @ApiOperation("导入")
+    @HasPermission({"base-data:custome-abbreviation:import"})
+    @PostMapping("/import")
+    public InvokeResult<Void> importExcel(@NotBlank(message = "ID不能为空") String id,
+                                          @NotNull(message = "请上传文件") MultipartFile file) {
+        CustomerShortImportListener listener = new CustomerShortImportListener();
+        listener.setTaskId(id);
+        ExcelUtil.read(file, CustomerShortModel.class, listener).sheet().doRead();
+
+        return InvokeResultBuilder.success();
+    }
 }

+package com.lframework.xingyun.basedata.excel.customer;
+
+import com.alibaba.excel.context.AnalysisContext;
+import com.baomidou.mybatisplus.core.conditions.Wrapper;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.lframework.starter.common.exceptions.impl.DefaultClientException;
+import com.lframework.starter.web.core.components.excel.ExcelImportListener;
+import com.lframework.starter.web.core.utils.ApplicationUtil;
+import com.lframework.starter.web.core.utils.JsonUtil;
+import com.lframework.starter.web.inner.entity.SysDept;
+import com.lframework.starter.web.inner.service.system.SysDeptService;
+import com.lframework.xingyun.basedata.entity.Customer;
+import com.lframework.xingyun.basedata.entity.CustomerShort;
+import com.lframework.xingyun.basedata.service.customer.CustomerService;
+import com.lframework.xingyun.basedata.service.customer.CustomerShortService;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.collections4.CollectionUtils;
+
+import java.util.List;
+import java.util.stream.Collectors;
+
+@Slf4j
+public class CustomerShortImportListener extends ExcelImportListener<CustomerShortModel> {
+
+    @Override
+    protected void doInvoke(CustomerShortModel data, AnalysisContext context) {
+        // 客户名称
+        CustomerService customerService = ApplicationUtil.getBean(CustomerService.class);
+        Wrapper<Customer> customerWrapper = Wrappers.lambdaQuery(Customer.class)
+                .eq(Customer::getName, data.getCustomerName());
+        Customer customer = customerService.getOne(customerWrapper);
+        if (customer == null) {
+            throw new DefaultClientException("第" + context.readRowHolder().getRowIndex() + "行，客户[" + data.getCustomerName()
+                    + "]不存在，请检查后再次导入！");
+        }
+        data.setCustomerId(customer.getId());
+        // 简称
+        CustomerShortService customerShortService = ApplicationUtil.getBean(CustomerShortService.class);
+        CustomerShort customerShort = customerShortService.getByCustomerId(customer.getId());
+        if (customerShort != null) {
+            throw new DefaultClientException("第" + context.readRowHolder().getRowIndex() + "行，客户[" + data.getCustomerName()
+                    + "]简称已存在，请检查后再次导入！");
+        }
+        // 客户类型
+        String type = data.getType();
+        if ("终端".equals(type)) {
+            type = "TERMINAL";
+        } else if ("经销".equals(type)) {
+            type = "DEALER";
+        } else {
+            throw new DefaultClientException("第" + context.readRowHolder().getRowIndex() + "行，客户类型只能填写“终端”或“经销”，请检查后再次导入！");
+        }
+        data.setType(type);
+        // 办事处
+        SysDeptService sysDeptService = ApplicationUtil.getBean(SysDeptService.class);
+        LambdaQueryWrapper<SysDept> deptWrapper = Wrappers.lambdaQuery(SysDept.class);
+        deptWrapper.eq(SysDept::getName, data.getDeptName());
+        SysDept office = sysDeptService.getOne(deptWrapper);
+        if (office == null) {
+            throw new DefaultClientException("第" + context.readRowHolder().getRowIndex() + "行，[" + data.getDeptName()
+                    + "]不存在，请检查后再次导入！");
+        }
+        data.setDeptId(office.getId());
+        // 区域
+        deptWrapper.eq(SysDept::getName, data.getRegionName());
+        SysDept region = sysDeptService.getOne(deptWrapper);
+        if (region == null) {
+            throw new DefaultClientException("第" + context.readRowHolder().getRowIndex() + "行，[" + data.getRegionName()
+                    + "]不存在，请检查后再次导入！");
+        }
+        List<SysDept> regionList = sysDeptService.queryArea(null, office.getId());
+        if (CollectionUtils.isEmpty(regionList)) {
+            throw new DefaultClientException("第" + context.readRowHolder().getRowIndex() + "行，[" + data.getRegionName()
+                    + "]不属于[" + data.getDeptName() + "]，请检查后再次导入！");
+        }
+        List<String> regionNameList = regionList.stream().map(SysDept::getName).collect(Collectors.toList());
+        if (regionNameList.contains(region.getName())) {
+            throw new DefaultClientException("第" + context.readRowHolder().getRowIndex() + "行，[" + data.getRegionName()
+                    + "]不属于[" + data.getDeptName() + "]，请检查后再次导入！");
+        }
+        data.setRegion(region.getId());
+    }
+
+    @Override
+    protected void afterAllAnalysed(AnalysisContext context) {
+        CustomerShortService customerShortService = ApplicationUtil.getBean(CustomerShortService.class);
+        for (int i = 0; i < datas.size(); i++) {
+            CustomerShortModel data = datas.get(i);
+            CustomerShort customerShort = JsonUtil.parseObject(JsonUtil.toJsonString(data), CustomerShort.class);
+            try {
+                customerShortService.save(customerShort);
+            } catch (Exception e) {
+                throw new DefaultClientException(
+                        "第" + (i + 1) + "行新增失败，失败原因：" + e.getMessage());
+            }
+            this.setSuccessProcess(i);
+        }
+    }
+
+    @Override
+    protected void doComplete() {
+
+    }
+}

+package com.lframework.xingyun.basedata.excel.customer;
+
+import com.alibaba.excel.annotation.ExcelIgnore;
+import com.alibaba.excel.annotation.ExcelProperty;
+import com.lframework.starter.web.core.annotations.excel.ExcelRequired;
+import com.lframework.starter.web.core.components.excel.ExcelModel;
+import lombok.Data;
+
+import javax.validation.constraints.Pattern;
+
+/**
+ * <p>
+ * 客户简称
+ * </p>
+ *
+ */
+@Data
+public class CustomerShortModel implements ExcelModel {
+
+    /**
+     * 客户简称
+     */
+    @ExcelRequired
+    @ExcelProperty("客户简称")
+    private String shortName;
+
+    /**
+     * 客户ID
+     */
+    @ExcelIgnore
+    private String customerId;
+
+    /**
+     * 客户名称
+     */
+    @ExcelRequired
+    @ExcelProperty("客户名称")
+    private String customerName;
+
+    /**
+     * 客户类型
+     */
+    @ExcelRequired
+    @ExcelProperty("客户类型")
+    @Pattern(regexp = "终端|经销", message = "客户类型只能填写“终端”或“经销”")
+    private String type;
+
+    /**
+     * 办事处ID
+     */
+    @ExcelIgnore
+    private String deptId;
+
+    /**
+     * 办事处名称
+     */
+    @ExcelRequired
+    @ExcelProperty("办事处名称")
+    private String deptName;
+
+    /**
+     * 区域ID
+     */
+    @ExcelIgnore
+    private String region;
+
+    /**
+     * 区域名称
+     */
+    @ExcelRequired
+    @ExcelProperty("区域名称")
+    private String regionName;
+}