com.elastic.support.diagnostics.RestModule.java Source code

Introduction

Here is the source code for com.elastic.support.diagnostics.RestModule.java

Source

package com.elastic.support.diagnostics;

import org.apache.commons.compress.utils.IOUtils;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.utils.HttpClientUtils;
import org.apache.http.conn.HttpHostConnectException;
import org.apache.http.util.EntityUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.FileOutputStream;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.concurrent.TimeUnit;

public class RestModule {

    HttpClient client;

    private final static Logger logger = LoggerFactory.getLogger(RestModule.class);

    public RestModule(HttpClient client) {
        this.client = client;
    }

    public String submitRequest(String url) {

        HttpResponse response = null;
        String result = "";
        try {
            HttpGet httpget = new HttpGet(url);
            response = client.execute(httpget);
            try {
                org.apache.http.HttpEntity entity = response.getEntity();
                if (entity != null) {
                    checkResponseCode(url, response);
                    result = EntityUtils.toString(entity);
                }
            } catch (Exception e) {
                logger.error("Error handling response for " + url, e);
            } finally {
                HttpClientUtils.closeQuietly(response);
            }
        } catch (HttpHostConnectException e) {
            throw new RuntimeException("Error connecting to host " + url, e);
        } catch (Exception e) {
            if (e.getMessage().contains("401 Unauthorized")) {
                logger.error("Auth failure", e);
                throw new RuntimeException("Authentication failure: invalid login credentials.", e);
            } else {
                logger.error("Diagnostic query: " + url + "failed.", e);
            }
        }

        return result;

    }

    public void submitRequest(String url, String queryName, String destination) {

        HttpResponse response = null;
        InputStream responseStream = null;

        try {
            FileOutputStream fos = new FileOutputStream(destination);
            HttpGet httpget = new HttpGet(url);
            response = client.execute(httpget);
            try {
                org.apache.http.HttpEntity entity = response.getEntity();
                if (entity != null) {
                    checkResponseCode(queryName, response);
                    responseStream = entity.getContent();
                    IOUtils.copy(responseStream, fos);
                } else {
                    Files.write(Paths.get(destination), ("No results for:" + queryName).getBytes());
                }
            } catch (Exception e) {
                logger.error("Error writing response for " + queryName + " to disk.", e);
            } finally {
                HttpClientUtils.closeQuietly(response);
                Thread.sleep(2000);
            }
        } catch (Exception e) {
            if (url.contains("_license")) {
                logger.info("There were no licenses installed");
            } else if (e.getMessage().contains("401 Unauthorized")) {
                logger.error("Auth failure", e);
                throw new RuntimeException("Authentication failure: invalid login credentials.", e);
            } else {
                logger.error("Diagnostic query: " + queryName + "failed.", e);
            }
        }

        logger.info("Diagnostic query: " + queryName + " was retrieved and saved to disk.");

    }

    private void checkResponseCode(String name, HttpResponse response) {

        int code = response.getStatusLine().getStatusCode();
        if (code != 200) {
            String msg = response.getStatusLine().getReasonPhrase();
            logger.warn(name + " query returned: " + response.getStatusLine().getStatusCode() + "/" + msg);
            if (code == 401) {
                throw new RuntimeException(
                        "Authentication failure: invalid login credentials. " + code + "/" + msg);
            }
        }
    }

}