All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.northernwall.hadrian.service.helper.HostDetailsHelper Maven / Gradle / Ivy

/*
 * Copyright 2015 Richard Thurston.
 *
 * 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.northernwall.hadrian.service.helper;

import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import com.northernwall.hadrian.Const;
import com.northernwall.hadrian.domain.Host;
import com.northernwall.hadrian.parameters.ParameterChangeListener;
import com.northernwall.hadrian.parameters.Parameters;
import com.northernwall.hadrian.service.dao.GetHostDetailsData;
import com.northernwall.hadrian.service.dao.GetPairData;
import com.squareup.okhttp.OkHttpClient;
import com.squareup.okhttp.Request;
import com.squareup.okhttp.Response;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class HostDetailsHelper implements ParameterChangeListener {
    private final static Logger logger = LoggerFactory.getLogger(HostDetailsHelper.class);

    private final OkHttpClient client;
    private final Parameters parameters;
    private final JsonParser parser;
    private String urlTemplate;
    private final List attributes;

    public HostDetailsHelper(OkHttpClient client, Parameters parameters) {
        this.client = client;
        this.parameters = parameters;
        this.parser = new JsonParser();
        this.attributes = new LinkedList<>();
        load();
        parameters.registerChangeListener(this);
    }

    @Override
    public void onChange(List keys) {
        load();
    }
    
    private void load() {
        urlTemplate = parameters.getString(Const.HOST_DETAILS_URL, null);
        String temp = parameters.getString(Const.HOST_DETAILS_ATTRIBUTES, null);
        attributes.clear();
        if (temp != null && !temp.isEmpty()) {
            String[] parts = temp.split(",");
            for (String part : parts) {
                part = part.trim();
                if (!part.isEmpty()) {
                    attributes.add(part);
                }
            }
        }
    }
    
    public GetHostDetailsData getDetails(Host host) {
        GetHostDetailsData details = new GetHostDetailsData();
        if (urlTemplate != null) {
            String url = urlTemplate.replace(Const.HOST, host.getHostName());
            Request httpRequest = new Request.Builder().url(url).build();
            try {
                Response resp = client.newCall(httpRequest).execute();
                if (resp.isSuccessful()) {
                    Reader reader = new InputStreamReader(resp.body().byteStream());

                    JsonElement jsonElement = parser.parse(reader);
                    if (jsonElement.isJsonObject()) {
                        JsonObject jsonObject = jsonElement.getAsJsonObject();
                        for (Map.Entry entry : jsonObject.entrySet()) {
                            if (entry.getValue().isJsonPrimitive()) {
                                addPair(entry.getKey(), entry.getValue().getAsString(), details);
                            } else if (entry.getValue().isJsonArray()) {
                                StringBuffer buffer = null;
                                JsonArray jsonArray = entry.getValue().getAsJsonArray();
                                for (int i = 0; i < jsonArray.size(); i++) {
                                    JsonElement arrayElement = jsonArray.get(i);
                                    if (arrayElement.isJsonPrimitive()) {
                                        if (buffer == null) {
                                            buffer = new StringBuffer(arrayElement.getAsString());
                                        } else {
                                            buffer.append(", ");
                                            buffer.append(arrayElement.getAsString());
                                        }
                                    }
                                }
                                if (buffer != null) {
                                    addPair(entry.getKey(), buffer.toString(), details);
                                }
                            }
                        }
                    }
                } else {
                    BufferedReader reader = new BufferedReader(new InputStreamReader(resp.body().byteStream()));
                    addPair("Error", reader.readLine(), details);
                }
            } catch (IOException ex) {
                logger.warn("Error while getting secondary host details for {}, error {}", host.getHostName(), ex.getMessage());
                addPair("Error", "Read failure", details);
            }
        }
        return details;
    }

    private void addPair(String label, String value, GetHostDetailsData details) {
        if (label == null || label.isEmpty() || value == null || value.isEmpty()) {
            return;
        }
        if (attributes.isEmpty() || attributes.contains(label)) {
            label = label.replace("-", " ").replace("_", " ");
            if (details.left.size() == details.right.size()) {
                details.left.add(new GetPairData(label, value));
            } else {
                details.right.add(new GetPairData(label, value));
            }
        }
    }

}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy