org.kuali.maven.plugins.HttpInspector Maven / Gradle / Ivy

Go to download
package org.kuali.maven.plugins;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.httpclient.DefaultHttpMethodRetryHandler;
import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.HttpMethod;
import org.apache.commons.httpclient.HttpMethodRetryHandler;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.commons.httpclient.params.HttpClientParams;
import org.apache.commons.httpclient.params.HttpMethodParams;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class HttpInspector {
	private final Logger logger = LoggerFactory.getLogger(HttpInspector.class);
	List successCodes = new ArrayList();
	int requestTimeout = 3000;
	int sleepInterval = 3000;
	int timeout = 180;

	protected boolean isSuccess(int resultCode) {
		for (int successCode : successCodes) {
			if (resultCode == successCode) {
				return true;
			}
		}
		return false;
	}

	protected String getMsg(String msg) {
		return getMsg(msg, -1);
	}

	protected String getMsg(String msg, long l) {
		StringBuilder sb = new StringBuilder();
		sb.append(msg);
		if (l == -1) {
			return sb.toString();
		}
		sb.append(" - (Timeout in " + l + "s)");
		return sb.toString();
	}

	public boolean wait(String url) {
		HttpClient client = getHttpClient();
		long now = System.currentTimeMillis();
		long end = now + (timeout * 1000);
		logger.info(getMsg("Determining status for '" + url + "'"));
		for (;;) {
			long secondsRemaining = (long) Math.ceil((end - System.currentTimeMillis()) / 1000D);
			boolean success = doRequest(client, url, secondsRemaining);
			if (success) {
				return true;
			}
			sleep(sleepInterval);
			if (System.currentTimeMillis() > end) {
				logger.info("Timed out waiting for response from '" + url + "'");
				return false;
			}
		}
	}

	protected HttpClient getHttpClient() {
		HttpClient client = new HttpClient();
		HttpClientParams clientParams = client.getParams();
		HttpMethodRetryHandler retryHandler = new DefaultHttpMethodRetryHandler(0, false);
		clientParams.setParameter(HttpMethodParams.RETRY_HANDLER, retryHandler);
		clientParams.setParameter(HttpMethodParams.SO_TIMEOUT, requestTimeout);
		return client;
	}

	protected boolean doRequest(HttpClient client, String url, long secondsRemaining) {
		try {
			HttpMethod method = new GetMethod(url);
			client.executeMethod(method);
			int statusCode = method.getStatusCode();
			String statusText = method.getStatusText();
			boolean success = isSuccess(statusCode);
			if (success) {
				logger.info(getMsg("Status for '" + url + "' is '" + statusCode + ":" + statusText + "'"));
				return true;
			} else {
				logger.info(getMsg("Status for '" + url + "' is '" + statusCode + ":" + statusText + "'",
						secondsRemaining));
			}
		} catch (IOException e) {
			logger.info(getMsg("Status for '" + url + "' is '" + e.getMessage() + "'", secondsRemaining));
		}
		return false;
	}

	protected void sleep(long millis) {
		try {
			Thread.sleep(millis);
		} catch (InterruptedException e) {
			throw new RuntimeException(e);
		}
	}

	public int getRequestTimeout() {
		return requestTimeout;
	}

	public void setRequestTimeout(int requestTimeout) {
		this.requestTimeout = requestTimeout;
	}

	public int getSleepInterval() {
		return sleepInterval;
	}

	public void setSleepInterval(int sleepInterval) {
		this.sleepInterval = sleepInterval;
	}

	public int getTimeout() {
		return timeout;
	}

	public void setTimeout(int waitTimeout) {
		this.timeout = waitTimeout;
	}

	public Logger getLogger() {
		return logger;
	}

	public List getSuccessCodes() {
		return successCodes;
	}

	public void setSuccessCodes(List successCodes) {
		this.successCodes = successCodes;
	}
}