This commit is contained in:
Zoltan Haindrich 2024-07-17 16:42:22 +00:00
parent 95ca0a9f5d
commit c59f1adcc8
2 changed files with 20 additions and 13 deletions

View File

@ -116,7 +116,7 @@ public class MSQDrillWindowQueryTest extends DrillWindowQueryTest
@Provides
@MultiStageQuery
@LazySingleton
public SqlStatementFactory makeNativeSqlStatementFactory(
public SqlStatementFactory makeMSQSqlStatementFactory(
final MSQTaskSqlEngine sqlEngine,
SqlToolbox toolbox
)
@ -129,18 +129,25 @@ public class MSQDrillWindowQueryTest extends DrillWindowQueryTest
public MSQTaskSqlEngine createEngine(
QueryLifecycleFactory qlf,
ObjectMapper queryJsonMapper,
Injector injector
Injector injector,
MSQTestOverlordServiceClient indexingServiceClient
)
{
final WorkerMemoryParameters workerMemoryParameters =
WorkerMemoryParameters.createInstance(
WorkerMemoryParameters.PROCESSING_MINIMUM_BYTES * 50,
2,
10,
2,
0,
0
);
return new MSQTaskSqlEngine(indexingServiceClient, queryJsonMapper);
}
@Provides
@LazySingleton
private MSQTestOverlordServiceClient extracted(ObjectMapper queryJsonMapper, Injector injector)
{
final WorkerMemoryParameters workerMemoryParameters = WorkerMemoryParameters.createInstance(
WorkerMemoryParameters.PROCESSING_MINIMUM_BYTES * 50,
2,
10,
2,
0,
0
);
final MSQTestOverlordServiceClient indexingServiceClient = new MSQTestOverlordServiceClient(
queryJsonMapper,
injector,
@ -148,7 +155,7 @@ public class MSQDrillWindowQueryTest extends DrillWindowQueryTest
workerMemoryParameters,
ImmutableList.of()
);
return new MSQTaskSqlEngine(indexingServiceClient, queryJsonMapper);
return indexingServiceClient;
}
@Provides

View File

@ -10,7 +10,7 @@ order by 1;
+--------------------------------------------+
| TASK |
+--------------------------------------------+
| query-eeb20bee-0a2a-4812-97d9-ae70a9107f80 |
| query-09ed2fdf-e74e-4d61-9a5b-b9ce4949aa48 |
+--------------------------------------------+
(1 row)