package com.ruoyi.web.controller.project;
|
|
import com.itextpdf.text.DocumentException;
|
import com.ruoyi.common.config.RuoYiConfig;
|
import com.ruoyi.common.core.controller.BaseController;
|
import com.ruoyi.common.utils.MergeFilesToPDFUtils;
|
import com.ruoyi.project.domain.MegerIsRemove;
|
import com.ruoyi.project.domain.ServiceDonateannex;
|
import com.ruoyi.project.domain.ServiceDonatebaseinfo;
|
import com.ruoyi.project.service.IBaseOnlyvalueService;
|
import com.ruoyi.project.service.IServiceDonateannexService;
|
import com.ruoyi.project.service.IServiceDonatebaseinfoService;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.util.CollectionUtils;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RequestParam;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.File;
|
import java.io.IOException;
|
import java.util.ArrayList;
|
import java.util.Collections;
|
import java.util.Comparator;
|
import java.util.List;
|
|
/**
|
* PDF合并
|
*
|
* @author ruoyi
|
* @date 2022-01-24
|
*/
|
@Slf4j
|
@Api("PDF合并")
|
@RestController
|
@RequestMapping("/project/pdfmerge")
|
public class PdfMergeController extends BaseController {
|
@Autowired
|
private IServiceDonateannexService serviceDonateannexService;
|
|
@Autowired
|
private MergeFilesToPDFUtils mergeFilesToPDFUtils;
|
|
@Autowired
|
private IBaseOnlyvalueService baseOnlyvalueService;
|
|
@Autowired
|
private IServiceDonatebaseinfoService donatebaseinfoService;
|
|
/**
|
* 图片、pdf、文档合并成pdf
|
*/
|
@ApiOperation("图片、pdf、文档合并成pdf")
|
@GetMapping("/merge")
|
public void merge(@RequestParam("infoId") Long infoId, HttpServletResponse response) {
|
log.info("merge的入参为辅：{}", infoId);
|
ServiceDonateannex serviceDonateannex = new ServiceDonateannex();
|
serviceDonateannex.setInfoid(infoId);
|
List<ServiceDonateannex> serviceDonateannexes = serviceDonateannexService.queryList(serviceDonateannex);
|
if (!CollectionUtils.isEmpty(serviceDonateannexes)) {
|
log.info("serviceDonateannexes查询的数据为：{}", serviceDonateannexes.size());
|
//根据Annexfilestype进行排序
|
Collections.sort(serviceDonateannexes, Comparator.comparingInt(ServiceDonateannex::getAnnexfilestype));
|
//获取文件存放的位置（配置文件里的Profile地址）
|
String localPath = RuoYiConfig.getProfile();
|
//用于转成一维数组
|
List<String> list = new ArrayList<>();
|
//用于记录该临时pdf文件是否删除
|
List<MegerIsRemove> megerIsRemoveList = new ArrayList<>();
|
for (ServiceDonateannex serviceDonateannex1 : serviceDonateannexes) {
|
String path = serviceDonateannex1.getAnnexurl().replace("/profile", localPath);
|
String suffix = path.substring(path.lastIndexOf("."));
|
Boolean aBoolean = false;
|
if (suffix.equals(".jpg") || suffix.equals(".JPG") || suffix.equals(".PNG") || suffix.equals(".png") || suffix.equals(".JPEG") || suffix.equals(".jpeg")) {
|
//图片处理
|
try {
|
aBoolean = mergeFilesToPDFUtils.addImageToPdf(path, getNewFileNameWithExtension(path, ".pdf"));
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
} else if (suffix.equals(".pdf") || suffix.equals(".PDF")) {
|
//pdf文件处理
|
String pdf = getNewFileNameWithExtension(path, ".pdf");
|
list.add(pdf);
|
megerIsRemoveList.add(new MegerIsRemove(false, pdf));
|
} else if (suffix.equals(".doc") || suffix.equals(".DOC") || suffix.equals(".docx") || suffix.equals(".DOCX")) {
|
//word文件处理
|
aBoolean = mergeFilesToPDFUtils.wordtoPdf(path, getNewFileNameWithExtension(path, ".pdf"));
|
}
|
if (aBoolean) {
|
//将图片和word文件的临时文件，存到集合中，以便后面删除
|
String pdf = getNewFileNameWithExtension(path, ".pdf");
|
list.add(pdf);
|
megerIsRemoveList.add(new MegerIsRemove(true, pdf));
|
}
|
}
|
//获取合并名称
|
ServiceDonatebaseinfo donatebaseinfo = donatebaseinfoService.getById(serviceDonateannexes.get(0).getInfoid());
|
|
String fileName = donatebaseinfo.getIdcardno() + donatebaseinfo.getName() + ".pdf";
|
try {
|
//文件合并
|
mergeFilesToPDFUtils.mergePdfFiles(list, fileName);
|
} catch (IOException e) {
|
e.printStackTrace();
|
} catch (DocumentException e) {
|
e.printStackTrace();
|
}
|
//删除临时文件
|
for (MegerIsRemove megerIsRemove : megerIsRemoveList) {
|
if (megerIsRemove.getIsdel()) {
|
new File(megerIsRemove.getPath()).delete();
|
}
|
}
|
}
|
}
|
|
private static String getNewFileNameWithExtension(String fileName, String extension) {
|
int dotIndex = fileName.lastIndexOf('.');
|
return dotIndex != -1 ? fileName.substring(0, dotIndex) + extension : fileName;
|
}
|
|
}
|
