第二张报表导出0915

chengf

2025-09-15 ebf6117ccea680eae81a91f2fa355ce3a76eb621

 service-job/src/main/java/com/java110/job/export/ExportDataExecutor.java

@@ -1,49 +1,77 @@
package com.java110.job.export;

import com.java110.core.client.*;
import com.java110.core.client.FileUploadTemplate;
import com.java110.core.log.LoggerFactory;
import com.java110.dto.data.ExportDataDto;
import com.java110.utils.cache.MappingCache;
import com.java110.dto.user.UserDownloadFileDto;
import com.java110.intf.job.IUserDownloadFileV1InnerServiceSMO;
import com.java110.po.user.UserDownloadFilePo;
import com.java110.utils.factory.ApplicationContextFactory;
import com.java110.utils.util.COSUtil;
import com.java110.utils.util.OSSUtil;
import com.java110.utils.util.ExceptionUtil;
import com.java110.utils.util.StringUtil;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
import org.apache.poi.xwpf.usermodel.XWPFDocument;
import org.slf4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.util.List;
import java.io.IOException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * 导出数据执行器
 */
public class ExportDataExecutor implements Runnable{
public class ExportDataExecutor implements Runnable {
    private static final Logger log = LoggerFactory.getLogger(ExportDataQueue.class);

    private IUserDownloadFileV1InnerServiceSMO userDownloadFileV1InnerServiceSMOImpl;

    private FileUploadTemplate fileUploadTemplate;

    //默认线程大小
    private static final int DEFAULT_EXPORT_POOL = 4;

    private boolean isRun = false;

    public ExportDataExecutor(boolean isRun) {

        this.isRun = isRun;
    }

    public ExportDataExecutor() {
    }

    @Override
    public void run() {

        try {
            doExportData();
        }catch (Exception e){
            log.error("处理消息异常",e);
            e.printStackTrace();
        while (isRun) {
            log.debug("导出数据线程开始处理");
            try {
                doExportData();
            } catch (Throwable e) {
                log.error("处理消息异常", e);
                e.printStackTrace();
            }
            log.debug("导出数据线程处理完成");

        }

    }

    private void doExportData() throws Exception {

        ExportDataDto exportDataDto =  ExportDataQueue.getData();
        ExportDataDto exportDataDto = ExportDataQueue.getData();

        if(exportDataDto == null){
        this.userDownloadFileV1InnerServiceSMOImpl = ApplicationContextFactory.getBean("userDownloadFileV1InnerServiceSMOImpl", IUserDownloadFileV1InnerServiceSMO.class);
        this.fileUploadTemplate = ApplicationContextFactory.getBean("fileUploadTemplate", FileUploadTemplate.class);

        ByteArrayInputStream inputStream = null;
        ByteArrayOutputStream os = null;
        Object document = null;
        String fileName = "";

        if (exportDataDto == null) {
            return;
        }

@@ -51,50 +79,83 @@

        IExportDataAdapt exportDataAdaptImpl = ApplicationContextFactory.getBean(businessAdapt, IExportDataAdapt.class);

        if(exportDataAdaptImpl == null){
            return ;
        if (exportDataAdaptImpl == null) {
            return;
        }
        SXSSFWorkbook workbook= exportDataAdaptImpl.exportData(exportDataDto);
        updateUserDownloadFile(exportDataDto, UserDownloadFileDto.STATE_DOING,"", "开始下载");

        //保存文件路径到 文件下载表
        FileUploadTemplate fileUploadTemplate = ApplicationContextFactory.getBean("fileUploadTemplate", FileUploadTemplate.class);


        ByteArrayInputStream inputStream = null;
        ByteArrayOutputStream os = null;
        try {

            document = exportDataAdaptImpl.exportData(exportDataDto);
            //保存文件路径到 文件下载表
//            os = new ByteArrayOutputStream();
//            workbook.write(os);
//            inputStream = new ByteArrayInputStream(os.toByteArray());
//
//            fileName = fileUploadTemplate.saveFile(inputStream, exportDataDto.getFileName());
//
//
//            updateUserDownloadFile(exportDataDto, UserDownloadFileDto.STATE_FINISH,fileName, "下载完成");
            // 根据文档类型，写入流
            os = new ByteArrayOutputStream();
            workbook.write(os);
            if (document instanceof SXSSFWorkbook) {
                ((SXSSFWorkbook) document).write(os);
            } else if (document instanceof XWPFDocument) {
                ((XWPFDocument) document).write(os);
            }
            inputStream = new ByteArrayInputStream(os.toByteArray());
            fileName = fileUploadTemplate.saveFile(inputStream, exportDataDto.getFileName() + "." + exportDataAdaptImpl.getFileType());

            fileUploadTemplate.saveFile(inputStream,exportDataDto.getFileName());

        }finally {
            try {
                workbook.close();
            }catch (Exception e){

            updateUserDownloadFile(exportDataDto, UserDownloadFileDto.STATE_FINISH, fileName, "下载完成");
        } catch (Throwable e) {
            e.printStackTrace();
            updateUserDownloadFile(exportDataDto, UserDownloadFileDto.STATE_FAIL, "","下载失败" + ExceptionUtil.getStackTrace(e));
        } finally {
            // 关闭文档对象（针对不同类型的文档进行资源释放）
            if (document != null) {
                try {
                    if (document instanceof SXSSFWorkbook) {
                        ((SXSSFWorkbook) document).close();
                    } else if (document instanceof XWPFDocument) {
                        ((XWPFDocument) document).close();
                    } else if (document instanceof ByteArrayInputStream) {
                        ((ByteArrayInputStream) document).close();
                    }
                } catch (IOException e) {
                    log.error("关闭文档对象失败", e);
                }
            }
            try {
                inputStream.close();
            }catch (Exception e){
            } catch (Exception e) {

            }
            try {
                os.close();
            }catch (Exception e){
            } catch (Exception e) {

            }

        }
    }

    private void updateUserDownloadFile(ExportDataDto exportDataDto, String state,String url, String remark) {
        UserDownloadFilePo userDownloadFilePo = new UserDownloadFilePo();
        userDownloadFilePo.setDownloadId(exportDataDto.getDownloadId());
        userDownloadFilePo.setState(state);
        if(!StringUtil.isEmpty(url)){
            userDownloadFilePo.setTempUrl(url);
        }
        userDownloadFilePo.setRemark(remark.length() > 512 ? remark.substring(0, 512): remark);
        userDownloadFileV1InnerServiceSMOImpl.updateUserDownloadFile(userDownloadFilePo);
    }

    /**
     * 线程启动器
     */
    public static void startExportDataExecutor(){
    public static void startExportDataExecutor() {
        log.debug("开始初始化导出队列");
        ExecutorService executorService = Executors.newFixedThreadPool(DEFAULT_EXPORT_POOL);
        executorService.execute(new ExportDataExecutor());
        executorService.execute(new ExportDataExecutor(true));
        log.debug("初始化导出队列完成");

    }
}