#appengine javaでURLFetchのdeadlineを自動的に設定する

AppEngine/JavaでのURLFetchサービスのdeadlineは5秒だと思うけど、全てのURLFetch#fetchのdeadlineを自動的に10秒に設定するためのApiProxy.Delegateの実装を作成してみた。

	import java.util.concurrent.Future;
	import java.util.logging.Level;
	import java.util.logging.Logger;
	import org.apache.commons.lang.StringUtils;
	import com.google.appengine.api.urlfetch.URLFetchServicePb.URLFetchRequest;
	import com.google.appengine.repackaged.com.google.protobuf.InvalidProtocolBufferException;
	import com.google.apphosting.api.ApiProxy;
	import com.google.apphosting.api.ApiProxy.ApiConfig;
	import com.google.apphosting.api.ApiProxy.ApiProxyException;
	import com.google.apphosting.api.ApiProxy.Delegate;
	import com.google.apphosting.api.ApiProxy.Environment;
	import com.google.apphosting.api.ApiProxy.LogRecord;
	/**
	* URLFetchサービスへのDeadlineを強制的に10秒に設定する{@link ApiProxy.Delegate}.
	* @author shin1ogawa
	*/
	public class IncreaseURLFetchDeadlineDelegate implements ApiProxy.Delegate<Environment> {
	static final Logger logger = Logger.getLogger(IncreaseURLFetchDeadlineDelegate.class.getName());
	final ApiProxy.Delegate<Environment> delegate;
	/**
	* the constructor.
	* @param delegate
	* @category constructor
	*/
	public IncreaseURLFetchDeadlineDelegate(Delegate<Environment> delegate) {
	this.delegate = delegate;
	}
	@Override
	public void log(Environment env, LogRecord logRecord) {
	delegate.log(env, logRecord);
	}
	@Override
	public Future<byte[]> makeAsyncCall(Environment env, String service, String method,
	byte[] requestBytes, ApiConfig config) {
	if (StringUtils.equalsIgnoreCase("urlfetch", service) == false) {
	return delegate.makeAsyncCall(env, service, method, requestBytes, config);
	}
	if (StringUtils.equalsIgnoreCase("fetch", method) == false) {
	return delegate.makeAsyncCall(env, service, method, requestBytes, config);
	}
	try {
	config.setDeadlineInSeconds(10000.0);
	byte[] newRequestBytes = increaseDeadline(requestBytes);
	return delegate.makeAsyncCall(env, service, method, newRequestBytes, config);
	} catch (InvalidProtocolBufferException e) {
	throw new RuntimeException(e);
	}
	}
	@Override
	public byte[] makeSyncCall(Environment env, String service, String method, byte[] requestBytes)
	throws ApiProxyException {
	if (StringUtils.equalsIgnoreCase("urlfetch", service) == false) {
	return delegate.makeSyncCall(env, service, method, requestBytes);
	}
	if (StringUtils.equalsIgnoreCase("fetch", method) == false) {
	return delegate.makeSyncCall(env, service, method, requestBytes);
	}
	try {
	byte[] newRequestBytes = increaseDeadline(requestBytes);
	return delegate.makeSyncCall(env, service, method, newRequestBytes);
	} catch (InvalidProtocolBufferException e) {
	throw new RuntimeException(e);
	}
	}
	static byte[] increaseDeadline(byte[] orignialRequestBytes)
	throws InvalidProtocolBufferException {
	URLFetchRequest requestPB = URLFetchRequest.parseFrom(orignialRequestBytes);
	URLFetchRequest newRequestPB = requestPB.toBuilder().setDeadline(10000.0).build();
	byte[] newRequestBytes = newRequestPB.toByteArray();
	if (logger.isLoggable(Level.FINEST)) {
	logger.log(Level.FINEST, "newRequest=" + newRequestPB);
	}
	return newRequestBytes;
	}
	}

view raw IncreaseURLFetchDeadlineDelegate.java hosted with ❤ by GitHub

仕組みはとても簡単、

1. URLFetch#fetchをフックして、サービスへリクエストされるバイト配列からURLFetchRequestに組み立てなおす。
2. 組み立てなおしたURLFetchRequestのURLFetchRequest#toBuilder()を使って、新たにURLFetchRequestを作成し、それに対してURLFetchRequest.Builder#setDeadline()する。
3. setDeadline()したオブジェクトをバイト配列に変換し、
4. サービスには何食わぬ顔でそのバイト配列を送りつける。

gdata-apiを使うときなんかは5秒じゃ済まない事も多いので、Slim3なら基底クラスとしてGDataAPIControllerBaseとか作って、setUp()/tearDown()でこのDelegateのつけはずしをしたりして使います。