自动化立体仓库 - WMS系统
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
whycq
2024-07-04 09a482e8fb4d4dac63aed1503fd91e738ddb070d 
 src/main/java/com/zy/asrs/controller/MatController.java


@@ -1,6 +1,8 @@


package com.zy.asrs.controller;




import com.alibaba.excel.EasyExcel;


import com.alibaba.excel.ExcelWriter;


import com.alibaba.excel.write.metadata.WriteSheet;


import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;


import com.alibaba.fastjson.JSONArray;


import com.alibaba.fastjson.JSONObject;


@@ -10,11 +12,9 @@


import com.core.annotations.ManagerAuth;


import com.core.common.*;


import com.core.exception.CoolException;


import com.zy.asrs.entity.ManPakOut;


import com.zy.asrs.entity.Mat;


import com.zy.asrs.entity.MatPrint;


import com.zy.asrs.entity.OrderDetl;


import com.zy.asrs.entity.*;


import com.zy.asrs.entity.result.KeyValueVo;


import com.zy.asrs.mapper.TagMapper;


import com.zy.asrs.service.ManPakOutService;


import com.zy.asrs.service.MatService;


import com.zy.asrs.service.OrderDetlService;


@@ -25,19 +25,24 @@


import com.zy.common.utils.BarcodeUtils;


import com.zy.common.utils.QrCode;


import com.zy.common.web.BaseController;


import lombok.extern.slf4j.Slf4j;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.transaction.annotation.Transactional;


import org.springframework.web.bind.annotation.*;


import org.springframework.web.multipart.MultipartFile;




import javax.annotation.Resource;


import javax.imageio.ImageIO;


import javax.servlet.http.HttpServletResponse;


import java.awt.image.BufferedImage;


import java.io.IOException;


import java.io.OutputStream;


import java.net.URLEncoder;


import java.util.*;


import java.util.stream.Collectors;




@RestController


@Slf4j


public class MatController extends BaseController {




    @Autowired


@@ -176,16 +181,72 @@


        }


        return R.ok();


    }


    @Resource


    private TagMapper tagMapper;




    @RequestMapping(value = "/mat/export/auth")


    @ManagerAuth


    public R export(@RequestBody JSONObject param){


        EntityWrapper<Mat> wrapper = new EntityWrapper<>();


        List<String> fields = JSONObject.parseArray(param.getJSONArray("fields").toJSONString(), String.class);


        Map<String, Object> map = excludeTrash(param.getJSONObject("mat"));


        convert(map, wrapper);


        List<Mat> list = matService.selectList(wrapper);


        return R.ok(exportSupport(list, fields));


    public void export(@RequestBody JSONObject param, HttpServletResponse response) throws IOException {


        //文件名


        String fileName =String.valueOf(System.currentTimeMillis());




        OutputStream outputStream =null;


        try {


            //记录总数:实际中需要根据查询条件进行统计即可:一共多少条


            int totalCount=matService.selectCount(null);


            //每一个Sheet存放100w条数据


            Integer sheetDataRows = 400000;


            //每次写入的数据量20w,每页查询20W


            Integer writeDataRows = 200000;


            //计算需要的Sheet数量


            Integer sheetNum = totalCount % sheetDataRows == 0 ? (totalCount / sheetDataRows) : (totalCount / sheetDataRows + 1);


            //计算一般情况下每一个Sheet需要写入的次数(一般情况不包含最后一个sheet,因为最后一个sheet不确定会写入多少条数据)


            Integer oneSheetWriteCount = sheetDataRows / writeDataRows;


            //计算最后一个sheet需要写入的次数


            Integer lastSheetWriteCount = totalCount % sheetDataRows == 0 ? oneSheetWriteCount : (totalCount % sheetDataRows % writeDataRows == 0 ? (totalCount / sheetDataRows / writeDataRows) : ((totalCount % sheetDataRows) / writeDataRows + 1));


            outputStream = response.getOutputStream();


            //必须放到循环外，否则会刷新流


            ExcelWriter excelWriter = EasyExcel.write(outputStream).build();


            //开始分批查询分次写入


            for (int i = 0; i < sheetNum; i++) {


                //创建Sheet


                WriteSheet sheet = new WriteSheet();


                sheet.setSheetName("Sheet"+i);


                sheet.setSheetNo(i);


                //循环写入次数: j的自增条件是当不是最后一个Sheet的时候写入次数为正常的每个Sheet写入的次数,如果是最后一个就需要使用计算的次数lastSheetWriteCount


                for (int j = 0; j < (i != sheetNum - 1 ? oneSheetWriteCount : lastSheetWriteCount); j++) {


                    //分页查询一次20w


                    Page page1 = new Page(j + 1 + oneSheetWriteCount * i, writeDataRows);


                    //查询分页列表---按照自己的业务查列表，分页这个一定要使用这个：page1.getPageNum(),page1.getPageSize()！！！


                    List<Mat> list = matService.selectPage(new Page<>(page1.getCurrent(), page1.getLimit())).getRecords();


                    List<Tag> tags = tagMapper.selectList(null);


                    // 所有类别


                    Map<Long, String> tagMaps = tags.stream().collect(Collectors.toMap(Tag::getId,Tag::getName,(v1,v2) -> v1));


                    // 归类id寻找归类名称


                    list = list.stream().peek(mat -> mat.setUuid(tagMaps.get(mat.getTagId()))).collect(Collectors.toList());


                    //写入到excel:


                    /**************z只需要选择一种方式即可*****************/


                    //这里可以通过设置includeColumnFiledNames、excludeColumnFiledNames导出什么字段，可以动态配置，前端传过来那些列，就导出那些列


                    //方式3、不做设置，全部导出


                    WriteSheet writeSheet = EasyExcel.writerSheet(i, "Sheet" + (i + 1)).head(Mat.class)


                            .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).build();


                    excelWriter.write(list, writeSheet);


                }


            }


            // 下载EXCEL，返回给前段stream流


            response.setContentType("application/octet-stream");


            response.setCharacterEncoding("utf-8");


            response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");


            excelWriter.finish();


            outputStream.flush();


            outputStream.close();


        } catch (Exception e) {


            e.printStackTrace();


        }finally {


            if (outputStream != null) {


                outputStream.close();


            }


        }


    }




    @RequestMapping(value = "/matQuery/auth")