库位导入数据

zss

2023-11-03 22c67fd681856d2b6513d02329b63d76017e1120

 mes-basic/src/main/java/com/chinaztt/mes/basic/excel/LocationUploadListener.java

@@ -5,12 +5,10 @@
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.chinaztt.mes.basic.entity.Warehouse;
import com.chinaztt.mes.basic.service.LocationService;
import com.chinaztt.mes.basic.service.WarehouseService;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @author ZTT
@@ -20,25 +18,13 @@
   List<LocationData> list = new ArrayList<>();

   private LocationService locationService;
   private Map<String, Warehouse> warehouseMap;

   public LocationUploadListener(LocationService locationService, Map<String, Warehouse> warehouseMap) {
   public LocationUploadListener(LocationService locationService) {
      this.locationService = locationService;
      this.warehouseMap = warehouseMap;
   }

   @Override
   public void invoke(LocationData data, AnalysisContext analysisContext) {

      if (StringUtils.isBlank(data.getWarehouseName())) {
         throw new RuntimeException("缺少仓库名称");
      }
      if (null == warehouseMap.get(data.getWarehouseName())) {
         throw new RuntimeException("请填写正确的库位");
      }
      Warehouse warehouse = warehouseMap.get(data.getWarehouseName());
      data.setFactoryId(warehouse.getFactoryId());
      data.setLocGroup(warehouse.getId());
      list.add(data);
      if (list.size() >= BATCH_COUNT) {
         save();