From 122e0a0f4a15588609250b063aa64301a76c88a3 Mon Sep 17 00:00:00 2001 From: Xianzhe Liang Date: Sat, 2 Mar 2024 11:08:30 -0800 Subject: [PATCH 1/2] Add MultiTransaction.run which takes TransactionPayload. For user who constructs the payload directly in Rust, this is slightly easier. --- cozo-core/src/lib.rs | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/cozo-core/src/lib.rs b/cozo-core/src/lib.rs index c0969a01..e5598199 100644 --- a/cozo-core/src/lib.rs +++ b/cozo-core/src/lib.rs @@ -527,6 +527,16 @@ pub struct MultiTransaction { impl MultiTransaction { /// Runs a single script in the transaction. + pub fn run( &self, payload: TransactionPayload) -> Result { + if let Err(err) = self.sender.send(payload) { + bail!(err); + } + match self.receiver.recv() { + Ok(r) => r, + Err(err) => bail!(err), + } + } + /// Runs a single script in the transaction. Same as run but takes query string and parameters. pub fn run_script( &self, payload: &str, From 3a3621db8fc0d9e17099238208e3638c885a23fa Mon Sep 17 00:00:00 2001 From: Xianzhe Liang Date: Wed, 6 Mar 2024 10:10:38 -0800 Subject: [PATCH 2/2] Minor code format change. --- cozo-core/src/lib.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cozo-core/src/lib.rs b/cozo-core/src/lib.rs index e5598199..36fb8cf9 100644 --- a/cozo-core/src/lib.rs +++ b/cozo-core/src/lib.rs @@ -527,7 +527,7 @@ pub struct MultiTransaction { impl MultiTransaction { /// Runs a single script in the transaction. - pub fn run( &self, payload: TransactionPayload) -> Result { + pub fn run(&self, payload: TransactionPayload) -> Result { if let Err(err) = self.sender.send(payload) { bail!(err); }