io.druid.query.FluentQueryRunnerBuilder Maven / Gradle / Ivy
/*
* 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;
import com.metamx.emitter.service.ServiceEmitter;
import io.druid.java.util.common.guava.Sequence;
import java.util.Map;
import java.util.concurrent.atomic.AtomicLong;
public class FluentQueryRunnerBuilder
{
final QueryToolChest> toolChest;
public FluentQueryRunner create(QueryRunner baseRunner)
{
return new FluentQueryRunner(baseRunner);
}
public FluentQueryRunnerBuilder(QueryToolChest> toolChest)
{
this.toolChest = toolChest;
}
public class FluentQueryRunner implements QueryRunner
{
private QueryRunner baseRunner;
public FluentQueryRunner(QueryRunner runner)
{
this.baseRunner = runner;
}
@Override
public Sequence run(
QueryPlus queryPlus, Map responseContext
)
{
return baseRunner.run(queryPlus, responseContext);
}
public FluentQueryRunner from(QueryRunner runner)
{
return new FluentQueryRunner(runner);
}
public FluentQueryRunner applyPostMergeDecoration()
{
return from(
new FinalizeResultsQueryRunner(
toolChest.postMergeQueryDecoration(
baseRunner
),
toolChest
)
);
}
public FluentQueryRunner applyPreMergeDecoration()
{
return from(
new UnionQueryRunner(
toolChest.preMergeQueryDecoration(
baseRunner
)
)
);
}
public FluentQueryRunner emitCPUTimeMetric(ServiceEmitter emitter)
{
return from(
CPUTimeMetricQueryRunner.safeBuild(
baseRunner,
toolChest,
emitter,
new AtomicLong(0L),
true
)
);
}
public FluentQueryRunner postProcess(PostProcessingOperator postProcessing)
{
return from(
postProcessing != null ?
postProcessing.postProcess(baseRunner) : baseRunner
);
}
public FluentQueryRunner mergeResults()
{
return from(
toolChest.mergeResults(baseRunner)
);
}
}
}