org.elasticsearch.search.aggregations.metrics.InternalBounds Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of elasticsearch Show documentation
Show all versions of elasticsearch Show documentation
Elasticsearch subproject :server
/*
* 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.search.aggregations.metrics;
import org.elasticsearch.common.geo.BoundingBox;
import org.elasticsearch.common.geo.GeoBoundingBox;
import org.elasticsearch.common.geo.SpatialPoint;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;
import org.elasticsearch.search.aggregations.InternalAggregation;
import org.elasticsearch.search.aggregations.support.SamplingContext;
import org.elasticsearch.xcontent.XContentBuilder;
import java.io.IOException;
import java.util.List;
import java.util.Map;
public abstract class InternalBounds extends InternalAggregation implements SpatialBounds {
public final double top;
public final double bottom;
public InternalBounds(String name, double top, double bottom, Map metadata) {
super(name, metadata);
this.top = top;
this.bottom = bottom;
}
/**
* Read from a stream.
*/
public InternalBounds(StreamInput in) throws IOException {
super(in);
top = in.readDouble();
bottom = in.readDouble();
}
@Override
protected void doWriteTo(StreamOutput out) throws IOException {
out.writeDouble(top);
out.writeDouble(bottom);
}
@Override
public InternalAggregation finalizeSampling(SamplingContext samplingContext) {
return this;
}
@Override
protected boolean mustReduceOnSingleInternalAgg() {
return false;
}
@Override
public Object getProperty(List path) {
if (path.isEmpty()) {
return this;
} else if (path.size() == 1) {
BoundingBox bbox = resolveBoundingBox();
String bBoxSide = path.get(0);
return switch (bBoxSide) {
case "top" -> bbox.top();
case "left" -> bbox.left();
case "bottom" -> bbox.bottom();
case "right" -> bbox.right();
default -> throw new IllegalArgumentException("Found unknown path element [" + bBoxSide + "] in [" + getName() + "]");
};
} else if (path.size() == 2) {
BoundingBox bbox = resolveBoundingBox();
T cornerPoint = null;
String cornerString = path.get(0);
cornerPoint = switch (cornerString) {
case "top_left" -> bbox.topLeft();
case "bottom_right" -> bbox.bottomRight();
default -> throw new IllegalArgumentException("Found unknown path element [" + cornerString + "] in [" + getName() + "]");
};
return selectCoordinate(path.get(1), cornerPoint);
} else {
throw new IllegalArgumentException("path not supported for [" + getName() + "]: " + path);
}
}
protected abstract Object selectCoordinate(String coordinateString, T cornerPoint);
@Override
public XContentBuilder doXContentBody(XContentBuilder builder, Params params) throws IOException {
BoundingBox bbox = resolveBoundingBox();
if (bbox != null) {
builder.startObject(GeoBoundingBox.BOUNDS_FIELD.getPreferredName());
bbox.toXContentFragment(builder);
builder.endObject();
}
return builder;
}
protected abstract BoundingBox resolveBoundingBox();
@Override
public T topLeft() {
BoundingBox bbox = resolveBoundingBox();
if (bbox == null) {
return null;
} else {
return bbox.topLeft();
}
}
@Override
public T bottomRight() {
BoundingBox bbox = resolveBoundingBox();
if (bbox == null) {
return null;
} else {
return bbox.bottomRight();
}
}
}