Java使用Puppeteer将网页指定区域转成pdf

刺骨的言语ヽ痛彻心扉 2022-12-01 15:56 412阅读 0赞

# 友情提示：本文章只提供方法思路，不要复制粘贴后说跑不通 #

# 一、需要先安装NodeJs，网上教程很多 #

# 二、安装puppeteer依赖，如果npm下载不成功就使用cnpm命令（cnpm需要先安装） #

npm install puppeteer --save

# 三、在安装puppeteer依赖的目录下创建puppeteer.js #

# 四、Java项目中创建PuppeteerHtmlToPdf.class #

/** * * @author zx * @date 2020/8/5 * 用谷歌提供的node实现的Puppeteer，实现网页截图及把网页生成pdf文件采 */
    public class PuppeteerHtmlToPdf { 
        /** * html转pdf,直接通过流输出到浏览器 * @param response 浏览器响应 * @param fileName 文件名称 * @param puppeteerjs 要采用哪个js文件执行 * @param webSiteUrl 要生成pdf/图片的网页 * @param types 类型 pdf代表要生成pdf文件，jpg代表要生成jpg图片 * @param mobileWidth 需要转化的网页宽度 * @param mobileHight 需要转化的网页长度 */
        public static void parseHtml2Pdf(HttpServletResponse response, String fileName, String puppeteerjs, String webSiteUrl, String types, String mobileWidth, String mobileHight) { 
            try { 
                Runtime rt = Runtime.getRuntime();
                //Process p = rt.exec("node C:\\Users\\boshi\\Desktop\\iview-admin-master\\hn.js http://yunfu.ifutest.com/adminAchievement/viewPdfHtml/7e6bba53b5d34124a54c0648a825bcda pdf");
                Process p = rt.exec("node " + puppeteerjs + " " + webSiteUrl + " " + types + " " + mobileWidth + " " + mobileHight);
                InputStream is = p.getInputStream();
                BufferedInputStream bf = new BufferedInputStream(is);
                byte[] data = IOUtils.toByteArray(bf);
                fileName = URLEncoder.encode(fileName, "UTF-8");
                response.setHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\"");
                response.addHeader("Content-Length", "" + data.length);
                response.setContentType("application/octet-stream;charset=UTF-8");
                OutputStream outputStream = new BufferedOutputStream(response.getOutputStream());
                outputStream.write(data);
                outputStream.flush();
                outputStream.close();
            } catch (IOException e) { 
                e.printStackTrace();
            }
    
        }
    }

# 五、在你需要将网页转成pdf是去调用PuppeteerHtmlToPdf 的parseHtml2Pdf方法即可 #

# 六、注意：需要等页面加载完才去转pdf,需要在页面设置一个标志用于判断页面是否加载完成（这里前端使用的是vue的语法，页面加载完会把addNameFlags置为true,只有为true才设置class属性nodeCompleted，传统的可以在onload函数中给标签绑定class属性） #

![在这里插入图片描述][20200828112938998.png_pic_center]

[20200828112938998.png_pic_center]: /images/20221123/5273593fde2b4c4da7380d9ff4109e89.png