let additional batch queries block the session

risingwavelabs · Mar 8, 2024 · a31d6fd · a31d6fd
1 parent d5a77d3
commit a31d6fd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/frontend/src/scheduler/distributed/query_manager.rs b/src/frontend/src/scheduler/distributed/query_manager.rs
@@ -163,10 +163,10 @@ impl QueryManager {
         }
     }
 
-    fn get_permit(&self) -> SchedulerResult<Option<OwnedSemaphorePermit>> {
+    async fn get_permit(&self) -> SchedulerResult<Option<OwnedSemaphorePermit>> {
         match self.distributed_query_semaphore {
             Some(ref semaphore) => {
-                let permit = semaphore.clone().try_acquire_owned();
+                let permit = semaphore.clone().acquire_owned().await;
                 match permit {
                     Ok(permit) => Ok(Some(permit)),
                     Err(_) => {
@@ -198,7 +198,7 @@ impl QueryManager {
             ));
         }
         let query_id = query.query_id.clone();
-        let permit = self.get_permit()?;
+        let permit = self.get_permit().await?;
         let query_execution = Arc::new(QueryExecution::new(query, context.session().id(), permit));
 
         // Add queries status when begin.