wms-serve-mule/src/main/java/com/wms/controller/ExcelController.java

package com.wms.controller;

import com.alibaba.fastjson2.JSON;
import com.wms.constants.enums.KateTaskStatus;
import com.wms.constants.enums.ResponseCode;
import com.wms.entity.app.ResponseEntity;
import com.wms.entity.table.*;
import com.wms.service.*;
import com.wms.utils.HttpUtils;
import com.wms.utils.StringUtils;
import com.wms.utils.WmsUtils;
import com.wms.utils.excel.ExcelUtils;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.transaction.annotation.Isolation;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.interceptor.TransactionAspectSupport;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.util.*;

/**
 *
 */
@Controller
@CrossOrigin
@RequiredArgsConstructor(onConstructor = @__(@Autowired))
@RequestMapping(value = "/wms/excel")
public class ExcelController extends BaseController {
    private final StockService stockService;// 库存服务
    private final PartInfoService partInfoService;// 零件服务
    private final HttpServletRequest servletRequest;// 请求服务
    private final TaskRecordService taskRecordService;// 任务记录服务

    /**
     * 导入零件信息
     *
     * @param file 文件
     * @return 导入结果
     */
    @PostMapping("/uploadPartInfos")
    @ResponseBody
    @Transactional(isolation = Isolation.REPEATABLE_READ, propagation = Propagation.REQUIRED)
    public String uploadPartInfos(@RequestPart("file") MultipartFile file) {
        logger.info("接收到导入零件信息请求，ip地址：{}", HttpUtils.getIpAddr(servletRequest));
        ResponseEntity response = new ResponseEntity();
        try {
            List<PartInfo> files = ExcelUtils.readMultipartFile(file, PartInfo.class);
            // 添加进物料表
            for (PartInfo pageInfo : files) {
                if (partInfoService.selPartByPartNo(pageInfo.getMaterial()) != null) {// 当前零件号的数据已经存在过
                    partInfoService.modifyPart(pageInfo);
                } else {// 新零件
                    partInfoService.addPart(pageInfo);
                }
            }
            response.setCode(ResponseCode.OK.getCode());
            response.setMessage("导入excel成功");
            response.setReturnData(files);
        } catch (Exception e) {
            // 回滚事务
            TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
            response.setCode(ResponseCode.ERROR.getCode());
            response.setMessage(e.getMessage());
        }
        return JSON.toJSONString(response);
    }

    /**
     * 导出库存数据
     *
     * @param response 请求
     */
    @GetMapping("/downloadStockExcel")
    @ResponseBody
    public void downloadStockExcel(HttpServletResponse response) {
        List<Stock> stocks = stockService.selStocks(new Stock());
        ExcelUtils.export(response, "库存报表", stocks, Stock.class);
    }

    /**
     * 导出入库记录
     *
     * @param response 请求
     */
    @GetMapping("/downloadRukuExcel")
    @ResponseBody
    public void downloadRukuExcel(HttpServletResponse response) {
        List<Task> ruku = taskRecordService.selTasks(new Task());
        ExcelUtils.export(response, "入库记录报表", ruku, Task.class);
    }

    /**
     * 导出物料信息
     *
     * @param response 请求
     */
    @GetMapping("/downloadMaterialExcel")
    @ResponseBody
    public void downloadMaterialExcel(HttpServletResponse response) {
        List<PartInfo> Material = partInfoService.selParts(new PartInfo());
        ExcelUtils.export(response, "物料信息", Material, PartInfo.class);
    }

}