组织导入数据

zss

2023-11-03 99d9153a8c33ce32b898c7d9098038caa3447206

 mes-basic/src/main/java/com/chinaztt/mes/basic/controller/StaffController.java

@@ -17,12 +17,17 @@

package com.chinaztt.mes.basic.controller;

import com.alibaba.excel.EasyExcel;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.chinaztt.mes.basic.entity.Division;
import com.chinaztt.mes.basic.entity.Factory;
import com.chinaztt.mes.basic.entity.Location;
import com.chinaztt.mes.basic.entity.Staff;
import com.chinaztt.mes.basic.excel.DivisionData;
import com.chinaztt.mes.basic.excel.DivisionUploadListener;
import com.chinaztt.mes.basic.excel.StaffData;
import com.chinaztt.mes.basic.excel.StaffUploadListener;
import com.chinaztt.mes.basic.service.DivisionService;
import com.chinaztt.mes.basic.service.StaffService;
import com.chinaztt.mes.common.wrapper.QueryWrapperUtil;
@@ -34,7 +39,9 @@
import lombok.AllArgsConstructor;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.List;


@@ -168,4 +175,19 @@
      return R.ok(staffService.list(Wrappers.query(staff)));
   }

   /**
    * excel上传
    *
    * @return
    */
   @PostMapping("/upload")
   public R upload(@RequestParam("file") MultipartFile file) {
      try {
         EasyExcel.read(file.getInputStream(), StaffData.class, new StaffUploadListener(staffService)).sheet().doRead();
      } catch (IOException e) {
         e.printStackTrace();
      }
      return R.ok();
   }

}