67 lines
2.5 KiB
Java
67 lines
2.5 KiB
Java
|
package cn.keking.utils;
|
||
|
|
|
||
|
|
import org.apache.pdfbox.pdmodel.PDDocument;
|
||
|
|
import org.apache.pdfbox.rendering.ImageType;
|
||
|
|
import org.apache.pdfbox.rendering.PDFRenderer;
|
||
|
|
import org.apache.pdfbox.tools.imageio.ImageIOUtil;
|
||
|
|
import org.slf4j.Logger;
|
||
|
|
import org.slf4j.LoggerFactory;
|
||
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
||
|
|
import org.springframework.stereotype.Component;
|
||
|
|
|
||
|
|
import java.awt.image.BufferedImage;
|
||
|
|
import java.io.File;
|
||
|
|
import java.io.IOException;
|
||
|
|
import java.util.ArrayList;
|
||
|
|
import java.util.List;
|
||
|
|
|
||
|
|
@Component
|
||
|
|
public class PdfUtils {
|
||
|
|
|
||
|
|
private final Logger LOGGER = LoggerFactory.getLogger(PdfUtils.class);
|
||
|
|
|
||
|
|
@Autowired
|
||
|
|
FileUtils fileUtils;
|
||
|
|
|
||
|
|
public List<String> pdf2jpg(String pdfFilePath, String pdfName, String url) {
|
||
|
|
List<String> imageUrls = new ArrayList<>();
|
||
|
|
Integer imageCount = fileUtils.getConvertedPdfImage(pdfFilePath);
|
||
|
|
String imageFileSuffix = ".jpg";
|
||
|
|
// https://8个字符 http://7个字符 从这后面开始出现的第一个/就是当前file.Dir下的根目录
|
||
|
|
int index1 = url.indexOf("/", 8);
|
||
|
|
String pdfFolder = pdfName.substring(0, pdfName.length() - 4);
|
||
|
|
String urlPrefix = url.substring(0, index1 + 1) + pdfFolder;
|
||
|
|
if (imageCount != null && imageCount.intValue() > 0) {
|
||
|
|
for (int i = 0; i < imageCount ; i++)
|
||
|
|
imageUrls.add(urlPrefix + "/" + i + imageFileSuffix);
|
||
|
|
return imageUrls;
|
||
|
|
}
|
||
|
|
try {
|
||
|
|
File pdfFile = new File(pdfFilePath);
|
||
|
|
PDDocument doc = PDDocument.load(pdfFile);
|
||
|
|
int pageCount = doc.getNumberOfPages();
|
||
|
|
PDFRenderer pdfRenderer = new PDFRenderer(doc);
|
||
|
|
|
||
|
|
int index = pdfFilePath.lastIndexOf(".");
|
||
|
|
String folder = pdfFilePath.substring(0, index);
|
||
|
|
|
||
|
|
File path = new File(folder);
|
||
|
|
if (!path.exists()) {
|
||
|
|
path.mkdirs();
|
||
|
|
}
|
||
|
|
String imageFilePath;
|
||
|
|
for (int pageIndex = 0; pageIndex < pageCount; pageIndex++) {
|
||
|
|
imageFilePath = folder + File.separator + pageIndex + imageFileSuffix;
|
||
|
|
BufferedImage image = pdfRenderer.renderImageWithDPI(pageIndex, 105, ImageType.RGB);
|
||
|
|
ImageIOUtil.writeImage(image, imageFilePath, 105);
|
||
|
|
imageUrls.add(urlPrefix + "/" + pageIndex + imageFileSuffix);
|
||
|
|
}
|
||
|
|
doc.close();
|
||
|
|
fileUtils.addConvertedPdfImage(pdfFilePath, pageCount);
|
||
|
|
} catch (IOException e) {
|
||
|
|
LOGGER.error("Convert pdf to jpg exception", e);
|
||
|
|
}
|
||
|
|
return imageUrls;
|
||
|
|
}
|
||
|
|
}
|