org.graylog2.inputs.misc.jsonpath.Collector Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of graylog2-inputs Show documentation

Graylog message inputs

There is a newer version: 1.3.4

/**
 * This file is part of Graylog2.
 *
 * Graylog2 is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Graylog2 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Graylog2.  If not, see .
 */
package org.graylog2.inputs.misc.jsonpath;

import com.ning.http.client.AsyncHttpClient;
import com.ning.http.client.Response;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

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

/**
 * @author Lennart Koopmann 
 */
public class Collector {

    private static final Logger LOG = LoggerFactory.getLogger(Collector.class);

    private final String url;
    private final Map headers;
    private final String inputId;

    public Collector(String url,  Map headers, String inputId) {
        this.url = url;
        this.headers = headers;
        this.inputId = inputId;
    }

    public String getJson() throws InterruptedException, ExecutionException, IOException {
        LOG.debug("Getting JSON for JsonPathInput <{}> from [{}].", inputId, url);

        AsyncHttpClient client = new AsyncHttpClient();
        try {
            AsyncHttpClient.BoundRequestBuilder requestBuilder = client.prepareGet(url);

            // Add custom headers if there are some.
            if (headers != null) {
                for (Map.Entry header : headers.entrySet()) {
                    requestBuilder.addHeader(header.getKey(), header.getValue());
                }
            }

            Response r = requestBuilder.execute().get();

            if (r.getStatusCode() != 200) {
                throw new RuntimeException("Expected HTTP status code 200, got " + r.getStatusCode());
            }

            return r.getResponseBody();
        } finally {
            client.close();
        }
    }

}