/**
* Represents a query to aggregation. Contains the list of requested keys, fields, predicates and
* orderings.
*/
public final class AggregationQuery {
private List<String> keys = new ArrayList<>();
private List<String> measures = new ArrayList<>();
private AggregationPredicate predicate = AggregationPredicates.alwaysTrue();
public static AggregationQuery create() {
return new AggregationQuery();
}
public static AggregationQuery create(List<String> keys, List<String> measures) {
return new AggregationQuery(keys, measures);
}
public static AggregationQuery create(
List<String> keys, List<String> measures, AggregationPredicate predicate) {
return new AggregationQuery(keys, measures, predicate);
}
private AggregationQuery() {}
private AggregationQuery(List<String> keys, List<String> measures) {
this.keys.addAll(keys);
this.measures.addAll(measures);
}
private AggregationQuery(
List<String> keys, List<String> measures, AggregationPredicate predicate) {
this.keys.addAll(keys);
this.measures.addAll(measures);
this.predicate = predicate;
}
public List<String> getKeys() {
return keys;
}
public List<String> getMeasures() {
return unmodifiableList(measures);
}
public Set<String> getAllKeys() {
LinkedHashSet<String> result = new LinkedHashSet<>();
result.addAll(keys);
result.addAll(predicate.getDimensions());
return result;
}
public AggregationPredicate getPredicate() {
return predicate;
}
public AggregationQuery withKey(String key) {
this.keys.add(key);
return this;
}
public AggregationQuery withKeys(List<String> keys) {
this.keys.addAll(keys);
return this;
}
public AggregationQuery withKeys(String... keys) {
this.keys.addAll(Arrays.asList(keys));
return this;
}
public AggregationQuery withMeasures(List<String> fields) {
this.measures.addAll(fields);
return this;
}
public AggregationQuery withMeasures(String... fields) {
this.measures.addAll(Arrays.asList(fields));
return this;
}
public AggregationQuery withMeasure(String field) {
this.measures.add(field);
return this;
}
public AggregationQuery withPredicate(AggregationPredicate predicate) {
this.predicate = predicate;
return this;
}
public AggregationQuery withPredicates(List<AggregationPredicate> predicates) {
this.predicate = AggregationPredicates.and(predicates);
return this;
}
// public AggregationQuery addPredicates(List<Predicate> predicates) {
// for (Predicate predicate : predicates) {
// this.predicates.add(predicate);
// }
// return this;
// }
@Override
public String toString() {
return "AggregationQuery{"
+ "keys="
+ keys
+ ", fields="
+ measures
+ ", predicate="
+ predicate
+ '}';
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
AggregationQuery query = (AggregationQuery) o;
if (!keys.equals(query.keys)) return false;
if (!measures.equals(query.measures)) return false;
if (!predicate.equals(query.predicate)) return false;
return true;
}
@Override
public int hashCode() {
int result = keys.hashCode();
result = 31 * result + measures.hashCode();
result = 31 * result + predicate.hashCode();
return result;
}
}
public AggregationQuery withPredicates(List<AggregationPredicate> predicates) {
this.predicate = AggregationPredicates.and(predicates);
return this;
}