表示しているページのサイズ（画面キャプチャのサイズ、もしくは類似のサイズ）をselenumの何かしらのクラスで取得する方法

Question

seleniumで、巨大なページの画面キャプチャを取得するとOutOfMemoryが発生してしまいます。
短時間で大量のJavaヒープを消費するためOutOfMemoryErrorになっているようなので、
巨大なページであればキャプチャを取らないように条件分岐させたいと思っています。

** そこで表示したページのサイズ（画面キャプチャのサイズ、もしくは類似のサイズ）をselenumの何かしらのクラスで取得する方法**
をご存知な方いらっしゃいましたらお教え頂きたいです。

▼具体的な問題▼
具体的には、
http://backyard.imjp.co.jp/static/feed/atom.xml
を以下のコードで画面キャプチャを採取しようとするとOutOfMemoryが発生します。

▼キャプチャ採取のコード▼

				//取得したURLのスクリーンショット作成
				File scrFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
				FileUtils.copyFile(scrFile, new File(scrFileName));

▼発生している例外▼

Exception in thread "main" java.lang.OutOfMemoryError: Java heap space
	at java.lang.StringCoding.decode(StringCoding.java:215)
	at java.lang.String.<init>(String.java:451)
	at java.lang.String.<init>(String.java:503)
	at org.openqa.selenium.remote.http.HttpMessage.getContentString(HttpMessage.java:110)
	at org.openqa.selenium.remote.http.HttpResponse.getContentString(HttpResponse.java:1)
	at org.openqa.selenium.remote.http.JsonHttpResponseCodec.decode(JsonHttpResponseCodec.java:76)
	at org.openqa.selenium.remote.HttpCommandExecutor.execute(HttpCommandExecutor.java:137)
	at org.openqa.selenium.firefox.internal.NewProfileExtensionConnection.execute(NewProfileExtensionConnection.java:170)
	at org.openqa.selenium.firefox.FirefoxDriver$LazyCommandExecutor.execute(FirefoxDriver.java:393)
	at org.openqa.selenium.remote.RemoteWebDriver.execute(RemoteWebDriver.java:568)
	at org.openqa.selenium.remote.RemoteWebDriver.execute(RemoteWebDriver.java:618)
	at org.openqa.selenium.firefox.FirefoxDriver.getScreenshotAs(FirefoxDriver.java:354)
	at sample.SandBox.main(SandBox.java:168)

▼試してみたけどダメだったこと▼
0. JVMの起動パラメータで、-Xms1024m、-Xmx1024mにした。
0. ファイルコピーをせずに一時ディレクトリでのキャプチャ採取のみにした。(FileUtils.copyFile(scrFile, new File(scrFileName)); をコメントアウト)

▼やりたいこと。質問▼
短時間で大量のJavaヒープを消費するためOutOfMemoryErrorになっているようなので、
巨大なページの画面キャプチャ採取をあきらめ、
巨大なページであればキャプチャを取らないように条件分岐させたいと思っています。

** そこで表示したページのサイズ（画面キャプチャのサイズ、もしくは類似のサイズ）をselenumの何かしらのクラスで取得する方法**
をご存知の方はお教え頂きたいです。

Answer 1

skyfishさんに伺った以下の方法でJSを動かしてスクロールするサイズを取得することで、
サイズの大きいページを判定することができました。
（実際には若干お教え頂いたコードとは異なってましたがそこは推測できる範囲なので大丈夫でしたｗ）

//skyfishさんに伺った方法
int scrollLength = Integer.parseInt(((JavascriptExecutor) driver).executeScript("return document.body.scrollHeight").toString());

---

ですが、JSを実行しなくても、selenium自体（org.openqa.selenium.Dimension）でスクロールした時のサイズが以下取れることがわかりました。

//selenium自体（org.openqa.selenium.Dimension）を使う方法
Dimension rootDomSize = driver.findElement(By.tagName("body")).getSize();
int height = rootDomSize.height;
int width = rootDomSize.width;

skyfishさんにお教え頂いた方法とorg.openqa.selenium.Dimensionを使う方法と値を比較してみましたが以下で一緒でした。

//2つの方法の値比較
	int scrollLength = Integer.parseInt(((JavascriptExecutor) driver).executeScript("return document.body.scrollHeight").toString());
	Dimension rootDomSize = driver.findElement(By.tagName("body")).getSize();
	System.out.println("height:"+rootDomSize.height+" width:"+rootDomSize.width+" scrollLength:"+scrollLength);

結果：
height:38780 width:1407 scrollLength:38780

中身は同じことしてるのかもしれませんが、一応JS呼び出すよりもJava内で実行できたほうがいいかなと思い、後者を使って以下のようにすることにしました。

				if (driver instanceof JavascriptExecutor) {
					Dimension rootDomSize = driver.findElement(By.tagName("body")).getSize();
					//長さが2万px以下の時はそのままスクリーンショットを撮る
					if(rootDomSize.height  >= 20000){

						//取得したURLのスクリーンショット作成
						File scrFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
						FileUtils.copyFile(scrFile, new File(scrFileName));

					}
				}

ただ、ちょっとこれだと逃げてる感があるので、
サイズの大きい画面では、画面サイズを小さくしてキャプチャを採取することにしました。

		//ウィンドウサイズを小さくする
		driver.manage().window().setSize(new Dimension(100,100));

結果は、成功しました！
ですが、画面サイズを元に戻さないと次のキャプチャ採取の時に、
小さいままになってしまうので、その辺りがちょっと面倒。。。

今度は、サイズの大きい画面では、
スクロールしないで表示されているところだけキャプチャするようにできないか試したく、
また質問しちゃってます↓
FirefoxDriverでスクロールさせないでキャプチャする方法

ということでアドバイスありがとうございました！！

Answer 2

SeleniumというよりWebDriver側の機能ですが、
WebDriverに向けてjavascriptのコードを叩いて値を返してもらう事で実装できます。

int scrollLength = Integer.parseInt(webdriver.execute_script("return document.body.scrollHeight"));
System.out.println(scrollLength);
if(scrollLength < 100000){
 //長さが10万以下の時はスクリーンショットを撮る
}

上記のようにして得られたscrollLengthの値がいくら以下の時はSSを取る。
みたいに実装すると良いのではないでしょうか。

面白い内容だったので動くコードをgithubに残しました。rubySeleniumですが参考までに
https://github.com/Lyptica/seleniumScrollTest/blob/master/seleniumScroll.rb

参考:
Efficient method to scroll though pages using Selenium
http://stackoverflow.com/questions/19803963/efficient-method-to-scroll-though-pages-using-selenium

Comments

[tasuku]

おー。
webdriverにjsを実行させるメソッドがあるんですね（webdriver.execute_script）
本題とはちがったことが学べて得した気分です。

早速試してみます。

PS：Rubyも勉強しなくては。。。ｗ