com.sematext.ag.sink.AbstractHttpSink.java Source code

Introduction

Here is the source code for com.sematext.ag.sink.AbstractHttpSink.java

Source

/**
 * Copyright 2010 Sematext International
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.sematext.ag.sink;

import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpRequestBase;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.impl.conn.PoolingClientConnectionManager;
import org.apache.log4j.Logger;

import com.sematext.ag.PlayerConfig;
import com.sematext.ag.event.Event;
import com.sematext.ag.exception.InitializationFailedException;
import com.sematext.ag.http.HttpUtils;

/**
 * Abstract base {@link Sink} implementation for sinks using HTTP calls.
 * 
 * @author sematext, http://www.sematext.com/
 */
public abstract class AbstractHttpSink<T extends Event> extends Sink<T> {
    public static final Logger LOG = Logger.getLogger(AbstractHttpSink.class);
    public static final String MAXIMUM_CONNECTIONS_PER_ROUTE_KEY = "abstracthttpsink.max_conn_per_route";
    public static final String MAXIMUM_CONNECTIONS_TOTAL_KEY = "abstracthttpsink.max_conn_total";
    public static final int DEFAULT_MAXIMUM_CONNECTIONS_PER_ROUTE = 100;
    public static final int DEFAULT_MAXIMUM_CONNECTIONS_TOTAL = 100;
    protected static final HttpClient HTTP_CLIENT_INSTANCE;

    static {
        PoolingClientConnectionManager pccm = new PoolingClientConnectionManager();
        pccm.setDefaultMaxPerRoute(DEFAULT_MAXIMUM_CONNECTIONS_PER_ROUTE);
        pccm.setMaxTotal(DEFAULT_MAXIMUM_CONNECTIONS_TOTAL);
        HTTP_CLIENT_INSTANCE = new DefaultHttpClient(pccm);
    }

    /**
     * Executes HTTP request.
     * 
     * @param request
     *          request to execute
     * @return <code>true</code> if request was run successfully, <code>false</code> otherwise
     */
    public boolean execute(HttpRequestBase request) {
        LOG.info("Sending event");
        return HttpUtils.processRequestSilently(HTTP_CLIENT_INSTANCE, request);
    }

    /**
     * (non-Javadoc)
     * 
     * @see com.sematext.ag.sink.Sink#init(com.sematext.ag.PlayerConfig)
     */
    @Override
    public void init(PlayerConfig config) throws InitializationFailedException {
        super.init(config);
        PoolingClientConnectionManager connectionManager = (PoolingClientConnectionManager) HTTP_CLIENT_INSTANCE
                .getConnectionManager();
        if (config.get(MAXIMUM_CONNECTIONS_PER_ROUTE_KEY) != null
                && !config.get(MAXIMUM_CONNECTIONS_PER_ROUTE_KEY).isEmpty()) {
            connectionManager
                    .setDefaultMaxPerRoute(Integer.parseInt(config.get(MAXIMUM_CONNECTIONS_PER_ROUTE_KEY)));
        }
        if (config.get(MAXIMUM_CONNECTIONS_TOTAL_KEY) != null
                && !config.get(MAXIMUM_CONNECTIONS_TOTAL_KEY).isEmpty()) {
            connectionManager.setMaxTotal(Integer.parseInt(config.get(MAXIMUM_CONNECTIONS_TOTAL_KEY)));
        }
    }
}