Refactor assigning source ID

risingwavelabs · Sep 19, 2023 · dbbe9db · dbbe9db
1 parent 26c8e1f
commit dbbe9db
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/meta/src/rpc/service/ddl_service.rs b/src/meta/src/rpc/service/ddl_service.rs
@@ -779,8 +779,11 @@ fn fill_table_source(
         visit_fragment(fragment, |node_body| {
             if let NodeBody::Source(source_node) = node_body {
                 // TODO: Refactor using source id.
-                source_node.source_inner.as_mut().unwrap().source_id = source_id;
-                source_count += 1;
+
+                if let Some(source_inner) = source_node.source_inner.as_mut() {
+                    source_inner.source_id = source_id;
+                    source_count += 1;
+                }
             }
         });
     }