add other

This commit is contained in:
Zoltan Haindrich 2024-05-16 11:23:26 +00:00
parent 607ef174c5
commit 59be71c068
1 changed files with 11 additions and 2 deletions

View File

@ -53,6 +53,7 @@ import org.apache.druid.server.security.AuthorizerMapper;
import org.apache.druid.sql.SqlStatementFactory; import org.apache.druid.sql.SqlStatementFactory;
import org.apache.druid.sql.calcite.SqlTestFrameworkConfig; import org.apache.druid.sql.calcite.SqlTestFrameworkConfig;
import org.apache.druid.sql.calcite.TempDirProducer; import org.apache.druid.sql.calcite.TempDirProducer;
import org.apache.druid.sql.calcite.aggregation.SqlAggregationModule;
import org.apache.druid.sql.calcite.planner.CalciteRulesManager; import org.apache.druid.sql.calcite.planner.CalciteRulesManager;
import org.apache.druid.sql.calcite.planner.CatalogResolver; import org.apache.druid.sql.calcite.planner.CatalogResolver;
import org.apache.druid.sql.calcite.planner.DruidOperatorTable; import org.apache.druid.sql.calcite.planner.DruidOperatorTable;
@ -569,7 +570,14 @@ public class SqlTestFramework
@Provides @Provides
@LazySingleton @LazySingleton
public QuerySegmentWalker segmentsQuerySegmentWalker(final Injector injector) public QuerySegmentWalker querySegmentWalker(final Injector injector)
{
return injector.getInstance(SpecificSegmentsQuerySegmentWalker.class);
}
@Provides
@LazySingleton
public SpecificSegmentsQuerySegmentWalker specificSegmentsQuerySegmentWalker(final Injector injector)
{ {
SpecificSegmentsQuerySegmentWalker walker = componentSupplier.createQuerySegmentWalker( SpecificSegmentsQuerySegmentWalker walker = componentSupplier.createQuerySegmentWalker(
injector.getInstance(QueryRunnerFactoryConglomerate.class), injector.getInstance(QueryRunnerFactoryConglomerate.class),
@ -648,7 +656,8 @@ return componentSupplier.getPlannerComponentSupplier().createViewManager();
.ignoreLoadScopes() .ignoreLoadScopes()
.addModule(new LookylooModule()) .addModule(new LookylooModule())
.addModule(new SegmentWranglerModule()) .addModule(new SegmentWranglerModule())
// .addModule(new SqlAggregationModule()) .addModule(new SqlAggregationModule())
// .addModule(new SqlModule())
.addModule(new ExpressionModule()); .addModule(new ExpressionModule());
// .addModule(testSetupModule()); // .addModule(testSetupModule());