fix v1

risingwavelabs · Jun 28, 2024 · 044525f · 044525f
1 parent 2dfb3e5
commit 044525f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/frontend/src/catalog/root_catalog.rs b/src/frontend/src/catalog/root_catalog.rs
@@ -311,7 +311,7 @@ impl Catalog {
     pub fn update_table(&mut self, proto: &PbTable) {
         let database = self.get_database_mut(proto.database_id).unwrap();
         let schema = database.get_schema_mut(proto.schema_id).unwrap();
-        let table = if schema.get_created_table_by_id(&proto.id.into()).is_some() {
+        let table = if schema.get_table_by_id(&proto.id.into()).is_some() {
             schema.update_table(proto)
         } else {
             // Enter this branch when schema is changed by `ALTER ... SET SCHEMA ...` statement.

diff --git a/src/frontend/src/catalog/schema_catalog.rs b/src/frontend/src/catalog/schema_catalog.rs
@@ -615,6 +615,10 @@ impl SchemaCatalog {
             .filter(|&table| table.stream_job_status != StreamJobStatus::Creating)
     }
 
+    pub fn get_table_by_id(&self, table_id: &TableId) -> Option<&Arc<TableCatalog>> {
+        self.table_by_id.get(table_id)
+    }
+
     pub fn get_created_table_by_id(&self, table_id: &TableId) -> Option<&Arc<TableCatalog>> {
         self.table_by_id
             .get(table_id)