io.druid.query.aggregation.datasketches.theta.SketchMergeAggregatorFactory Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of druid-datasketches Show documentation
Show all versions of druid-datasketches Show documentation
Druid Aggregators based on datasketches lib http://datasketches.github.io/
The newest version!
/*
* Licensed to Metamarkets Group Inc. (Metamarkets) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. Metamarkets licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package io.druid.query.aggregation.datasketches.theta;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
import io.druid.query.aggregation.AggregatorFactory;
import io.druid.query.aggregation.AggregatorFactoryNotMergeableException;
import io.druid.query.aggregation.AggregatorUtil;
import java.util.Collections;
import java.util.List;
public class SketchMergeAggregatorFactory extends SketchAggregatorFactory
{
private final boolean shouldFinalize;
private final boolean isInputThetaSketch;
private final Integer errorBoundsStdDev;
@JsonCreator
public SketchMergeAggregatorFactory(
@JsonProperty("name") String name,
@JsonProperty("fieldName") String fieldName,
@JsonProperty("size") Integer size,
@JsonProperty("shouldFinalize") Boolean shouldFinalize,
@JsonProperty("isInputThetaSketch") Boolean isInputThetaSketch,
@JsonProperty("errorBoundsStdDev") Integer errorBoundsStdDev
)
{
super(name, fieldName, size, AggregatorUtil.SKETCH_MERGE_CACHE_TYPE_ID);
this.shouldFinalize = (shouldFinalize == null) ? true : shouldFinalize.booleanValue();
this.isInputThetaSketch = (isInputThetaSketch == null) ? false : isInputThetaSketch.booleanValue();
this.errorBoundsStdDev = errorBoundsStdDev;
}
@Override
public List getRequiredColumns()
{
return Collections.singletonList(
new SketchMergeAggregatorFactory(
fieldName,
fieldName,
size,
shouldFinalize,
isInputThetaSketch,
errorBoundsStdDev
)
);
}
@Override
public AggregatorFactory getCombiningFactory()
{
return new SketchMergeAggregatorFactory(name, name, size, shouldFinalize, false, errorBoundsStdDev);
}
@Override
public AggregatorFactory getMergingFactory(AggregatorFactory other) throws AggregatorFactoryNotMergeableException
{
if (other.getName().equals(this.getName()) && other instanceof SketchMergeAggregatorFactory) {
SketchMergeAggregatorFactory castedOther = (SketchMergeAggregatorFactory) other;
return new SketchMergeAggregatorFactory(
name,
name,
Math.max(size, castedOther.size),
shouldFinalize,
false,
errorBoundsStdDev
);
} else {
throw new AggregatorFactoryNotMergeableException(this, other);
}
}
@JsonProperty
public boolean getShouldFinalize()
{
return shouldFinalize;
}
@JsonProperty
public boolean getIsInputThetaSketch()
{
return isInputThetaSketch;
}
@JsonProperty
public Integer getErrorBoundsStdDev()
{
return errorBoundsStdDev;
}
/**
* Finalize the computation on sketch object and returns estimate from underlying
* sketch.
*
* @param object the sketch object
*
* @return sketch object
*/
@Override
public Object finalizeComputation(Object object)
{
if (shouldFinalize) {
SketchHolder holder = (SketchHolder) object;
if (errorBoundsStdDev != null) {
return holder.getEstimateWithErrorBounds(errorBoundsStdDev);
} else {
return holder.getEstimate();
}
} else {
return object;
}
}
@Override
public String getTypeName()
{
if (isInputThetaSketch) {
return SketchModule.THETA_SKETCH_MERGE_AGG;
} else {
return SketchModule.THETA_SKETCH_BUILD_AGG;
}
}
@Override
public boolean equals(Object o)
{
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
if (!super.equals(o)) {
return false;
}
SketchMergeAggregatorFactory that = (SketchMergeAggregatorFactory) o;
if (shouldFinalize != that.shouldFinalize) {
return false;
}
if (errorBoundsStdDev == null ^ that.errorBoundsStdDev == null) {
// one of the two stddevs (not both) are null
return false;
}
if (errorBoundsStdDev != null && that.errorBoundsStdDev != null &&
errorBoundsStdDev.intValue() != that.errorBoundsStdDev.intValue()) {
// neither stddevs are null, Integer values don't match
return false;
}
return isInputThetaSketch == that.isInputThetaSketch;
}
@Override
public int hashCode()
{
int result = super.hashCode();
result = 31 * result + (shouldFinalize ? 1 : 0);
result = 31 * result + (isInputThetaSketch ? 1 : 0);
result = 31 * result + (errorBoundsStdDev != null ? errorBoundsStdDev.hashCode() : 0);
return result;
}
@Override
public String toString()
{
return "SketchMergeAggregatorFactory{"
+ "fieldName=" + fieldName
+ ", name=" + name
+ ", size=" + size
+ ", shouldFinalize=" + shouldFinalize
+ ", isInputThetaSketch=" + isInputThetaSketch
+ ", errorBoundsStdDev=" + errorBoundsStdDev
+ "}";
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy