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

Add current INSERT value as parameter value to ON DUPLICATE KEY clause

Browse source 
Closes #1376
This commit is contained in:
Tobie Morgan Hitchcock 2022-12-04 20:59:30 +00:00
parent 3c4f194610
commit a3d0f46994
1 changed files with 29 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
lib/src/doc/alter.rs
View file

@ -2,6 +2,7 @@ use crate::ctx::Context;
use crate::dbs::Options; use crate::dbs::Options;
use crate::dbs::Statement; use crate::dbs::Statement;
use crate::dbs::Transaction; use crate::dbs::Transaction;
use crate::dbs::Workable;
use crate::doc::Document; use crate::doc::Document;
use crate::err::Error; use crate::err::Error;
use crate::sql::data::Data; use crate::sql::data::Data;
@ -23,6 +24,22 @@ impl<'a> Document<'a> {
// The statement has a data clause // The statement has a data clause
if let Some(v) = stm.data() { if let Some(v) = stm.data() {
match v { match v {
Data::PatchExpression(data) => {
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
self.current.to_mut().patch(ctx, opt, txn, data).await?
}
Data::MergeExpression(data) => {
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
self.current.to_mut().merge(ctx, opt, txn, data).await?
}
Data::ReplaceExpression(data) => {
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
self.current.to_mut().replace(ctx, opt, txn, data).await?
}
Data::ContentExpression(data) => {
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
self.current.to_mut().replace(ctx, opt, txn, data).await?
}
Data::SetExpression(x) => { Data::SetExpression(x) => {
for x in x.iter() { for x in x.iter() {
let v = x.2.compute(ctx, opt, txn, Some(&self.current)).await?; let v = x.2.compute(ctx, opt, txn, Some(&self.current)).await?;
@ -44,41 +61,32 @@ impl<'a> Document<'a> {
} }
} }
Data::UpdateExpression(x) => { Data::UpdateExpression(x) => {
// Duplicate context
let mut ctx = Context::new(ctx);
// Add insertable value
if let Workable::Insert(value) = &self.extras {
ctx.add_value("value".into(), value);
}
// Process ON DUPLICATE KEY clause
for x in x.iter() { for x in x.iter() {
let v = x.2.compute(ctx, opt, txn, Some(&self.current)).await?; let v = x.2.compute(&ctx, opt, txn, Some(&self.current)).await?;
match x.1 { match x.1 {
Operator::Equal => match v { Operator::Equal => match v {
Value::None => { Value::None => {
self.current.to_mut().del(ctx, opt, txn, &x.0).await? self.current.to_mut().del(&ctx, opt, txn, &x.0).await?
} }
_ => self.current.to_mut().set(ctx, opt, txn, &x.0, v).await?, _ => self.current.to_mut().set(&ctx, opt, txn, &x.0, v).await?,
}, },
Operator::Inc => { Operator::Inc => {
self.current.to_mut().increment(ctx, opt, txn, &x.0, v).await? self.current.to_mut().increment(&ctx, opt, txn, &x.0, v).await?
} }
Operator::Dec => { Operator::Dec => {
self.current.to_mut().decrement(ctx, opt, txn, &x.0, v).await? self.current.to_mut().decrement(&ctx, opt, txn, &x.0, v).await?
} }
_ => unreachable!(), _ => unreachable!(),
} }
} }
} }
Data::PatchExpression(data) => {
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
self.current.to_mut().patch(ctx, opt, txn, data).await?
}
Data::MergeExpression(data) => {
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
self.current.to_mut().merge(ctx, opt, txn, data).await?
}
Data::ReplaceExpression(data) => {
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
self.current.to_mut().replace(ctx, opt, txn, data).await?
}
Data::ContentExpression(data) => {
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
self.current.to_mut().replace(ctx, opt, txn, data).await?
}
_ => unreachable!(), _ => unreachable!(),
}; };
}; };