Java利用Phantomjs實(shí)現(xiàn)生成圖片的方法

"use strict";
var page = require('webpage').create(),
  system = require('system'),
  address, output, size;

if (system.args.length < 3 || system.args.length > 5) {
  console.log('Usage: rasterize.js URL filename [paperwidth*paperheight|paperformat] [zoom]');
  console.log(' paper (pdf output) examples: "5in*7.5in", "10cm*20cm", "A4", "Letter"');
  console.log(' image (png/jpg output) examples: "1920px" entire page, window width 1920px');
  console.log('                  "800px*600px" window, clipped to 800x600');
  phantom.exit(1);
} else {
  address = system.args[1];
  output = system.args[2];
  page.viewportSize = { width: 800, height: 200 };
  if (system.args.length > 3 && system.args[2].substr(-4) === ".pdf") {
    size = system.args[3].split('*');
    page.paperSize = size.length === 2 &#63; { width: size[0], height: size[1], margin: '0px' }
                      : { format: system.args[3], orientation: 'portrait', margin: '1cm' };
  } else if (system.args.length > 3 && system.args[3].substr(-2) === "px") {
    size = system.args[3].split('*');
    if (size.length === 2) {
      var pageWidth = parseInt(size[0].substr(0,size[0].indexOf("px")), 10);
      var pageHeight = parseInt(size[1].substr(0,size[1].indexOf("px")), 10);
      page.viewportSize = { width: pageWidth, height: pageHeight };
      page.clipRect = { top: 0, left: 0, width: pageWidth, height: pageHeight };
    } else {
      var pageWidth = parseInt(system.args[3].substr(0,system.args[3].indexOf("px")), 10);
      var pageHeight = parseInt(pageWidth * 3/4, 10); // it's as good an assumption as any
      page.viewportSize = { width: pageWidth, height: pageHeight };
    }
  }
  if (system.args.length > 4) {
    page.zoomFactor = system.args[4];
  }
  page.open(address, function (status) {
    if (status !== 'success') {
      console.log('Unable to load the address!');
      phantom.exit(1);
    } else {
      window.setTimeout(function () {
        page.render(output);
        phantom.exit();
      }, 200);
    }
  });
}

package test;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.UUID;

/**
 * 網(wǎng)頁轉(zhuǎn)圖片處理類，使用外部CMD
 *
 * @author lekkoli
 */
@Slf4j
public class PhantomTools {

  /**
   * 可執(zhí)行文件phantomjs.exe路徑
   */
  private final String phantomjsPath;
  /**
   * 快照?qǐng)D生成JS路徑
   */
  private final String rasterizePath;
  /**
   * 臨時(shí)圖片前綴
   */
  private static final String FILE_PREFIX = "TIG-AE-";
  /**
   * 臨時(shí)圖片后綴
   */
  private static final String FILE_SUFFIX = ".jpg";

  /**
   * 構(gòu)造參數(shù)
   * 獲取phantomjs路徑
   */
  public PhantomTools() {
    String bootPath = new File(this.getClass().getResource("/").getPath()).getPath();
    phantomjsPath = String.join(File.separator, bootPath, "phantomjs-2.1.1-windows", "bin", "phantomjs");
    rasterizePath = String.join(File.separator, bootPath, "phantomjs-2.1.1-windows", "examples", "rasterize.js");
  }

  /**
   * url 中需要轉(zhuǎn)義的字符
   * 1. + URL 中+號(hào)表示空格 %2B
   * 2. 空格 URL中的空格可以用+號(hào)或者編碼 %20
   * 3. / 分隔目錄和子目錄 %2F
   * 4. ? 分隔實(shí)際的 URL 和參數(shù) %3F
   * 5. % 指定特殊字符 %25
   * 6. # 表示書簽 %23
   * 7. & URL 中指定的參數(shù)間的分隔符 %26
   * 8. = URL 中指定參數(shù)的值 %3D
   *
   * @param url 需要轉(zhuǎn)義的URL
   * @return 轉(zhuǎn)義后的URL
   */
  public String parseUrl(String url) {
    String parsedUrl = StringUtils.replace(url, "&", "%26");
    log.info("[解析后的URL：{}]", parsedUrl);
    return parsedUrl;
  }

  /**
   * 根據(jù)URL生成指定fileName的字節(jié)數(shù)組
   *
   * @param url 請(qǐng)求URL
   * @return 圖片字節(jié)數(shù)組
   */
  public byte[] create(String url) {
    return create(url, null);
  }

  /**
   * 根據(jù)URL生成指定fileName的字節(jié)數(shù)組
   *
   * @param url 請(qǐng)求URL
   * @param size 指定圖片尺寸，例如：1000px*800px
   * @return 圖片字節(jié)數(shù)組
   */
  public byte[] create(String url, String size) {
    // 服務(wù)器文件存放地址
    String filePath = FileUtils.getTempDirectoryPath() + FILE_PREFIX + UUID.randomUUID().toString() + FILE_SUFFIX;
    try {
      // 執(zhí)行快照命令
      String command = String.join(StringUtils.SPACE, phantomjsPath, rasterizePath, url, filePath, size);
      log.info("[執(zhí)行命令：{}]", command);
      // 執(zhí)行命令操作
      Process process = Runtime.getRuntime().exec(command);
      // 一直掛起，直到子進(jìn)程執(zhí)行結(jié)束，返回值0表示正常退出
      if (process.waitFor() != 0) {
        log.error("[執(zhí)行本地Command命令失敗] [Command：{}]", command);
        return new byte[0];
      }
      // 判斷生成的圖片是否存在
      File file = FileUtils.getFile(filePath);
      if (!file.exists()) {
        log.error("[本地文件\"{}\"不存在]", file.getName());
        return new byte[0];
      }
      // 將快照?qǐng)D片生成字節(jié)數(shù)組
      byte[] bytes = IOUtils.toByteArray(new FileInputStream(file));
      log.info("[圖片生成結(jié)束] [圖片大?。簕}KB]", bytes.length / 1024);
      return bytes;
    } catch (IOException | InterruptedException e) {
      log.error("[圖片生成失敗]", e);
    } finally {
      FileUtils.deleteQuietly(FileUtils.getFile(filePath));
    }
    return new byte[0];
  }
}

public Process exec(String[] cmdarray, String[] envp, File dir)
  throws IOException {
  return new ProcessBuilder(cmdarray)
    .environment(envp)
    .directory(dir)
    .start();
}

public static void main(String[] arg) throws IOException {
  String url = "https://www.cnblogs.com/ason-wxs/";
  PhantomTools phantomTools = new PhantomTools();
  String parsedUrl = phantomTools.parseUrl(url);
  byte[] byteImg = phantomTools.create(parsedUrl);
  File descFile = new File(FileUtils.getTempDirectoryPath() + "test.png");
  FileUtils.touch(descFile);
  FileUtils.writeByteArrayToFile(descFile, byteImg);
}