This is an automated email from the ASF dual-hosted git repository.

yashmayya pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/pinot.git


The following commit(s) were added to refs/heads/master by this push:
     new 52c5330f4e Initialize MSE worker thread context for TSE queries 
(#15353)
52c5330f4e is described below

commit 52c5330f4e85eae643cc32d40d986bcfa8cb64d4
Author: Yash Mayya <[email protected]>
AuthorDate: Mon Mar 24 17:34:29 2025 +0530

    Initialize MSE worker thread context for TSE queries (#15353)
---
 .../main/java/org/apache/pinot/query/service/server/QueryServer.java   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git 
a/pinot-query-runtime/src/main/java/org/apache/pinot/query/service/server/QueryServer.java
 
b/pinot-query-runtime/src/main/java/org/apache/pinot/query/service/server/QueryServer.java
index 37e99e45bf..a9f2b9933f 100644
--- 
a/pinot-query-runtime/src/main/java/org/apache/pinot/query/service/server/QueryServer.java
+++ 
b/pinot-query-runtime/src/main/java/org/apache/pinot/query/service/server/QueryServer.java
@@ -237,7 +237,8 @@ public class QueryServer extends 
PinotQueryWorkerGrpc.PinotQueryWorkerImplBase {
   @Override
   public void submitTimeSeries(Worker.TimeSeriesQueryRequest request,
       StreamObserver<Worker.TimeSeriesResponse> responseObserver) {
-    try (QueryThreadContext.CloseableContext closeable = 
QueryThreadContext.open()) {
+    try (QueryThreadContext.CloseableContext queryTlClosable = 
QueryThreadContext.open();
+        QueryThreadContext.CloseableContext mseTlCloseable = 
MseWorkerThreadContext.open()) {
       // TODO: populate the thread context with TSE information
       QueryThreadContext.setQueryEngine("tse");
       _queryRunner.processTimeSeriesQuery(request.getDispatchPlanList(), 
request.getMetadataMap(), responseObserver);


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to