java selenium保存canvas为图片

关于canvas保存成为图片基本都不能和selenium进行很好的配合，canvas保存成图片应用较多的是图片验证或滑动验证的图片保存,下面是利用selenium直接将canvas保存成图片的代码

  public static void main(String[] args) {
    // web驱动的存放路径 
    String webDriverDir = "web驱动的存放路径 ";
	// 加载驱动
	System.setProperty("webdriver.chrome.driver", webDriverDir);

	// 设置浏览器参数
	ChromeOptions options = new ChromeOptions();
	// 创建驱动对象
	WebDriver driver = new ChromeDriver(options);
	// 打开链接
	driver.get("链接地址");
	// 查找画布这里用的是tagName用其他的id或className也完全可以
	WebElement canvas = driver .findElement(By.tagName("canvas"));
	// 这句到代码是关键执行完后返回的图片的base64 字符串编码
	Object base64 = ((JavascriptExecutor) driver)
			.executeScript("return arguments[0].toDataURL('image/png').substring(21);", canvas);
	String base64Str = base64.toString();
	// 将base64把字符串装换成图片
	generateImage(base64Str , "要存放图片的路径");
}

   /**
	 * @Description: 将base64编码字符串转换为图片
	 * @Author:
	 * @CreateTime:
	 * @param imgStr base64编码字符串
	 * @param path   图片路径-具体到文件
	 * @return
	 */
	public static boolean generateImage(String imgStr, String path) {
		if (imgStr == null)
			return false;
		// 解密
		try {
			Decoder decoder = java.util.Base64.getDecoder();
			byte[] b = decoder.decode(imgStr);
			// 处理数据
			for (int i = 0; i < b.length; ++i) {
				if (b[i] < 0) {
					b[i] += 256;
				}
			}
			OutputStream out = new FileOutputStream(path);
			out.write(b);
			out.flush();
			out.close();
			return true;
		} catch (IOException e) {
			return false;
		}
	}