Java实现网页截屏功能（图片下载功能）的几种方式（整理）

最近的项目中，需要实现一个将网页中显示的内容导出为图片的功能，故研究了一下可以实现这个功能的各种方式，整理在此，以供以后使用。

p.s. 本人为java研发，所有涉及的代码，实现方式均为java的方式。

前台方式

1. 利用html2canvas实现前台截屏

html2canvas项目的gitHub地址
html2canvas示例

html2canvas可以将HTML代码块进行截取，并生成快照形式的canvas，然后可以利用html5的下载功能提供图片的下载接口，供用户下载。

优点：前台技术，实现比较容易。

缺点：此项目使用了html5的技术，因此在对IE的兼容上，只可以在IE9+的版本上使用。

使用时，需要引用jQuery.js和html2canvas.js。（html2canvas.js可以在github上下载）

具体JavaScript代码如下:

var canvas = function () {
    html2canvas($("#chart"), {
              onrendered: function (canvas) {
                  $("#class11").html(canvas);//将id为“class11”部分的代码转换为canvas
                  var type = 'png';//图片类型为png
                  var imgData = canvas.toDataURL('png');//将图片转换为png类型的流

                  var _fixType = function (type) {
                      type = type.toLowerCase().replace(/jpg/i, 'jpeg');
                      var r = type.match(/png|jpeg|bmp|gif/)[0];
                      return 'image/' + r;
                  };

                  var saveFile = function (data, filename) {
                      var save_link = document.createElementNS('http://www.w3.org/1999/xhtml', 'a');
                      save_link.href = data;
                      save_link.download = filename;

                      var event = document.createEvent('MouseEvents');
                      event.initMouseEvent('click', true, false, window, 0, 0, 0, 0, 0, false, false, false, false, 0, null);
                      save_link.dispatchEvent(event);
                  };

                  //下载时的文件名
                  var filename = 'baidufe_' + (new Date()).getTime() + '.' + type;

                  // 下载文件
                  saveFile(imgData, filename);
              }
          });
};

后台方式（java）

1. 利用html2image实现

html2image的地址（google）
html2image是可以识别html标签并将html转换成图片的java项目。

优点：后台转换，故对浏览器的版本基本没有要求。

缺点：对样式的识别不是很好，转换出来的图片比较简单，基本没有可以兼容的样式。

使用方式：maven中引用如下依赖：

<dependency>
  <groupId>com.github.xuwei-k</groupId>
  <artifactId>html2image</artifactId>
  <version>0.1.0</version>
</dependency>

2. 利用DJNativeSwing项目实现

纯java代码实现。使用DJNativeSwing这个项目，构建一个swing浏览器，打开网址，之后截取图片。（需要eclipse的swt库支持）

优点：纯java实现。

缺点：需要多线程的支持，只在控制台方式实现过（由于本人对于多线程的掌握较弱，目前没有在spring项目中实现这个方法，求大神指导~~）

使用这个方法，需要引用的maven依赖如下：

<dependency>
    <groupId>com.hynnet</groupId>
    <artifactId>DJNativeSwing</artifactId>
    <version>1.0.0</version>
</dependency>

<dependency>
    <groupId>com.hynnet</groupId>
    <artifactId>DJNativeSwing-SWT</artifactId>
    <version>1.0.0</version>
</dependency>

<dependency>                            <groupId>org.eclipse.swt.org.eclipse.swt.win32.win32.x86_64.4.3.swt</groupId>
    <artifactId>org.eclipse.swt.win32.win32.x86_64</artifactId>
    <version>4.3</version>
</dependency>

实现代码如下（引用自http://blog.csdn.net/buddyuu/article/details/52699997）：

public class PrintScreen4DJNativeSwingUtils extends JPanel {

    private static final long serialVersionUID = 1L;
    // 行分隔符
    final static public String LS = System.getProperty("line.separator", "/n");
    // 文件分割符
    final static public String FS = System.getProperty("file.separator", "//");
    // 当网页超出目标大小时 截取
    final static public int maxWidth = 2000;
    final static public int maxHeight = 1400;

    /**
     * @param file
     *            预生成的图片全路径
     * @param url
     *            网页地址
     * @param width
     *            打开网页宽度 ，0 = 全屏
     * @param height
     *            打开网页高度 ，0 = 全屏
     * @return boolean
     */
    public PrintScreen4DJNativeSwingUtils(final String file, final String url, final String WithResult) {
        super(new BorderLayout());
        JPanel webBrowserPanel = new JPanel(new BorderLayout());
        final JWebBrowser webBrowser = new JWebBrowser(null);
        webBrowser.setBarsVisible(false);
        webBrowser.navigate(url);
        webBrowserPanel.add(webBrowser, BorderLayout.CENTER);
        add(webBrowserPanel, BorderLayout.CENTER);
        JPanel panel = new JPanel(new FlowLayout());
        webBrowser.addWebBrowserListener(new WebBrowserAdapter() {
            // 监听加载进度
            public void loadingProgressChanged(WebBrowserEvent e) {
                // 当加载完毕时
                if (e.getWebBrowser().getLoadingProgress() == 100) {
                    String result = (String) webBrowser.executeJavascriptWithResult(WithResult);
                    int index = result == null ? -1 : result.indexOf(":");
                    NativeComponent nativeComponent = webBrowser.getNativeComponent();
                    Dimension originalSize = nativeComponent.getSize();
                    Dimension imageSize = new Dimension(Integer.parseInt(result.substring(0, index)),
                            Integer.parseInt(result.substring(index + 1)));
                    imageSize.width = Math.max(originalSize.width, imageSize.width + 50);
                    imageSize.height = Math.max(originalSize.height, imageSize.height + 50);
                    nativeComponent.setSize(imageSize);
                    BufferedImage image = new BufferedImage(imageSize.width, imageSize.height,
                            BufferedImage.TYPE_INT_RGB);
                    nativeComponent.paintComponent(image);
                    nativeComponent.setSize(originalSize);
                    // 当网页超出目标大小时
                    if (imageSize.width > maxWidth || imageSize.height > maxHeight) {
                        // 截图部分图形
                        image = image.getSubimage(0, 0, maxWidth, maxHeight);
                        // 此部分为使用缩略图
                        /*
                         * int width = image.getWidth(), height = image
                         * .getHeight(); AffineTransform tx = new
                         * AffineTransform(); tx.scale((double) maxWidth /
                         * width, (double) maxHeight / height);
                         * AffineTransformOp op = new AffineTransformOp(tx,
                         * AffineTransformOp.TYPE_NEAREST_NEIGHBOR); //缩小 image
                         * = op.filter(image, null);
                         */
                    }
                    try {
                        // 输出图像
                        ImageIO.write(image, "jpg", new File(file));
                    } catch (IOException ex) {
                        ex.printStackTrace();
                    }
                    // 退出操作
                    System.exit(0);
                }
            }
        });
        add(panel, BorderLayout.SOUTH);

    }

    // 以javascript脚本获得网页全屏后大小
    public static String getScreenWidthHeight() {

        StringBuffer jsDimension = new StringBuffer();
        jsDimension.append("var width = 0;").append(LS);
        jsDimension.append("var height = 0;").append(LS);
        jsDimension.append("if(document.documentElement) {").append(LS);
        jsDimension.append("  width = Math.max(width, document.documentElement.scrollWidth);").append(LS);
        jsDimension.append("  height = Math.max(height, document.documentElement.scrollHeight);").append(LS);
        jsDimension.append("}").append(LS);
        jsDimension.append("if(self.innerWidth) {").append(LS);
        jsDimension.append("  width = Math.max(width, self.innerWidth);").append(LS);
        jsDimension.append("  height = Math.max(height, self.innerHeight);").append(LS);
        jsDimension.append("}").append(LS);
        jsDimension.append("if(document.body.scrollWidth) {").append(LS);
        jsDimension.append("  width = Math.max(width, document.body.scrollWidth);").append(LS);
        jsDimension.append("  height = Math.max(height, document.body.scrollHeight);").append(LS);
        jsDimension.append("}").append(LS);
        jsDimension.append("return width + ':' + height;");

        return jsDimension.toString();
    }

    public static boolean printUrlScreen2jpg(final String file, final String url, final int width, final int height) {

        NativeInterface.open();
        SwingUtilities.invokeLater(new Runnable() {
            public void run() {
                String withResult = "var width = " + width + ";var height = " + height + ";return width +':' + height;";
                if (width == 0 || height == 0)
                    withResult = getScreenWidthHeight();

                JFrame frame = new JFrame("网页截图");
                // 加载指定页面，最大保存为640x480的截图
                frame.getContentPane().add(new PrintScreen4DJNativeSwingUtils(file, url, withResult),
                        BorderLayout.CENTER);
                frame.setSize(640, 480);
                // 仅初始化，但不显示
                frame.invalidate();

                frame.pack();
                frame.setVisible(false);
            }
        });
        NativeInterface.runEventPump();

        return true;
    }

    public static void main(String[] args) {  

        PrintScreen4DJNativeSwingUtils.printUrlScreen2jpg("1122.jpg", "https://www.baidu.com", 1400, 900);

    }
}

3. 利用phantomJs实现

phantomJs的GitHub地址
phantomJs官网

phantom利用的是webKit内核，全面支持web而不需浏览器支持，快速，原生支持各种Web标准。

优点：速度快，使用简单。

缺点：需要在项目的服务器端引用exe文件进行截图，并发性能有待考证。

使用这个方式，需要去官网下载最新的phantomJs脚本。官网地址在上面的链接中。

具体代码如下（引用自http://www.cnblogs.com/lekko/p/4796062.html）：

/**
 * 网页转图片处理类，使用外部CMD
 */
public class PhantomTools {

    private static final Logger _logger = Logger.getLogger(PhantomTools.class);

    private static final String _tempPath = "D:/data/temp/phantom_";
    private String basePath;
    private static final String _shellCommand1 = "phantomjs ";
    private static final String _shellCommand2 = "rasterize.js ";

    private String _file;
    private String _size;

    /**
     * 构造截图类
     * @param hash 用于临时文件的目录唯一化
     * @param basePath phantomJs所在路径
     */
    public PhantomTools(int hash, String basePath) {
        _file = _tempPath + hash + ".png";
        this.basePath = basePath;
    }

    /**
     * 构造截图类
     * @param hash 用于临时文件的目录唯一化
     * @param size 图片的大小，如800px*600px（此时高度会裁切），或800px（此时 高度最少=宽度*9/16，高度不裁切）
     * @param basePath phantomJs所在路径
     */
    public PhantomTools(int hash, String size, String basePath) {
        _file = _tempPath + hash + ".png";
        if (size != null)
            _size = " " + size;
        this.basePath = basePath;
    }

    /**
     * 将目标网页转为图片字节流
     * @param url 目标网页地址
     * @return 字节流
     */
    public byte[] getByteImg(String url) throws IOException {
        BufferedInputStream in = null;
        ByteArrayOutputStream out = null;
        File file = null;
        byte[] ret = null;
        try {
            if (exeCmd(basePath + _shellCommand1 + basePath + _shellCommand2 + url + " " + _file + (_size != null ? _size : ""))) {
                file = new File(_file);
                if (file.exists()) {
                    out = new ByteArrayOutputStream();
                    byte[] b = new byte[5120];
                    in = new BufferedInputStream(new FileInputStream(file));
                    int n;
                    while ((n = in.read(b, 0, 5120)) != -1) {
                        out.write(b, 0, n);
                    }
                    file.delete();
                    ret = out.toByteArray();
                }
            } else {
                ret = new byte[] {};
            }
        } finally {
            try {
                if (out != null) {
                    out.close();
                }
            } catch (IOException e) {
                _logger.error(e);
            }
            try {
                if (in != null) {
                    in.close();
                }
            } catch (IOException e) {
                _logger.error(e);
            }
            if (file != null && file.exists()) {
                file.delete();
            }
        }
        return ret;
    }

    /**
     * 执行CMD命令
     */
    private static boolean exeCmd(String commandStr) {
        BufferedReader br = null;
        try {
            Process p = Runtime.getRuntime().exec(commandStr);
            if (p.waitFor() != 0 && p.exitValue() == 1) {
                return false;
            }
        } catch (Exception e) {
            _logger.error(e);
        } finally {
            if (br != null) {
                try {
                    br.close();
                } catch (Exception e) {
                    _logger.error(e);
                }
            }
        }
        return true;
    }
}