你的位置:首页 > Java教程

[Java教程]使用PhantomJS实现网页截图服务


    这是上半年遇到的一个小需求,想实现网页的抓取,并保存为图片。研究了不少工具,效果都不理想,不是显示太差了(Canvas、Html2Image、Cobra),就是性能不怎么样(如SWT的Brower)。后发现无界面浏览器可以满足这个条件,大致研究了一下PhantomJS与CutyCapt,两者都是Webkit内核,其中PhantomJS使用上更方便一些,尤其在Windows平台上,如果在Linux下,从2.0版本后需要自己去机器上编译了(大概要编译3个小时,不得不说,g++就是个渣渣,同样的项目,在vc下编译快得,不谈了,毕竟是免费开源的编译器)。下面介绍PhantomJS结合Java代码实现的网页截图技术:

   

 一、环境准备

    1、PhantomJS脚本的目录:D:/xxx/phantomjs-2.0.0-windows/bin/phantomjs

    2、截图脚本:D:/xxx/phantomjs-2.0.0-windows/bin/rasterize.js

    截图的脚本在官网上有提供,但是我这里需要说明一下它的高宽度设计原理:

page.viewportSize = { width: 600, height: 600 };

   这个是默认的高度,也就是600X600,我建议大家把height设置小一点,我这边设置的是width:800,height:200。因为实际上,在不同时设置高度与亮度的情况下,如果真实的网页的高度大于设置值时,图片会自动扩充高宽度的,直到整个页面显示完(当你想截取小的图片时,可能由于默认设置的太大,会使图片有很大一块空的)。如果同时设置了高宽度,下面的代码会被执行,就会对网页的部分进行截取了:

page.clipRect = { top: 0, left: 0, width: pageWidth, height: pageHeight };

    3、先用命令行测试一下:    

D:/xxx/phantomjs-2.0.0-windows/bin/phantomjs D:/xxx/phantomjs-2.0.0-windows/bin/rasterize.js http://www.qq.com D:/test.png

    如果配置好了,应该可以看到生成的图片了。当然还可以配置高宽度的参数,在上面的命令后加上:" 1000px"或" 1000px*400px",都是可以的。

 

二、服务器代码

    作为一个网页截图服务,这部分代码片段应当被布署在服务器上,当然不必全照搬啦,根据自己的需求来用就好了:

 1 package lekkoli.test; 2  3 import java.io.BufferedInputStream; 4 import java.io.BufferedReader; 5 import java.io.ByteArrayOutputStream; 6 import java.io.File; 7 import java.io.FileInputStream; 8 import java.io.IOException; 9 import org.apache.log4j.Logger;  10  11 /** 12  * 网页转图片处理类,使用外部CMD 13  * @author lekkoli 14 */ 15 public class PhantomTools { 16  17   private static final Logger _logger = Logger.getLogger(PhantomTools.class); 18  19   // private static final String _tempPath = "/data/temp/phantom_"; 20   // private static final String _shellCommand = "/usr/local/xxx/phantomjs /usr/local/xxx/rasterize.js "; Linux下的命令 21   private static final String _tempPath = "D:/data/temp/phantom_"; 22   private static final String _shellCommand = "D:/xxx/phantomjs-2.0.0-windows/bin/phantomjs D:/xxx/phantomjs-2.0.0-windows/bin/rasterize.js ";   23  24   private String _file; 25   private String _size; 26  27   /** 28    * 构造截图类 29    * @parm hash 用于临时文件的目录唯一化 30   */ 31   public PhantomTools(int hash) { 32     _file = _tempPath + hash + ".png"; 33   } 34  35   /** 36    * 构造截图类 37    * @parm hash 用于临时文件的目录唯一化 38    * @param size 图片的大小,如800px*600px(此时高度会裁切),或800px(此时 高度最少=宽度*9/16,高度不裁切) 39   */ 40   public PhantomTools(int hash, String size) { 41     this(hash); 42     if (size != null) 43       _size = " " + size; 44   } 45  46   /** 47    * 将目标网页转为图片字节流 48    * @param url 目标网页地址 49    * @return 字节流 50   */ 51   public byte[] getByteImg(String url) throws IOException { 52     BufferedInputStream in = null; 53     ByteArrayOutputStream out = null; 54     File file = null; 55     byte[] ret = null; 56     try { 57       if (exeCmd(_shellCommand + url + " " + _file + (_size != null ? _size : ""))) { 58         file = new File(_file); 59         if (file.exists()) { 60           out = new ByteArrayOutputStream(); 61           byte[] b = new byte[5120]; 62           in = new BufferedInputStream(new FileInputStream(file)); 63           int n; 64           while ((n = in.read(b, 0, 5120)) != -1) { 65             out.write(b, 0, n); 66           } 67           file.delete(); 68           ret = out.toByteArray(); 69         } 70       } else { 71         ret = new byte[] {}; 72       } 73     } finally { 74       try { 75         if (out != null) { 76           out.close(); 77         } 78       } catch (IOException e) { 79         _logger.error(e); 80       } 81       try { 82         if (in != null) { 83           in.close(); 84         } 85       } catch (IOException e) { 86         _logger.error(e); 87       } 88       if (file != null && file.exists()) { 89         file.delete(); 90       } 91     } 92     return ret; 93   } 94  95   /** 96    * 执行CMD命令 97   */ 98   private static boolean exeCmd(String commandStr) { 99     BufferedReader br = null;100     try {101       Process p = Runtime.getRuntime().exec(commandStr);102       if (p.waitFor() != 0 && p.exitValue() == 1) {103         return false;104       }105     } catch (Exception e) {106       _logger.error(e);107     } finally {108       if (br != null) {109         try {110           br.close();111         } catch (Exception e) {112           _logger.error(e);113         }114       }115     }116     return true;117   }118 }

    使用上面的PhantomTools类,可以很方便地调用getByteImg方法来生成并获取图片内容。  

     

    附上我的截图配置脚本:rasterize.js,至于PhantomJS,大家就自行去官网下载吧。

    转载请注明原址:http://www.cnblogs.com/lekko/p/4796062.html