产品结构导入数据

zss

2023-11-07 3c34085a21511610477f555cebf3f6fa42272402

 mes-technology/src/main/java/com/chinaztt/mes/technology/controller/StructureController.java

@@ -19,14 +19,14 @@

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelReader;
import com.alibaba.excel.read.builder.ExcelReaderBuilder;
import com.alibaba.excel.read.metadata.ReadSheet;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.chinaztt.mes.basic.util.DictUtils;
import com.chinaztt.mes.common.wrapper.QueryWrapperUtil;
import com.chinaztt.mes.technology.excel.*;
import com.chinaztt.mes.technology.dto.StructureDTO;
import com.chinaztt.mes.technology.entity.StructureComponent;
import com.chinaztt.mes.technology.excel.StructureData;
import com.chinaztt.mes.technology.excel.StructureUploadListener;
import com.chinaztt.mes.technology.service.StructureService;
import com.chinaztt.ztt.admin.api.entity.SysDictItem;
import com.chinaztt.ztt.admin.api.feign.RemoteDictService;
@@ -41,7 +41,6 @@

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;


@@ -71,15 +70,10 @@
   public R uploadExt(@RequestParam("file") MultipartFile file) {
      ExcelReader excelReader = null;
      try {
         ExcelReaderBuilder excelReaderBuilder = EasyExcel.read(file.getInputStream());
         excelReader = excelReaderBuilder.build();
         List<ReadSheet> sheets = excelReader.excelExecutor().sheetList();
         List<ReadSheet> readSheetList = new ArrayList<>();
         for (ReadSheet sheet : sheets) {
            ReadSheet readSheet = EasyExcel.readSheet(sheet.getSheetName()).registerReadListener(new StructureUploadExtListener(structureService)).build();
            readSheetList.add(readSheet);
         }
         excelReader.read(readSheetList);
         excelReader = EasyExcel.read(file.getInputStream(), StructureData.class, new StructureUploadListener(structureService)).build();
         ReadSheet readSheet = EasyExcel.readSheet(0).build();
         ReadSheet readSheet2 = EasyExcel.readSheet(1).build();
         excelReader.read(readSheet,readSheet2);
      } catch (IOException e) {
         e.printStackTrace();
         return R.failed(e.getMessage());
@@ -96,26 +90,27 @@
    * 竖着导入
    *
    * @param file
    * @return
    */
   @PostMapping("/upload")
   public R simpleRead(@RequestParam("file") MultipartFile file) {
      ExcelReader excelReader = null;
      try {
         excelReader = EasyExcel.read(file.getInputStream(), StructureData.class, new StructureUploadListener(structureService, dictUtils)).build();
         ReadSheet readSheet = EasyExcel.readSheet(0).build();
         excelReader.read(readSheet);
      } catch (IOException e) {
         e.printStackTrace();
         return R.failed(e.getMessage());
      } finally {
         if (excelReader != null) {
            // 这里千万别忘记关闭，读的时候会创建临时文件，到时磁盘会崩的
            excelReader.finish();
         }
      }
      return R.ok();
   }
//    * @return
//    */
//   @PostMapping("/upload")
//   public R simpleRead(@RequestParam("file") MultipartFile file) {
//      ExcelReader excelReader = null;
//      try {
//         excelReader = EasyExcel.read(file.getInputStream(), StructureData.class, new StructureUploadListener(structureService)).build();
//         ReadSheet readSheet = EasyExcel.readSheet(0).build();
//         ReadSheet readSheet2 = EasyExcel.readSheet(1).build();
//         excelReader.read(readSheet,readSheet2);
//      } catch (IOException e) {
//         e.printStackTrace();
//         return R.failed(e.getMessage());
//      } finally {
//         if (excelReader != null) {
//            // 这里千万别忘记关闭，读的时候会创建临时文件，到时磁盘会崩的
//            excelReader.finish();
//         }
//      }
//      return R.ok();
//   }

   /**
    * 根据零件号查询默认BOM
@@ -195,6 +190,7 @@
   @PostMapping
   @PreAuthorize("@pms.hasPermission('technology_structure_add')")
   public R save(@RequestBody StructureDTO structure) {
      System.out.println("===========================>"+structure);
      return structureService.fullSave(structure);
   }