package com.java110.job.export;
|
|
import com.java110.core.client.*;
|
import com.java110.core.log.LoggerFactory;
|
import com.java110.dto.data.ExportDataDto;
|
import com.java110.utils.cache.MappingCache;
|
import com.java110.utils.factory.ApplicationContextFactory;
|
import com.java110.utils.util.COSUtil;
|
import com.java110.utils.util.OSSUtil;
|
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
|
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.util.concurrent.ExecutorService;
|
import java.util.concurrent.Executors;
|
|
/**
|
* 导出数据执行器
|
*/
|
public class ExportDataExecutor implements Runnable{
|
private static final Logger log = LoggerFactory.getLogger(ExportDataQueue.class);
|
|
|
//默认线程大小
|
private static final int DEFAULT_EXPORT_POOL = 4;
|
|
@Override
|
public void run() {
|
|
try {
|
doExportData();
|
}catch (Exception e){
|
log.error("处理消息异常",e);
|
e.printStackTrace();
|
}
|
|
}
|
|
private void doExportData() throws Exception {
|
|
ExportDataDto exportDataDto = ExportDataQueue.getData();
|
|
if(exportDataDto == null){
|
return;
|
}
|
|
String businessAdapt = exportDataDto.getBusinessAdapt();
|
|
IExportDataAdapt exportDataAdaptImpl = ApplicationContextFactory.getBean(businessAdapt, IExportDataAdapt.class);
|
|
if(exportDataAdaptImpl == null){
|
return ;
|
}
|
SXSSFWorkbook workbook= exportDataAdaptImpl.exportData(exportDataDto);
|
|
//保存文件路径到 文件下载表
|
FileUploadTemplate fileUploadTemplate = ApplicationContextFactory.getBean("fileUploadTemplate", FileUploadTemplate.class);
|
|
|
ByteArrayInputStream inputStream = null;
|
ByteArrayOutputStream os = null;
|
try {
|
|
os = new ByteArrayOutputStream();
|
workbook.write(os);
|
inputStream = new ByteArrayInputStream(os.toByteArray());
|
|
fileUploadTemplate.saveFile(inputStream,exportDataDto.getFileName());
|
|
}finally {
|
try {
|
workbook.close();
|
}catch (Exception e){
|
|
}
|
try {
|
inputStream.close();
|
}catch (Exception e){
|
|
}
|
try {
|
os.close();
|
}catch (Exception e){
|
|
}
|
|
}
|
}
|
|
/**
|
* 线程启动器
|
*/
|
public static void startExportDataExecutor(){
|
ExecutorService executorService = Executors.newFixedThreadPool(DEFAULT_EXPORT_POOL);
|
executorService.execute(new ExportDataExecutor());
|
}
|
}
|
