org.elasticsearch.client.eql.EqlStatsResponse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of elasticsearch-rest-high-level-client Show documentation
Elasticsearch subproject :client:rest-high-level
There is a newer version: 8.0.0-alpha2
Show newest version
/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0 and the Server Side Public License, v 1; you may not use this file except
 * in compliance with, at your election, the Elastic License 2.0 or the Server
 * Side Public License, v 1.
 */
package org.elasticsearch.client.eql;

import org.elasticsearch.client.NodesResponseHeader;
import org.elasticsearch.xcontent.ConstructingObjectParser;
import org.elasticsearch.xcontent.ParseField;
import org.elasticsearch.xcontent.XContentParser;

import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Objects;

public class EqlStatsResponse {
    private final NodesResponseHeader header;
    private final String clusterName;
    private final List nodes;

    public EqlStatsResponse(NodesResponseHeader header, String clusterName, List nodes) {
        this.header = header;
        this.clusterName = clusterName;
        this.nodes = nodes;
    }

    @SuppressWarnings("unchecked")
    private static final ConstructingObjectParser PARSER = new ConstructingObjectParser<>(
        "eql/stats_response",
        true,
        args -> {
            int i = 0;
            NodesResponseHeader header = (NodesResponseHeader) args[i++];
            String clusterName = (String) args[i++];
            List nodes = (List) args[i];
            return new EqlStatsResponse(header, clusterName, nodes);
        }
    );

    static {
        PARSER.declareObject(ConstructingObjectParser.constructorArg(), NodesResponseHeader::fromXContent, new ParseField("_nodes"));
        PARSER.declareString(ConstructingObjectParser.constructorArg(), new ParseField("cluster_name"));
        PARSER.declareObjectArray(
            ConstructingObjectParser.constructorArg(),
            (p, c) -> EqlStatsResponse.Node.PARSER.apply(p, null),
            new ParseField("stats")
        );
    }

    public static EqlStatsResponse fromXContent(XContentParser parser) throws IOException {
        return PARSER.parse(parser, null);
    }

    public NodesResponseHeader getHeader() {
        return header;
    }

    public List getNodes() {
        return nodes;
    }

    public String getClusterName() {
        return clusterName;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        EqlStatsResponse that = (EqlStatsResponse) o;
        return Objects.equals(nodes, that.nodes) && Objects.equals(header, that.header) && Objects.equals(clusterName, that.clusterName);
    }

    @Override
    public int hashCode() {
        return Objects.hash(nodes, header, clusterName);
    }

    public static class Node {
        @SuppressWarnings("unchecked")
        public static final ConstructingObjectParser PARSER = new ConstructingObjectParser<>(
            "eql/stats_response_node",
            true,
            (args, c) -> new Node((Map) args[0])
        );

        static {
            PARSER.declareObject(ConstructingObjectParser.optionalConstructorArg(), (p, c) -> p.map(), new ParseField("stats"));
        }

        private Map stats;

        public Node(Map stats) {
            this.stats = stats;
        }

        public Map getStats() {
            return stats;
        }

        @Override
        public boolean equals(Object o) {
            if (this == o) return true;
            if (o == null || getClass() != o.getClass()) return false;
            Node node = (Node) o;
            return Objects.equals(stats, node.stats);
        }

        @Override
        public int hashCode() {
            return Objects.hash(stats);
        }
    }
}