Fix(kclvm-codegen): fix schema back ref. (#232)

* Fix(kclvm-codegen): fix schema back ref. fix schema back reference exception. issue #231 * fix mistake
kcl-lang · Oct 11, 2022 · 93ee39e · 93ee39e
1 parent 1a5ddd7
commit 93ee39e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/kclvm/compiler/src/codegen/llvm/module.rs b/kclvm/compiler/src/codegen/llvm/module.rs
@@ -17,9 +17,13 @@ impl<'ctx> LLVMCodeGenContext<'ctx> {
                 }
                 ast::Stmt::Schema(schema_stmt) => {
                     self.predefine_global_types(&schema_stmt.name.node);
+                    self.walk_schema_stmt(schema_stmt)
+                        .expect(kcl_error::COMPILE_ERROR_MSG);
                 }
                 ast::Stmt::Rule(rule_stmt) => {
                     self.predefine_global_types(&rule_stmt.name.node);
+                    self.walk_rule_stmt(rule_stmt)
+                        .expect(kcl_error::COMPILE_ERROR_MSG);
                 }
                 _ => {}
             };

diff --git a/test/grammar/schema/back_ref/back_ref_1.k b/test/grammar/schema/back_ref/back_ref_1.k
@@ -0,0 +1,5 @@
+x0 = Person {}
+
+schema Person:
+    name: str = "kcl"
+    age: int = 1
diff --git a/test/grammar/schema/back_ref/stdout.golden b/test/grammar/schema/back_ref/stdout.golden
@@ -0,0 +1,3 @@
+x0:
+  name: kcl
+  age: 1