Java根据url生成图片、截图效果

node -v
npm -v

npm config set registry https://registry.npmmirror.com

npm install puppeteer

const puppeteer = require('puppeteer');
// 获取java传过来的参数
const url = process.argv[2];
(async () => {
    const browser = await puppeteer.launch();
    const page = await browser.newPage();
    // 设置视口大小
    await page.setViewport({
        width: 1920,
        height: 1080,
        deviceScaleFactor: 2, // 像素比，默认为 1。在高 DPI 屏幕上，可以将其设置为 2 或更高，以模拟高分辨率显示。
    });
    await page.goto(url);
    const screenshotBuffer = await page.screenshot({
        fullPage: true;//是否截全屏
    });
    // 输出截图的 Buffer 到标准输出流
    process.stdout.write(screenshotBuffer);
    await browser.close();
})();

node screenshot.js https://www.baidu.com/

package com.assoft.erms.common.utils;
import java.io.ByteArrayOutputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
public class ScreenshotUtil {
    public static void main(String[] args) {
        //js脚本文件地址
        String screenshotJs = "D:\\workspace\\javascript\\screenshot.js";
        //要截图的网页（有拦截器，需要验证身份的不行）
        String url = "http://politics.people.com.cn/n1/2024/1031/c1024-40350705.html";
        //保存的文件名
        String filePath = "screenshot.png";
        try {
            //两种返回类型，逻辑是一样的；只是返回类型不同 根据情况自行选择
            // 模式一：获取截图的字节数组
            byte[] screenshotBytes = ScreenshotUtil.captureScreenshotAsBytes(url,screenshotJs);
            System.out.println("屏幕截图字节数组长度: " + screenshotBytes.length);
            // 模式二：将截图保存到指定路径
            ScreenshotUtil.captureScreenshotAsFile(url, filePath,screenshotJs);
            System.out.println("屏幕截图已保存到: " + filePath);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    /**
     * 获取网页截图的字节数组
     * @param url
     * @return
     * @throws Exception
     */
    public static byte[] captureScreenshotAsBytes(String url,String screenshotJs) throws Exception {
        Process process = Runtime.getRuntime().exec(new String[]{"node", screenshotJs, url});
        // 获取 Process 的输入流
        try (InputStream inputStream = process.getInputStream();
             ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream()) {
            byte[] buffer = new byte[1024];
            int length;
            while ((length = inputStream.read(buffer)) != -1) {
                byteArrayOutputStream.write(buffer, 0, length);
            }
            process.waitFor(); // 等待进程结束
            return byteArrayOutputStream.toByteArray(); // 返回字节数组
        }
    }
    /**
     * 将网页截图保存到指定路径
     * @param url
     * @param filePath
     * @throws Exception
     */
    public static void captureScreenshotAsFile(String url, String filePath,String screenshotJs) throws Exception {
        byte[] screenshotBytes = captureScreenshotAsBytes(url,screenshotJs); // 获取截图字节数组
        // 将字节数组写入文件
        try (FileOutputStream fileOutputStream = new FileOutputStream(filePath)) {
            fileOutputStream.write(screenshotBytes);
        }
    }
}

|-- node_modules
|-- screenshot.js

|-- node_modules
|-- js
	|-- screenshot.js

|-- screenshot.js
|-- static
	|-- node_modules