feat: extend support for table commit opts

supabase-community · Jan 2, 2024 · 4763488 · 4763488
1 parent 1809ea0
commit 4763488
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/crates/codegen/src/get_node_properties.rs b/crates/codegen/src/get_node_properties.rs
@@ -484,6 +484,12 @@ fn custom_handlers(node: &Node) -> TokenStream {
                 tokens.push(TokenProperty::from(Token::All));
             }
         },
+        "TransactionStmt" => quote! {
+            match n.kind() {
+                protobuf::TransactionStmtKind::TransStmtCommit => tokens.push(TokenProperty::from(Token::Commit)),
+                _ => panic!("Unknown TransactionStmt {:#?}", n.kind())
+            }
+        },
         "PartitionBoundSpec" => quote! {
             tokens.push(TokenProperty::from(Token::From));
             tokens.push(TokenProperty::from(Token::To));