用SpringBoot对网页截屏并轻松保存为图片

2024-01-07 18:01:54

使用SpringBoot和Selenium截取网页屏幕：分步指南

简介

在当今数字时代，截取屏幕已成为一种普遍需求，用于保存、共享或记录信息。通过使用SpringBoot和Selenium，我们可以轻松实现网页屏幕截图并将其保存为图片。

配置Selenium

首先，我们需要配置Selenium。在Maven项目中，添加以下依赖项：

<dependency>
    <groupId>org.seleniumhq.selenium</groupId>
    <artifactId>selenium-java</artifactId>
    <version>4.7.0</version>
</dependency>

下载Selenium WebDriver并将其解压到项目中。

使用Selenium驱动无头浏览器

Selenium可以控制各种浏览器，如Chrome和Firefox。在本文中，我们将使用Chrome。

WebDriver driver = new ChromeDriver();

使用WebDriver实例，我们打开浏览器并导航到目标网页。

driver.get("https://www.google.com");

截取屏幕

File screenshot = driver.getScreenshotAs(OutputType.FILE);

getScreenshotAs方法返回一个包含屏幕截图的文件对象。

保存屏幕截图

FileUtils.copyFile(screenshot, new File("screenshot.png"));

copyFile方法将屏幕截图文件复制到指定路径。

完整代码示例

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import java.io.File;
import org.apache.commons.io.FileUtils;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

@SpringBootApplication
public class ScreenshotApplication {

    public static void main(String[] args) {
        SpringApplication.run(ScreenshotApplication.class, args);
    }

    @RestController
    public class ScreenshotController {

        @GetMapping("/screenshot")
        public String takeScreenshot(@RequestParam String url) {
            WebDriver driver = new ChromeDriver();
            driver.get(url);
            File screenshot = driver.getScreenshotAs(OutputType.FILE);
            FileUtils.copyFile(screenshot, new File("screenshot.png"));
            driver.quit();
            return "Screenshot saved to screenshot.png";
        }
    }
}