报价：价格表导入

@@ -7,9 +7,12 @@ import com.lframework.starter.web.core.components.resp.InvokeResult;
 import com.lframework.starter.web.core.components.resp.InvokeResultBuilder;
 import com.lframework.starter.web.core.components.resp.PageResult;
 import com.lframework.starter.web.core.controller.DefaultBaseController;
+import com.lframework.starter.web.core.utils.ExcelUtil;
 import com.lframework.starter.web.core.utils.PageResultUtil;
 import com.lframework.xingyun.basedata.bo.price.QueryPriceBo;
 import com.lframework.xingyun.basedata.entity.Price;
+import com.lframework.xingyun.basedata.excel.price.PriceImportListener;
+import com.lframework.xingyun.basedata.excel.price.PriceImportModel;
 import com.lframework.xingyun.basedata.service.price.PriceService;
 import com.lframework.xingyun.basedata.vo.price.CreatePriceVo;
 import com.lframework.xingyun.basedata.vo.price.QueryPriceVo;
@@ -22,8 +25,12 @@ import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 import javax.validation.Valid;
 import javax.validation.constraints.NotBlank;
+import javax.validation.constraints.NotNull;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
 import java.util.List;
 import java.util.stream.Collectors;
+import org.springframework.web.multipart.MultipartFile;
 
 /**
  * 价格表 Controller
@@ -116,4 +123,30 @@ public class PriceController extends DefaultBaseController {
 
         return InvokeResultBuilder.success();
     }
+
+    /**
+     * 下载导入模板
+     */
+    @ApiOperation("下载导入模板")
+    @HasPermission({"base-data:price-table:import"})
+    @GetMapping("/import/template")
+    public void downloadImportTemplate(HttpServletResponse response) throws IOException {
+        ExcelUtil.exportXls("价格表导入模板", PriceImportModel.class);
+    }
+
+    /**
+     * 导入
+     */
+    @ApiOperation("导入")
+    @HasPermission({"base-data:price-table:import"})
+    @PostMapping("/import")
+    public InvokeResult<Void> importExcel(@NotBlank(message = "ID不能为空") String id,
+                                          @NotNull(message = "请上传文件") MultipartFile file) {
+
+        PriceImportListener listener = new PriceImportListener();
+        listener.setTaskId(id);
+        ExcelUtil.read(file, PriceImportModel.class, listener).sheet().doRead();
+
+        return InvokeResultBuilder.success();
+    }
 }

+package com.lframework.xingyun.basedata.excel.price;
+
+import com.alibaba.excel.context.AnalysisContext;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.lframework.starter.common.exceptions.impl.DefaultClientException;
+import com.lframework.starter.common.utils.StringUtil;
+import com.lframework.starter.web.core.components.excel.ExcelImportListener;
+import com.lframework.starter.web.core.utils.ApplicationUtil;
+import com.lframework.starter.web.core.utils.IdUtil;
+import com.lframework.xingyun.basedata.entity.BreedRelationship;
+import com.lframework.xingyun.basedata.entity.Price;
+import com.lframework.xingyun.basedata.service.breed.BreedRelationshipService;
+import com.lframework.xingyun.basedata.service.price.PriceService;
+import com.lframework.xingyun.basedata.vo.price.CreatePriceSubTableVo;
+import com.lframework.xingyun.basedata.vo.price.CreatePriceVo;
+import java.math.BigDecimal;
+import java.util.ArrayList;
+import java.util.List;
+import lombok.extern.slf4j.Slf4j;
+
+/**
+ * 价格表导入监听器
+ */
+@Slf4j
+public class PriceImportListener extends ExcelImportListener<PriceImportModel> {
+
+    /**
+     * 导入文件中的价格表编号
+     */
+    private String importCode;
+
+    /**
+     * 编号所在行号
+     */
+    private Integer codeRowIndex;
+
+    /**
+     * 导入文件中的品名校验列表
+     */
+    private final List<String> checkProductNameList = new ArrayList<>();
+
+    @Override
+    protected void doInvoke(PriceImportModel data, AnalysisContext context) {
+        int rowIndex = context.readRowHolder().getRowIndex();
+        PriceService priceService = ApplicationUtil.getBean(PriceService.class);
+        BreedRelationshipService breedRelationshipService = ApplicationUtil.getBean(BreedRelationshipService.class);
+
+        if (StringUtil.isNotBlank(data.getCode())) {
+            data.setCode(data.getCode().trim());
+        }
+        if (StringUtil.isNotBlank(data.getProductName())) {
+            data.setProductName(data.getProductName().trim());
+        }
+
+        validateCode(data.getCode(), rowIndex, priceService);
+
+        if (StringUtil.isBlank(data.getProductName())) {
+            throw new DefaultClientException("第" + rowIndex + "行“品名”不能为空");
+        }
+
+        BreedRelationship breedRelationship = breedRelationshipService.getOne(
+                Wrappers.lambdaQuery(BreedRelationship.class)
+                        .eq(BreedRelationship::getProductName, data.getProductName())
+                        .last("LIMIT 1"));
+        if (breedRelationship == null) {
+            throw new DefaultClientException("第" + rowIndex + "行“品名”不存在于品种关系中");
+        }
+        data.setProductId(breedRelationship.getId());
+
+        if (checkProductNameList.contains(data.getProductName())) {
+            throw new DefaultClientException(
+                    "第" + rowIndex + "行“品名”与第" + (checkProductNameList.indexOf(data.getProductName()) + 1) + "行重复");
+        }
+        checkProductNameList.add(data.getProductName());
+
+        data.setPriceValue(parsePrice(data.getPrice(), rowIndex));
+    }
+
+    @Override
+    protected void afterAllAnalysed(AnalysisContext context) {
+        PriceService priceService = ApplicationUtil.getBean(PriceService.class);
+
+        List<PriceImportModel> datas = this.getDatas();
+        if (datas.isEmpty()) {
+            return;
+        }
+
+        CreatePriceVo vo = new CreatePriceVo();
+        vo.setCode(importCode);
+
+        List<CreatePriceSubTableVo> priceSubTableList = new ArrayList<>(datas.size());
+        for (PriceImportModel data : datas) {
+            CreatePriceSubTableVo subTableVo = new CreatePriceSubTableVo();
+            subTableVo.setProductId(data.getProductId());
+            subTableVo.setPrice(data.getPriceValue());
+            subTableVo.setItemId(IdUtil.getId());
+            priceSubTableList.add(subTableVo);
+        }
+        vo.setPriceSubTableList(priceSubTableList);
+
+        priceService.create(vo);
+        for (int i = 0; i < datas.size(); i++) {
+            this.setSuccessProcess(i);
+        }
+    }
+
+    @Override
+    protected void doComplete() {
+
+    }
+
+    /**
+     * 校验导入编号
+     */
+    private void validateCode(String code, int rowIndex, PriceService priceService) {
+        if (importCode == null) {
+            if (StringUtil.isBlank(code)) {
+                throw new DefaultClientException("第" + rowIndex + "行“编号”不能为空");
+            }
+
+            Price existsPrice = priceService.getOne(Wrappers.lambdaQuery(Price.class)
+                    .eq(Price::getCode, code)
+                    .last("LIMIT 1"));
+            if (existsPrice != null) {
+                throw new DefaultClientException("第" + rowIndex + "行“编号”已存在");
+            }
+
+            importCode = code;
+            codeRowIndex = rowIndex;
+            return;
+        }
+
+        if (StringUtil.isNotBlank(code)) {
+            throw new DefaultClientException(
+                    "第" + rowIndex + "行“编号”重复填写，请仅在第" + codeRowIndex + "行填写一次");
+        }
+    }
+
+    /**
+     * 解析采购价，并按统一口径返回行级错误
+     */
+    private BigDecimal parsePrice(String value, int rowIndex) {
+        if (StringUtil.isBlank(value)) {
+            throw new DefaultClientException("第" + rowIndex + "行“采购价”不能为空");
+        }
+
+        BigDecimal decimalValue;
+        try {
+            decimalValue = new BigDecimal(value.trim());
+        } catch (NumberFormatException e) {
+            throw new DefaultClientException("第" + rowIndex + "行“采购价”格式有误");
+        }
+
+        BigDecimal normalizedValue = decimalValue.stripTrailingZeros();
+        int scale = Math.max(normalizedValue.scale(), 0);
+        if (scale > 10) {
+            throw new DefaultClientException("第" + rowIndex + "行“采购价”最多允许10位小数");
+        }
+
+        return decimalValue;
+    }
+}

+package com.lframework.xingyun.basedata.excel.price;
+
+import com.alibaba.excel.annotation.ExcelIgnore;
+import com.alibaba.excel.annotation.ExcelProperty;
+import com.lframework.starter.web.core.components.excel.ExcelModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+
+/**
+ * 价格表导入模型
+ */
+@Data
+public class PriceImportModel implements ExcelModel {
+
+    /**
+     * 编号
+     */
+    @ApiModelProperty("编号")
+    @ExcelProperty("编号")
+    private String code;
+
+    /**
+     * 品名
+     */
+    @ApiModelProperty("品名")
+    @ExcelProperty("品名")
+    private String productName;
+
+    /**
+     * 采购价
+     */
+    @ApiModelProperty("采购价")
+    @ExcelProperty("采购价")
+    private String price;
+
+    /**
+     * 导入时解析出的品种ID
+     */
+    @ExcelIgnore
+    private String productId;
+
+    /**
+     * 导入时解析出的采购价
+     */
+    @ExcelIgnore
+    private java.math.BigDecimal priceValue;
+}