org.jgroups.ping.kube.Client Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kubernetes Show documentation
Kubernetes PING - Kubernetes
There is a newer version: 0.9.3
/**
 *  Copyright 2014 Red Hat, Inc.
 *
 *  Red Hat licenses this file to you 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 org.jgroups.ping.kube;

import static org.jgroups.ping.common.Utils.openStream;
import static org.jgroups.ping.common.Utils.urlencode;

import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.TreeMap;
import java.util.logging.Level;
import java.util.logging.Logger;

import org.jboss.dmr.ModelNode;
import org.jgroups.ping.common.stream.StreamProvider;

/**
 * @author Ales Justin
 */
public class Client {
    private static final Logger log = Logger.getLogger(Client.class.getName());

    private final String masterUrl;
    private final Map headers;
    private final int connectTimeout;
    private final int readTimeout;
    private final int operationAttempts;
    private final long operationSleep;
    private final StreamProvider streamProvider;
    private final int serverPort;
    private final String info;

    public Client(String masterUrl, Map headers, int connectTimeout, int readTimeout, int operationAttempts, long operationSleep, StreamProvider streamProvider, int serverPort) {
        this.masterUrl = masterUrl;
        this.headers = headers;
        this.connectTimeout = connectTimeout;
        this.readTimeout = readTimeout;
        this.operationAttempts = operationAttempts;
        this.operationSleep = operationSleep;
        this.streamProvider = streamProvider;
        this.serverPort = serverPort;
        Map maskedHeaders = new TreeMap();
        if (headers != null) {
            for (Map.Entry header : headers.entrySet()) {
                String key = header.getKey();
                String value = header.getValue();
                if ("Authorization".equalsIgnoreCase(key) && value != null) {
                    value = "#MASKED:" + value.length() + "#";
                }
                maskedHeaders.put(key, value);
            }
        }
        this.info = String.format("%s[masterUrl=%s, headers=%s, connectTimeout=%s, readTimeout=%s, operationAttempts=%s, operationSleep=%s, streamProvider=%s]",
                getClass().getSimpleName(), masterUrl, maskedHeaders, connectTimeout, readTimeout, operationAttempts, operationSleep, streamProvider);
    }

    public final String info() {
        return info;
    }

    protected ModelNode getNode(String op, String namespace, String labels) throws Exception {
        String url = masterUrl;
        if (namespace != null && namespace.length() > 0) {
            url = url + "/namespaces/" + urlencode(namespace);
        }
        url = url + "/" + op;
        if (labels != null && labels.length() > 0) {
            url = url + "?labelSelector=" + urlencode(labels);
        }
        try (InputStream stream = openStream(url, headers, connectTimeout, readTimeout, operationAttempts, operationSleep, streamProvider)) {
            return ModelNode.fromJSONStream(stream);
        }
    }

    public final List getPods(String namespace, String labels) throws Exception {
        ModelNode root = getNode("pods", namespace, labels);
        List pods = new ArrayList();
        List itemNodes = root.get("items").asList();
        for (ModelNode itemNode : itemNodes) {
            //ModelNode metadataNode = itemNode.get("metadata");
            //String podName = metadataNode.get("name").asString(); // eap-app-1-43wra
            //String podNamespace = metadataNode.get("namespace").asString(); // dward
            ModelNode specNode = itemNode.get("spec");
            //String serviceAccount = specNode.get("serviceAccount").asString(); // default
            //String host = specNode.get("host").asString(); // ce-openshift-rhel-minion-1.lab.eng.brq.redhat.com
            ModelNode statusNode = itemNode.get("status");
            ModelNode phaseNode = statusNode.get("phase");
            if (!phaseNode.isDefined() || !"Running".equals(phaseNode.asString())) {
                continue;
            }
            /* We don't want to filter on the following as that could result in MERGEs instead of JOINs.
            ModelNode conditionsNode = statusNode.get("conditions");
            if (!conditionsNode.isDefined()) {
                continue;
            }
            boolean ready = false;
            List conditions = conditionsNode.asList();
            for (ModelNode condition : conditions) {
                ModelNode conditionTypeNode = condition.get("type");
                ModelNode conditionStatusNode = condition.get("status");
                if (conditionTypeNode.isDefined() && "Ready".equals(conditionTypeNode.asString()) &&
                        conditionStatusNode.isDefined() && "True".equals(conditionStatusNode.asString())) {
                    ready = true;
                    break;
                }
            }
            if (!ready) {
                continue;
            }
            */
            //String hostIP = statusNode.get("hostIP").asString(); // 10.34.75.250
            ModelNode podIPNode = statusNode.get("podIP");
            if (!podIPNode.isDefined()) {
                continue;
            }
            String podIP = podIPNode.asString(); // 10.1.0.169
            Pod pod = new Pod(podIP);
            ModelNode containersNode = specNode.get("containers");
            if (!containersNode.isDefined()) {
                continue;
            }
            List containerNodes = containersNode.asList();
            for (ModelNode containerNode : containerNodes) {
                ModelNode portsNode = containerNode.get("ports");
                if (!portsNode.isDefined()) {
                    continue;
                }
                //String containerName = containerNode.get("name").asString(); // eap-app
                Container container = new Container();
                List portNodes = portsNode.asList();
                for (ModelNode portNode : portNodes) {
                    Optional portName = Optional.empty();

                    ModelNode portNameNode = portNode.get("name");
                    if (portNameNode.isDefined()) {
                        portName = Optional.of(portNameNode.asString());
                    }

                    ModelNode containerPortNode = portNode.get("containerPort");
                    if (!containerPortNode.isDefined()) {
                        continue;
                    }
                    int containerPort = containerPortNode.asInt(); // 8888
                    Port port = new Port(portName, containerPort);
                    container.addPort(port);
                }
                pod.addContainer(container);
            }
            pods.add(pod);
        }
        if (log.isLoggable(Level.FINE)) {
            log.log(Level.FINE, String.format("getPods(%s, %s) = %s", namespace, labels, pods));
        }
        return pods;
    }

    public boolean accept(Container container) {
        return container.getPorts().stream()
                .filter(port -> accept(port))
                .findAny().isPresent();
    }

    public boolean accept(Port port) {
        return serverPort == port.getContainerPort();
    }

}