io.tracee.contextlogger.connector.HttpConnector Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tracee-context-logger-http-connector Show documentation
Please refer to https://github.com/holisticon/tracee.
The newest version!
package io.tracee.contextlogger.connector;

import com.ning.http.client.AsyncCompletionHandler;
import com.ning.http.client.AsyncHttpClient;
import com.ning.http.client.AsyncHttpClientConfig;
import com.ning.http.client.AsyncHttpClientConfig.Builder;
import com.ning.http.client.ProxyServer;
import com.ning.http.client.Realm;
import com.ning.http.client.Realm.AuthScheme;
import com.ning.http.client.Response;
import io.tracee.Tracee;
import io.tracee.TraceeLogger;
import io.tracee.TraceeLoggerFactory;
import io.tracee.contextlogger.Connector;

import java.io.IOException;
import java.util.Map;

/**
 * A Connector to send error reports via http.
 * Created by Tobias Gindler, holisticon AG on 07.02.14.
 */
public class HttpConnector implements Connector {

	public static final String PROPERTY_URL = "url";
	public static final String PROPERTY_BASIC_AUTH_USER = "basicAuth.user";
	public static final String PROPERTY_BASIC_AUTH_PASSWORD = "basicAuth.password";
	public static final String PROPERTY_PROXY_HOST = "proxy.host";
	public static final String PROPERTY_PROXY_PORT = "proxy.port";
	public static final String PROPERTY_PROXY_USER = "proxy.user";
	public static final String PROPERTY_PROXY_PASSWORD = "proxy.password";
	public static final String PROPERTY_REQUEST_TIMEOUT = "request.timoutInMs";

	public static final int DEFAULT_REQUEST_TIMEOUT = 10000;
	public static final int DEFAULT_REQUEST_IDLE_TIMEOUT = 500;

	private final TraceeLogger logger;
	private final AsyncHttpClientProvider asyncHttpClientProvider;

	public HttpConnector() {
		this(Tracee.getBackend().getLoggerFactory(), new SimpleAsyncHttpProvider());
	}

	HttpConnector(TraceeLoggerFactory loggerFactory, AsyncHttpClientProvider asyncHttpClientProvider) {
		this.asyncHttpClientProvider = asyncHttpClientProvider;
		logger = loggerFactory.getLogger(HttpConnector.class);
	}

	private String url;

	private Builder builder;

	@Override
	public final void init(final Map properties) {
		// map properties
		url = properties.get(PROPERTY_URL);
		final String basicAuthUser = properties.get(PROPERTY_BASIC_AUTH_USER);
		final String basicAuthPassword = properties.get(PROPERTY_BASIC_AUTH_PASSWORD);
		final String proxyHost = properties.get(PROPERTY_PROXY_HOST);
		final String proxyUser = properties.get(PROPERTY_PROXY_USER);
		final String proxyPassword = properties.get(PROPERTY_PROXY_PASSWORD);

		final Integer proxyPort = convertStringToInt(properties.get(PROPERTY_PROXY_PORT), null);

		final Integer requestTimeoutInMs = convertStringToInt(properties.get(PROPERTY_REQUEST_TIMEOUT), DEFAULT_REQUEST_TIMEOUT);

		builder = new AsyncHttpClientConfig.Builder();

		if (basicAuthUser != null && basicAuthPassword != null) {
			final Realm realm = new Realm.RealmBuilder().setPrincipal(basicAuthUser).setPassword(basicAuthPassword).setUsePreemptiveAuth(true)
					.setScheme(AuthScheme.BASIC).build();

			builder.setRealm(realm);
		}

		// Add Proxy Support - use authentication if user and password properties are set
		if (proxyHost != null && proxyPort != null) {
			final ProxyServer proxy;
			if (proxyUser != null && proxyPassword != null) {
				proxy = new ProxyServer(proxyHost, proxyPort, proxyUser, proxyPassword);
			} else {
				proxy = new ProxyServer(proxyHost, proxyPort);
			}

			builder.setProxyServer(proxy);
		}

		builder.setRequestTimeoutInMs(requestTimeoutInMs);
		builder.setIdleConnectionTimeoutInMs(DEFAULT_REQUEST_IDLE_TIMEOUT);

	}

	@Override
	public final void sendErrorReport(final String json) {

		final String targetUrl = this.url;

		try {
			@SuppressWarnings("resource")
			final AsyncHttpClient asyncHttpClient = asyncHttpClientProvider.provideHttpClient(builder.build());
			asyncHttpClient.preparePost(targetUrl).setBody(json).setHeader("Content-type", "text/json;charset=utf-8").setBodyEncoding("UTF-8")
					.execute(new AsyncCompletionHandler() {

						@Override
						public Response onCompleted(final Response response) throws Exception {
							logger.info("Error report send via HTTP to '" + targetUrl + "'");
							return response;
						}

						@Override
						public void onThrowable(final Throwable t) {
							// Something wrong happened.
							logger.error("An error occurred while sending the error report via HTTP to '" + targetUrl + "'", t);
						}
					});
		} catch (final IOException e) {
			logger.error("An error occurred while sending the error report via HTTP to '" + this.url + "'", e);
		}

	}

	final Integer convertStringToInt(final String value, final Integer defaultValue) {
		try {
			if (value != null) {
				return Integer.valueOf(value);
			}
		} catch (final NumberFormatException e) {
			// ignore
		}
		return defaultValue;
	}
}