All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • io.polyglotted.eswrapper.query.QueryResponse Maven / Gradle / Ivy

    package io.polyglotted.eswrapper.query;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.Iterables;
import com.google.common.collect.Lists;
import io.polyglotted.eswrapper.query.response.Aggregation;
import io.polyglotted.eswrapper.query.response.ResponseHeader;
import lombok.AccessLevel;
import lombok.RequiredArgsConstructor;
import lombok.Setter;
import lombok.ToString;
import lombok.experimental.Accessors;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

import static com.google.common.base.Preconditions.checkNotNull;
import static io.polyglotted.eswrapper.query.StandardScroll.fromScrollId;

@RequiredArgsConstructor
@ToString(includeFieldNames = false, doNotUseGetters = true)
public final class QueryResponse {
    public final ResponseHeader header;
    public final ImmutableList results;
    public final ImmutableList aggregations;

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        QueryResponse that = (QueryResponse) o;
        return header.equals(that.header) && results.equals(that.results) && aggregations.equals(that.aggregations);
    }

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

    public  List resultsAs(Class tClass) {
        return Lists.transform(results, tClass::cast);
    }

    public StandardScroll nextScroll() {
        return fromScrollId(checkNotNull(header.id, "cannot scroll null scrollId"));
    }

    public static Builder responseBuilder() {
        return new Builder();
    }

    @Setter
    @Accessors(fluent = true, chain = true)
    @RequiredArgsConstructor(access = AccessLevel.PRIVATE)
    public static class Builder {
        private ResponseHeader header = null;
        private final List results = new ArrayList<>();
        private List aggregations = new ArrayList<>();

        public Builder results(Iterable objects) {
            Iterables.addAll(results, objects);
            return this;
        }

        public QueryResponse build() {
            return new QueryResponse(checkNotNull(header, "header cannot be null"),
               ImmutableList.copyOf(results), ImmutableList.copyOf(aggregations));
        }
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy