java pdf转base64图片工具类

java pdf转base64图片工具类

依赖

		<dependency>
			<groupId>org.apache.pdfbox</groupId>
			<artifactId>pdfbox</artifactId>
			<version>2.0.13</version>
		</dependency>

工具类


import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.rendering.ImageType;
import org.apache.pdfbox.rendering.PDFRenderer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import sun.misc.BASE64Encoder;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;

/**
 * @Author dume
 * @Description PDF转base64图片
 * @Date 2023/03/28/15:01
 * @Version 1.0
 */
public class PdfToImgUtils {
    private static Logger log = LoggerFactory.getLogger(PdfToImgUtils.class);
    public static final float DEFAULT_DPI = 105;

    public static void main(String[] args) {
        long start = System.currentTimeMillis();
//        System.out.println(getBase64("https://www.baidu.com/pdffile/20020230328132145.pdf"));
//        long end = System.currentTimeMillis();
//        System.out.println(end - start + "ms");
    }

    public static String getBase64(String pdfURL) {
        System.setProperty("sun.java2d.cmm", "sun.java2d.cmm.kcms.KcmsServiceProvider");

        //图像合并使用参数
        // 总宽度
        int width = 0;
        // 保存一张图片中的RGB数据
        int[] singleImgRGB;
        int shiftHeight = 0;
        //保存每张图片的像素值
        BufferedImage imageResult = null;

        InputStream pdfInputStream = null;
        ByteArrayOutputStream pdfoutStream = null;
        ByteArrayOutputStream bos = null;
        PDDocument pdDocument = null;
        byte[] buffer = null;
        try {
            // 创建URL
            URL url = new URL(pdfURL);
            // 创建链接
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            conn.setRequestMethod("GET");
            conn.setConnectTimeout(10 * 1000);
            conn.setReadTimeout(10 * 1000);
            pdfInputStream = conn.getInputStream();
            pdfoutStream = new ByteArrayOutputStream();
            // 将内容读取内存中
            buffer = new byte[4926];
            int lenn = -1;
            while ((lenn = pdfInputStream.read(buffer)) != -1) {
                pdfoutStream.write(buffer, 0, lenn);
            }
            buffer = pdfoutStream.toByteArray();
            pdDocument = PDDocument.load(buffer);
            PDFRenderer renderer = new PDFRenderer(pdDocument);
            int pageCount = pdDocument.getNumberOfPages();


            //循环每个页码
            for (int i = 0; i < pageCount; i++) {
                BufferedImage image = renderer.renderImageWithDPI(i, DEFAULT_DPI, ImageType.RGB);
                int imageHeight = image.getHeight();
                int imageWidth = image.getWidth();
                //计算高度和偏移量
                if (i == 0) {
                    //使用第一张图片宽度;
                    width = imageWidth;
                    //保存每页图片的像素值
                    imageResult = new BufferedImage(width, imageHeight * pageCount, BufferedImage.TYPE_INT_RGB);
                } else {
                    // 计算偏移高度
                    shiftHeight += imageHeight;
                }
                singleImgRGB = image.getRGB(0, 0, width, imageHeight, null, 0, width);
                // 写入流中
                imageResult.setRGB(0, shiftHeight, width, imageHeight, singleImgRGB, 0, width);

            }

            bos = new ByteArrayOutputStream();
            ImageIO.write(imageResult, "jpg", bos);
            byte[] bytes = bos.toByteArray();
            if (null == bytes) {
                throw new Exception("获取图片为空!");
            }
            String base = new BASE64Encoder().encode(bytes).trim().replaceAll("\n", "").replaceAll("\r", "");
            pdDocument.close();
           // return "data:image/jpg;base64," + base;
            return base;

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (null != pdfInputStream) {
                    pdfInputStream.close();
                }

            } catch (Exception e) {
                e.printStackTrace();
            }
            try {
                if (null != pdfoutStream) {
                    pdfoutStream.close();
                }

            } catch (Exception e) {
                e.printStackTrace();
            }

            try {
                if (null != bos) {
                    bos.close();
                }

            } catch (Exception e) {
                e.printStackTrace();
            }
            try {
                if (null != pdDocument) {
                    pdDocument.close();
                }

            } catch (Exception e) {
                e.printStackTrace();
            }
        }

        return null;

    }

}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小七蒙恩

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值