org.apache.druid.query.aggregation.TimestampBufferAggregator Maven / Gradle / Ivy
The newest version!
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF 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 org.apache.druid.query.aggregation;
import org.apache.druid.data.input.impl.TimestampSpec;
import org.apache.druid.query.monomorphicprocessing.RuntimeShapeInspector;
import org.apache.druid.segment.BaseObjectColumnValueSelector;
import java.nio.ByteBuffer;
import java.util.Comparator;
public class TimestampBufferAggregator implements BufferAggregator
{
private final BaseObjectColumnValueSelector selector;
private final TimestampSpec timestampSpec;
private final Comparator comparator;
private final Long initValue;
public TimestampBufferAggregator(
BaseObjectColumnValueSelector selector,
TimestampSpec timestampSpec,
Comparator comparator,
Long initValue)
{
this.selector = selector;
this.timestampSpec = timestampSpec;
this.comparator = comparator;
this.initValue = initValue;
}
@Override
public void init(ByteBuffer buf, int position)
{
buf.putLong(position, initValue);
}
@Override
public void aggregate(ByteBuffer buf, int position)
{
Long newTime = TimestampAggregatorFactory.convertLong(timestampSpec, selector.getObject());
if (newTime != null) {
long prev = buf.getLong(position);
buf.putLong(position, comparator.compare(prev, newTime) > 0 ? prev : newTime);
}
}
@Override
public Object get(ByteBuffer buf, int position)
{
return buf.getLong(position);
}
@Override
public float getFloat(ByteBuffer buf, int position)
{
return (float) buf.getLong(position);
}
@Override
public long getLong(ByteBuffer buf, int position)
{
return buf.getLong(position);
}
@Override
public double getDouble(ByteBuffer buf, int position)
{
return (double) buf.getLong(position);
}
@Override
public void close()
{
}
@Override
public void inspectRuntimeShape(RuntimeShapeInspector inspector)
{
inspector.visit("selector", selector);
inspector.visit("comparator", comparator);
}
}