报价：品种关系导入

@@ -3,13 +3,17 @@ package com.lframework.xingyun.basedata.controller;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 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.starter.web.core.utils.PageResultUtil;
 import com.lframework.starter.web.core.components.resp.PageResult;
 import com.lframework.starter.web.core.components.resp.InvokeResult;
+import javax.servlet.http.HttpServletResponse;
 import javax.validation.constraints.NotBlank;
 import com.lframework.xingyun.basedata.bo.breed.QueryBreedRelationshipBo;
 import com.lframework.xingyun.basedata.entity.BreedRelationship;
 import com.lframework.xingyun.basedata.entity.PriceSubTable;
+import com.lframework.xingyun.basedata.excel.breed.BreedRelationshipImportListener;
+import com.lframework.xingyun.basedata.excel.breed.BreedRelationshipImportModel;
 import com.lframework.xingyun.basedata.service.breed.BreedRelationshipService;
 import com.lframework.xingyun.basedata.service.price.PriceSubTableService;
 import com.lframework.xingyun.basedata.vo.breed.CreateBreedRelationshipVo;
@@ -25,7 +29,10 @@ import org.springframework.web.bind.annotation.DeleteMapping;
 import org.springframework.beans.factory.annotation.Autowired;
 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.io.IOException;
 import java.util.Collections;
 import java.util.HashSet;
 import java.util.List;
@@ -129,6 +136,26 @@ public class BreedRelationshipController extends DefaultBaseController {
         return InvokeResultBuilder.success();
     }
 
+    @ApiOperation("下载导入模板")
+    @HasPermission({"base-data:variety:import"})
+    @GetMapping("/import/template")
+    public void downloadImportTemplate(HttpServletResponse response) throws IOException {
+        ExcelUtil.exportXls("品种关系导入模板", BreedRelationshipImportModel.class);
+    }
+
+    @ApiOperation("导入")
+    @HasPermission({"base-data:variety:import"})
+    @PostMapping("/import")
+    public InvokeResult<Void> importExcel(@NotBlank(message = "ID不能为空") String id,
+                                          @NotNull(message = "请上传文件") MultipartFile file) {
+
+        BreedRelationshipImportListener listener = new BreedRelationshipImportListener();
+        listener.setTaskId(id);
+        ExcelUtil.read(file, BreedRelationshipImportModel.class, listener).sheet().doRead();
+
+        return InvokeResultBuilder.success();
+    }
+
     private void fillCanOperate(List<QueryBreedRelationshipBo> results) {
         if (CollectionUtil.isEmpty(results)) {
             return;

+package com.lframework.xingyun.basedata.excel.breed;
+
+import com.alibaba.excel.context.AnalysisContext;
+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.xingyun.basedata.service.breed.BreedRelationshipService;
+import com.lframework.xingyun.basedata.vo.breed.CreateBreedRelationshipVo;
+import lombok.extern.slf4j.Slf4j;
+import java.math.BigDecimal;
+import java.util.ArrayList;
+import java.util.List;
+
+@Slf4j
+public class BreedRelationshipImportListener extends ExcelImportListener<BreedRelationshipImportModel> {
+
+  private List<String> checkList = new ArrayList<>();
+
+  @Override
+  protected void doInvoke(BreedRelationshipImportModel data, AnalysisContext context) {
+    BreedRelationshipService breedRelationshipService = ApplicationUtil.getBean(BreedRelationshipService.class);
+    if (StringUtil.isBlank(data.getProductName())) {
+      throw new DefaultClientException(
+              "第" + context.readRowHolder().getRowIndex() + "行“品种”不能为空");
+    } else {
+      Integer integer = breedRelationshipService.checkProductName(data.getProductName());
+      if (integer > 0) {
+        throw new DefaultClientException(
+                "第" + context.readRowHolder().getRowIndex() + "行“品种”已存在");
+      }
+    }
+    data.setAttritionRateValue(parseRate(data.getAttritionRate(), "损耗率", context));
+    data.setMiscellaneousTaxRateValue(parseRate(data.getMiscellaneousTaxRate(), "杂率", context));
+    if (checkList.contains(data.getProductName())) {
+      throw new DefaultClientException(
+              "第" + context.readRowHolder().getRowIndex() + "行“品种”与第" + (checkList.indexOf(data.getProductName()) + 1) + "行重复");
+    }
+    checkList.add(data.getProductName());
+  }
+
+  @Override
+  protected void afterAllAnalysed(AnalysisContext context) {
+    BreedRelationshipService breedRelationshipService = ApplicationUtil.getBean(BreedRelationshipService.class);
+    List<BreedRelationshipImportModel> datas = this.getDatas();
+    for (int i = 0; i < datas.size(); i++) {
+      BreedRelationshipImportModel data = datas.get(i);
+      CreateBreedRelationshipVo vo = new CreateBreedRelationshipVo();
+      vo.setProductName(data.getProductName());
+      vo.setCode(data.getCode());
+      vo.setType(data.getType());
+      vo.setAttritionRate(data.getAttritionRateValue());
+      vo.setMiscellaneousTaxRate(data.getMiscellaneousTaxRateValue());
+      breedRelationshipService.create(vo);
+      this.setSuccessProcess(i);
+    }
+  }
+
+  @Override
+  protected void doComplete() {
+  }
+
+  /**
+   * 解析导入的比例字段，并按统一口径返回行级错误
+   */
+  private BigDecimal parseRate(String value, String fieldName, AnalysisContext context) {
+    if (StringUtil.isBlank(value)) {
+      return null;
+    }
+
+    BigDecimal decimalValue;
+    try {
+      decimalValue = new BigDecimal(value.trim());
+    } catch (NumberFormatException e) {
+      throw new DefaultClientException(
+              "第" + context.readRowHolder().getRowIndex() + "行“" + fieldName + "”格式有误");
+    }
+
+    BigDecimal normalizedValue = decimalValue.stripTrailingZeros();
+    int scale = Math.max(normalizedValue.scale(), 0);
+    if (scale > 10) {
+      throw new DefaultClientException(
+              "第" + context.readRowHolder().getRowIndex() + "行“" + fieldName + "”最多允许10位小数");
+    }
+    return decimalValue;
+  }
+}

+package com.lframework.xingyun.basedata.excel.breed;
+
+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;
+
+@Data
+public class BreedRelationshipImportModel implements ExcelModel {
+
+    @ExcelIgnore
+    private String id;
+
+    /**
+     * 品种
+     */
+    @ExcelRequired
+    @ExcelProperty("品种")
+    private String productName;
+
+    /**
+     * 代号
+     */
+    @ExcelProperty("代号")
+    private String code;
+
+    /**
+     * 类型
+     */
+    @ExcelProperty("类型")
+    private String type;
+
+    /**
+     * 损耗率
+     */
+    @ExcelProperty("损耗率")
+    private String attritionRate;
+
+    /**
+     * 杂率
+     */
+    @ExcelProperty("杂率")
+    private String miscellaneousTaxRate;
+
+    /**
+     * 损耗率转换后的值
+     */
+    @ExcelIgnore
+    private java.math.BigDecimal attritionRateValue;
+
+    /**
+     * 杂率转换后的值
+     */
+    @ExcelIgnore
+    private java.math.BigDecimal miscellaneousTaxRateValue;
+
+}

@@ -129,6 +129,17 @@ public class BreedRelationshipServiceImpl extends BaseMpServiceImpl<BreedRelatio
         }
     }
 
+    /**
+     * 校验品名是否已存在
+     *
+     * @param productName 品名
+     */
+    public Integer checkProductName(String productName) {
+        LambdaQueryWrapper<BreedRelationship> queryWrapper = Wrappers.lambdaQuery(BreedRelationship.class)
+                        .eq(BreedRelationship::getProductName, productName);
+        return getBaseMapper().selectCount(queryWrapper);
+    }
+
     @Override
     public void cleanCacheByKey(Serializable key) {
 

@@ -52,4 +52,6 @@ public interface BreedRelationshipService extends BaseMpService<BreedRelationshi
      * @return
      */
     void deleteById(String id);
+
+    Integer checkProductName(String productName);
 }