fix(exec): fix RemoteExec transformer deadlock (#1774)

filodb · May 15, 2024 · b0b35f1 · b0b35f1
1 parent 522ab33
commit b0b35f1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/query/src/main/scala/filodb/query/exec/PromQLGrpcRemoteExec.scala b/query/src/main/scala/filodb/query/exec/PromQLGrpcRemoteExec.scala
@@ -56,7 +56,8 @@ trait GrpcRemoteExec extends RemoteExec {
               .timed
               .map { case (elapsed, qresp) =>
                   val timeRemaining = Duration(requestTimeoutMs, TimeUnit.MILLISECONDS) - elapsed
-                  applyTransformers(qresp, querySession, source, timeRemaining)
+                  applyTransformers(qresp, querySession, source,
+                    timeRemaining)(monix.execution.Scheduler.Implicits.global)
               }
         }
     }

diff --git a/query/src/main/scala/filodb/query/exec/RemoteExec.scala b/query/src/main/scala/filodb/query/exec/RemoteExec.scala
@@ -82,7 +82,8 @@ trait RemoteExec extends LeafExecPlan with StrictLogging {
         .timed
         .map{ case (elapsed, qresp) =>
           val timeRemaining = Duration(requestTimeoutMs, TimeUnit.MILLISECONDS) - elapsed
-          applyTransformers(qresp, querySession, source, timeRemaining)
+          applyTransformers(qresp, querySession, source,
+            timeRemaining)(monix.execution.Scheduler.Implicits.global)
         }
     }
   }