noth/surrealpatch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Pass the current statement to the value iterator functions

Browse source
This commit is contained in:
Tobie Morgan Hitchcock 2022-03-18 19:51:13 +00:00
parent 5b1d727c25
commit f8b747374f
2 changed files with 19 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
lib/src/dbs/channel.rs
View file

@ -1,5 +1,6 @@
use crate::dbs::Options;
use crate::dbs::Runtime;
use crate::dbs::Statement;
use crate::dbs::Transaction;
use crate::err::Error;
use crate::key::thing;
@ -17,15 +18,16 @@ impl Value {
self,
ctx: Runtime,
opt: Options,
stm: Statement,
txn: Transaction,
chn: Sender<(Option<Thing>, Value)>,
) -> Result<(), Error> {
if ctx.is_ok() {
match self {
Value::Array(v) => v.process(&ctx, &opt, &txn, &chn).await?,
Value::Model(v) => v.process(&ctx, &opt, &txn, &chn).await?,
Value::Thing(v) => v.process(&ctx, &opt, &txn, &chn).await?,
Value::Table(v) => v.process(&ctx, &opt, &txn, &chn).await?,
Value::Array(v) => v.process(&ctx, &opt, &stm, &txn, &chn).await?,
Value::Model(v) => v.process(&ctx, &opt, &stm, &txn, &chn).await?,
Value::Thing(v) => v.process(&ctx, &opt, &stm, &txn, &chn).await?,
Value::Table(v) => v.process(&ctx, &opt, &stm, &txn, &chn).await?,
v => chn.send((None, v)).await?,
}
}
@ -39,16 +41,17 @@ impl Array {
self,
ctx: &Runtime,
opt: &Options,
stm: &Statement,
txn: &Transaction,
chn: &Sender<(Option<Thing>, Value)>,
) -> Result<(), Error> {
for v in self.value.into_iter() {
if ctx.is_ok() {
match v {
Value::Array(v) => v.process(ctx, opt, txn, chn).await?,
Value::Model(v) => v.process(ctx, opt, txn, chn).await?,
Value::Thing(v) => v.process(ctx, opt, txn, chn).await?,
Value::Table(v) => v.process(ctx, opt, txn, chn).await?,
Value::Array(v) => v.process(ctx, opt, stm, txn, chn).await?,
Value::Model(v) => v.process(ctx, opt, stm, txn, chn).await?,
Value::Thing(v) => v.process(ctx, opt, stm, txn, chn).await?,
Value::Table(v) => v.process(ctx, opt, stm, txn, chn).await?,
v => chn.send((None, v)).await?,
}
}
@ -62,6 +65,7 @@ impl Model {
self,
ctx: &Runtime,
opt: &Options,
stm: &Statement,
txn: &Transaction,
chn: &Sender<(Option<Thing>, Value)>,
) -> Result<(), Error> {
@ -72,7 +76,7 @@ impl Model {
tb: self.table.to_string(),
id: Id::rand(),
}
.process(ctx, opt, txn, chn)
.process(ctx, opt, stm, txn, chn)
.await?;
}
}
@ -82,7 +86,7 @@ impl Model {
tb: self.table.to_string(),
id: Id::from(x),
}
.process(ctx, opt, txn, chn)
.process(ctx, opt, stm, txn, chn)
.await?;
}
}
@ -96,6 +100,7 @@ impl Thing {
self,
ctx: &Runtime,
opt: &Options,
stm: &Statement,
txn: &Transaction,
chn: &Sender<(Option<Thing>, Value)>,
) -> Result<(), Error> {
@ -117,6 +122,7 @@ impl Table {
self,
ctx: &Runtime,
opt: &Options,
stm: &Statement,
txn: &Transaction,
chn: &Sender<(Option<Thing>, Value)>,
) -> Result<(), Error> {

3
lib/src/dbs/iterator.rs
View file

@ -213,6 +213,8 @@ impl Iterator {
// Run statements in parallel
true => {
let mut rcv = {
// Get current statement
let stm = &self.stm;
// Create an unbounded channel
let (chn, rx) = tokio::sync::mpsc::channel(MAX_CONCURRENT_TASKS);
// Process all prepared values
@ -221,6 +223,7 @@ impl Iterator {
tokio::spawn(v.channel(
ctx.clone(),
opt.clone(),
stm.clone(),
txn.clone(),
chn.clone(),
));