return timeout cancel on failure

risingwavelabs · Oct 25, 2023 · 8286d63 · 8286d63
1 parent 8221d3a
commit 8286d63
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/meta/service/src/ddl_service.rs b/src/meta/service/src/ddl_service.rs
@@ -734,7 +734,7 @@ impl DdlService for DdlServiceImpl {
     }
 
     async fn wait(&self, _request: Request<WaitRequest>) -> Result<Response<WaitResponse>, Status> {
-        self.ddl_controller.wait().await;
+        self.ddl_controller.wait().await?;
         Ok(Response::new(WaitResponse {}))
     }
 }

diff --git a/src/meta/src/rpc/ddl_controller.rs b/src/meta/src/rpc/ddl_controller.rs
@@ -1097,17 +1097,18 @@ impl DdlController {
         }
     }
 
-    pub async fn wait(&self) {
+    pub async fn wait(&self) -> MetaResult<()> {
         for _ in 0..30 * 60 {
             if self
                 .catalog_manager
                 .list_creating_background_mvs()
                 .await
                 .is_empty()
             {
-                break;
+                return Ok(());
             }
             sleep(Duration::from_secs(1)).await;
         }
+        return Err(MetaError::cancelled("timeout".into()));
     }
 }