com.github.imrafaelmerino.kafkacli.ConfigurationQueries Maven / Gradle / Ivy

Go to download
package com.github.imrafaelmerino.kafkacli;

import jsonvalues.JsObj;
import jsonvalues.JsPath;

import java.util.Set;
import java.util.stream.Collectors;

import static com.github.imrafaelmerino.kafkacli.ConfigurationFields.*;

public class ConfigurationQueries {

    public static JsObj getProducerProps(JsObj conf,
                                         String producerName
                                        ) {
        return conf.getObj(JsPath.fromKey(KAFKA)
                                 .key(PRODUCERS)
                                 .key(producerName)
                                 .key(PRODUCER_PROPS));
    }

    public static boolean isChannelUp(JsObj conf,
                                      String name,
                                      KafkaProducers producers
                                     ) {
        String producer = conf.getStr(JsPath.fromKey(CHANNELS)
                                            .key(name)
                                            .key(PRODUCER));
        return producer != null && producers.apply(producer) != null;
    }

    public static boolean existChannel(JsObj conf,
                                       String name
                                      ) {
        return conf.getObj(JsPath.fromKey(CHANNELS))
                   .containsKey(name);
    }

    public static String getChannelsInfo(JsObj conf,
                                         KafkaProducers producers
                                        ) {
        JsObj channels = conf.getObj(JsPath.fromKey(CHANNELS));
        return channels.keySet()
                       .stream()
                       .map(key -> {
                           String producerName = channels.getObj(key)
                                                         .getStr(PRODUCER);
                           return String.format("%-20s %-20s %-20s %-20s",
                                                key,
                                                producerName,
                                                producers.apply(producerName) != null ? "up" : "down",
                                                channels.getObj(key)
                                                        .getStr(TOPIC)
                                               );
                       })
                       .collect(Collectors.joining("\n",
                                                   String.format("%-20s %-20s %-20s %-20s\n%s\n",
                                                                 "Name",
                                                                 "Producer",
                                                                 "Status",
                                                                 "Topic",
                                                                 "--------------------------------------------------------------------------------"),
                                                   ""));
    }

    static Set getProducers(final JsObj conf) {
        JsObj producers = conf.getObj(JsPath.fromKey(KAFKA)
                                            .key(PRODUCERS));
        return producers.keySet();

    }

    static Set getConsumers(final JsObj conf) {
        JsObj consumers = conf.getObj(JsPath.fromKey(KAFKA)
                                            .key(CONSUMERS));
        return consumers.keySet();
    }


}