zone.cogni.asquare.service.elasticsearch.info.Info_clusterState Maven / Gradle / Ivy

Go to download
package zone.cogni.asquare.service.elasticsearch.info;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import org.apache.commons.collections4.IteratorUtils;
import zone.cogni.core.util.DateFormats;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Date;
import java.util.List;

/**
 * JSON wrapper from cluster/_state endpoint of elasticsearch.
 */
public class Info_clusterState {

  private final JsonNode root;

  public Info_clusterState(JsonNode root) {
    this.root = root;
  }

  public JsonNode getIndex(String name) {
    return root.get("metadata").get("indices").get(name);
  }

  public List getIndexNames() {
    ObjectNode indexes = (ObjectNode) root.get("metadata").get("indices");
    List result = IteratorUtils.toList(indexes.fieldNames());
    Collections.sort(result);
    return result;
  }

  public String getCreatedDate(String indexName) {
    JsonNode index = getIndex(indexName);
    String timestampAsString = index.get("settings").get("index").get("creation_date").textValue();
    Date date = new Date(Long.parseLong(timestampAsString));
    return DateFormats.formatXsdDateTimeFormat(date);
  }

  public List getAliases(String indexName) {
    JsonNode index = getIndex(indexName);
    JsonNode aliases = index.get("aliases");

    if (!aliases.isArray()) return Collections.emptyList();

    List result = new ArrayList<>();
    ArrayNode aliasArray = (ArrayNode) aliases;
    for (int i = 0; i < aliasArray.size(); i++) {
      JsonNode alias = aliasArray.get(i);
      result.add(alias.textValue());
    }

    return result;
  }
}