java實現網頁截圖

使用工具

java+selenium+phantomjs /chromedriver /firefox

1.分別是 phantomjs插件 google截圖插件 和 firefox火狐瀏覽器截圖插件
2. selenium工具 是對頁面進行點擊操作后在截圖 需要用到的工具。

插件的使用方法幾乎一致

上代碼：1.使用phantomjs截圖

    public static String getPicByPhantomjs(Map<String ,String> map,String pic,File sf){
                logger.warn("使用phantomjs截圖鏈接:"+map.get("url"));
                //定義圖片存儲路徑
                DesiredCapabilities dcaps = null;
                PhantomJSDriver driver = null;
                 String picName = null;
                 try {
                //設置必要參數
                dcaps =  new DesiredCapabilities();
                //ssl證書支持
                dcaps.setCapability("acceptSslCerts", true);
                //截屏支持
                dcaps.setCapability("takesScreenshot", true);
                //css搜索支持
                dcaps.setCapability("cssSelectorsEnabled", true);
                //js支持
                dcaps.setJavascriptEnabled(true);
                //驅動支持（第二參數表明的是你的phantomjs引擎所在的路徑）
                dcaps.setCapability(PhantomJSDriverService.PHANTOMJS_EXECUTABLE_PATH_PROPERTY,
                        "D:/phantomjs/phantomjs-2.1.1-windows/bin/phantomjs.exe");
//                        "F:/test/phantomjs-2.1.1-windows/bin/phantomjs.exe");
                //創建無界面瀏覽器對象
                driver = new PhantomJSDriver(dcaps);
                long start = System.currentTimeMillis();
//                 driver.get(url); 
                 driver.get(map.get("url")); 
                 //設置隱性等待（作用於全局）
                 driver.manage().timeouts().implicitlyWait(2, TimeUnit.SECONDS);
                 Thread.sleep(5* 1000);
                     if(!"".equals(map.get("str")) && map.get("str")!=null){
                     if(ElementExist(driver,By.className(map.get("str")))){
                         WebElement inputBox = driver.findElement(By.className(map.get("str")));
                         Actions action = new Actions(driver);
                        action.click(inputBox).build().perform();
                         //元素點擊 后等待加載
                         Thread.sleep(2 * 1000);
                       }
                     }
               
                 JavascriptExecutor js = driver;
                 //頁面下滑10次,每次下滑加載2s
                 for (int i = 0; i < 10; i++) {
                     js.executeScript("window.scrollBy(0,1000)");
                     //睡眠2s等js加載完成
                     Thread.sleep(2 * 1000);
                 }
                 //指定了OutputType.FILE做為參數傳遞給getScreenshotAs()方法，其含義是將截取的屏幕以文件形式返回。
                 File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
                 Thread.sleep(2000);
                 //利用FileUtils工具類的copyFile()方法保存getScreenshotAs()返回的文件對象
                 FileUtils.copyFile(srcFile, new File(sf.getPath()+File.separator+pic));
                 Thread.sleep(2000);
                 Thumbnails.of(sf.getPath()+File.separator+pic).scale(1.0f).toFile(sf.getPath()+File.separator+pic);
                 driver.close();
                 driver.quit();
                 picName = pic;
                 System.out.println("本次截圖耗時：" + (System.currentTimeMillis() - start) + " 毫秒");
//                 System.out.println("轉換后的鏈接:"+map.get("url"));
        } catch (Exception e) {
             driver.close();
             driver.quit();
             picName = "";
             logger.warn("使用phantomjs截圖時："+e.toString());
        }
        return picName;
    }

2.使用 chromedriver截圖

    public static String getPicByChrom(Map<String ,String> map,String pic,File sf){

                logger.warn("使用chromedriver截圖鏈接:"+map.get("url"));
                //定義圖片存儲路徑
                WebDriver driver = null ;
                String picName = null;
                try {
                long start = System.currentTimeMillis();
                String chromeDriverDir = "D:/chromedriver/chromedriver_win32/chromedriver.exe";
                System.setProperty("webdriver.chrome.driver", chromeDriverDir);
                driver = new ChromeDriver();
                driver.manage().window().maximize();
                 driver.get(map.get("url")); 
                 //設置隱性等待（作用於全局）
                 driver.manage().timeouts().implicitlyWait(2, TimeUnit.SECONDS);
                 Thread.sleep(5* 1000);
                     if(!"".equals(map.get("str")) && map.get("str")!=null){
                     if(ElementExist(driver,By.className(map.get("str")))){
                         WebElement inputBox = driver.findElement(By.className(map.get("str")));
                         Actions action = new Actions(driver);
                        action.click(inputBox).build().perform();
                         //元素點擊 后等待加載
                         Thread.sleep(2 * 1000);
                     }
                     }
               
//                 JavascriptExecutor js = (JavascriptExecutor) driver;
                 //頁面下滑10次,每次下滑加載2s
//                 for (int i = 0; i < 1; i++) {
//                     js.executeScript("window.scrollBy(0,1000)");
//                     //睡眠2s等js加載完成
//                     Thread.sleep(2 * 1000);
//                 }
                 //指定了OutputType.FILE做為參數傳遞給getScreenshotAs()方法，其含義是將截取的屏幕以文件形式返回。
                 File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
                 Thread.sleep(2000);
                 //利用FileUtils工具類的copyFile()方法保存getScreenshotAs()返回的文件對象
                 FileUtils.copyFile(srcFile, new File(sf.getPath()+File.separator+pic));
                 Thread.sleep(2000);
                 Thumbnails.of(sf.getPath()+File.separator+pic).scale(1.0f).toFile(sf.getPath()+File.separator+pic);
                 driver.close();
                 driver.quit();
                 picName = pic;
                 System.out.println("本次截圖耗時：" + (System.currentTimeMillis() - start) + " 毫秒");
//                 System.out.println("轉換后的鏈接:"+map.get("url"));
        } catch (Exception e) {
             driver.close();
             driver.quit();
             picName = "";
             logger.warn("使用chromedriver截圖時："+e.toString());
        }
        return picName;
    }

注 ：

1.phantomjs截圖會創建無界面瀏覽器進行截圖，可截取網頁長圖 （推薦使用）
2.chromedriver 截圖工具只能截取當前瀏覽器可見區域長度(調用瀏覽器打開url地址，插件版本需要與chrom瀏覽器版本一致)

3.Thumbnails 為圖片壓縮工具