多人同時(shí)導(dǎo)出 Excel 干崩服務(wù)器！阿里新來大佬給出的解決方案太優(yōu)雅了！

@Slf4j
@Component
public class ExportQueue {


    private final int MAX_CAPACITY = 10; // 隊(duì)列最大容量
    private LinkedList<ExportUser> queue; // 用戶隊(duì)列

    public ExportQueue(LinkedList<ExportUser> queue) {
        this.queue = new LinkedList<>();
    }

    /**
     * 排隊(duì)隊(duì)列添加
     * @param sysUser
     */
    public synchronized LinkedList<ExportUser> add(ExportUser sysUser) {
        while (queue.size() >= MAX_CAPACITY) {
            try {
                log.info("當(dāng)前排隊(duì)人已滿，請等待");
                wait();
            } catch (InterruptedException e) {
                e.getMessage();
            }
        }
        queue.add(sysUser);
        log.info("目前導(dǎo)出隊(duì)列排隊(duì)人數(shù)：" + queue.size());
        notifyAll();
        return queue;
    }


    /**
     * 獲取排隊(duì)隊(duì)列下一個(gè)人
     * @return
     */
    public synchronized ExportUser getNextSysUser() {
        while (queue.isEmpty()) {
            try {
                wait();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        ExportUser sysUser = queue.remove();
        notifyAll(); //喚醒
        return sysUser;
    }
}

@Slf4j
public abstract class AbstractExport<T, K> {


    public abstract void export(ExportUser sysUser) throws InterruptedException;

    /**
     * 導(dǎo)出
     *
     * @param response 輸出流
     * @param pageSize 每頁大小
     * @param t 導(dǎo)出條件
     * @param k Excel內(nèi)容實(shí)體類
     * @param fileName 文件名稱
     */
    public void export(HttpServletResponse response, int pageSize, T t, Class<K> k, String fileName) throws Exception {
        ExcelWriter writer = null;
        try {
            writer = getExcelWriter(response, fileName);
            //查詢導(dǎo)出總條數(shù)
            int total = this.countExport(t);
            //頁數(shù)
            int loopCount = PageUtil.totalPage(total, pageSize);
            BeanUtil.setProperty(t, "pageSize", pageSize);
            for (int i = 0; i < loopCount; i++) {
                //開始頁
                BeanUtil.setProperty(t, "pageNum", PageUtil.getStart(i + 1, pageSize));
                //獲取Excel導(dǎo)出信息
                List<K> kList = this.getExportDetail(t);
                WriteSheet writeSheet = EasyExcel.writerSheet(fileName).head(k).build();
                writer.write(kList, writeSheet);
            }
        } catch (Exception e) {
            String msg = "導(dǎo)出" + fileName + "異常";
            log.error(msg, e);
            throw new Exception(msg + e);
        } finally {
            if (writer != null) {
                writer.finish();
            }
        }
    }

    public com.alibaba.excel.ExcelWriter getExcelWriter(HttpServletResponse response, String fileName) throws IOException {
        response.setContentType("application/vnd.ms-excel");
        response.setCharacterEncoding("utf-8");
        // 這里URLEncoder.encode可以防止中文亂碼 當(dāng)然和easyexcel沒有關(guān)系
        String fileNameUtf = URLEncoder.encode(fileName, "UTF-8").replaceAll("\\+", "%20");
        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileNameUtf + ".xlsx");
        return EasyExcel.write(response.getOutputStream()).build();
    }


    /**
     * （模版導(dǎo)出）
     *
     * @param t
     * @param fileName
     * @param response
     */
    public abstract void complexFillWithTable(T t, String fileName, HttpServletResponse response);

    /**
     * 查詢導(dǎo)出總條數(shù)
     *
     * @param t
     * @return
     */
    public abstract int countExport(T t);

    /**
     * 查詢導(dǎo)出數(shù)據(jù)
     *
     * @param t
     * @return
     */
    public abstract List<K> getExportDetail(T t);
}

@Service
@Slf4j
public class ExportImpl extends AbstractExport {

    @Autowired
    private ExportQueue exportQueue;


    @Override
    public void export(ExportUser sysUser) throws InterruptedException {

        //導(dǎo)出
        log.info("導(dǎo)出文件方法執(zhí)行～～～～～～～～～");
// export(response,pageSize,t,k,fileName);
        LinkedList<ExportUser> queue = exportQueue.add(sysUser);
        log.info("導(dǎo)出隊(duì)列：" + queue);
        //休眠時(shí)間稍微設(shè)置大點(diǎn)，模擬導(dǎo)出處理時(shí)間
        Thread.sleep(20000);
        //導(dǎo)出成功后移除當(dāng)前導(dǎo)出用戶
        ExportUser nextSysUser = exportQueue.getNextSysUser();
        log.info("移除后獲取下一個(gè)排隊(duì)的用戶: " + nextSysUser.getUserName());

    }


    @Override
    public void export(HttpServletResponse response, int pageSize, Object o, Class k, String fileName) throws Exception {
        super.export(response, pageSize, o, k, fileName);
    }

    @Override
    public ExcelWriter getExcelWriter(HttpServletResponse response, String fileName) throws IOException {
        return super.getExcelWriter(response, fileName);
    }

    @Override
    public void complexFillWithTable(Object o, String fileName, HttpServletResponse response) {

    }

    @Override
    public int countExport(Object o) {
        return 0;
    }

    @Override
    public List getExportDetail(Object o) {
        return null;
    }
}

測試Controller

@RestController
@RequestMapping("/export")
@Slf4j
public class ExportController {

    @Autowired
    private ExportImpl export;


    @PostMapping("/exportFile")
    public void exportFile() {
            new Thread(new Runnable() {
                @SneakyThrows
                @Override
                public void run() {
                    Thread thread1 = Thread.currentThread();
                    ExportUser sysUser =new ExportUser();
                    sysUser.setUserName(thread1.getName());

                    export.export(sysUser);
                }
            }).start();
        }
}

測試結(jié)果