108 lines
3.2 KiB
Rust
108 lines
3.2 KiB
Rust
use crate::ctx::Context;
|
|
use crate::dbs::Statement;
|
|
use crate::dbs::Workable;
|
|
use crate::dbs::{Options, Transaction};
|
|
use crate::doc::Document;
|
|
use crate::err::Error;
|
|
use crate::sql::data::Data;
|
|
use crate::sql::operator::Operator;
|
|
use crate::sql::value::Value;
|
|
|
|
impl<'a> Document<'a> {
|
|
pub async fn alter(
|
|
&mut self,
|
|
ctx: &Context<'_>,
|
|
opt: &Options,
|
|
txn: &Transaction,
|
|
stm: &Statement<'_>,
|
|
) -> Result<(), Error> {
|
|
// Get the record id
|
|
let rid = self.id.as_ref().unwrap();
|
|
// Set default field values
|
|
self.current.doc.to_mut().def(rid);
|
|
// The statement has a data clause
|
|
if let Some(v) = stm.data() {
|
|
match v {
|
|
Data::PatchExpression(data) => {
|
|
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
|
|
self.current.doc.to_mut().patch(data)?
|
|
}
|
|
Data::MergeExpression(data) => {
|
|
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
|
|
self.current.doc.to_mut().merge(data)?
|
|
}
|
|
Data::ReplaceExpression(data) => {
|
|
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
|
|
self.current.doc.to_mut().replace(data)?
|
|
}
|
|
Data::ContentExpression(data) => {
|
|
let data = data.compute(ctx, opt, txn, Some(&self.current)).await?;
|
|
self.current.doc.to_mut().replace(data)?
|
|
}
|
|
Data::SetExpression(x) => {
|
|
for x in x.iter() {
|
|
let v = x.2.compute(ctx, opt, txn, Some(&self.current)).await?;
|
|
match x.1 {
|
|
Operator::Equal => match v {
|
|
Value::None => {
|
|
self.current.doc.to_mut().del(ctx, opt, txn, &x.0).await?
|
|
}
|
|
_ => self.current.doc.to_mut().set(ctx, opt, txn, &x.0, v).await?,
|
|
},
|
|
Operator::Inc => {
|
|
self.current.doc.to_mut().increment(ctx, opt, txn, &x.0, v).await?
|
|
}
|
|
Operator::Dec => {
|
|
self.current.doc.to_mut().decrement(ctx, opt, txn, &x.0, v).await?
|
|
}
|
|
Operator::Ext => {
|
|
self.current.doc.to_mut().extend(ctx, opt, txn, &x.0, v).await?
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
}
|
|
Data::UnsetExpression(i) => {
|
|
for i in i.iter() {
|
|
self.current.doc.to_mut().del(ctx, opt, txn, i).await?
|
|
}
|
|
}
|
|
Data::UpdateExpression(x) => {
|
|
// Duplicate context
|
|
let mut ctx = Context::new(ctx);
|
|
// Add insertable value
|
|
if let Workable::Insert(value) = &self.extras {
|
|
ctx.add_value("input", value);
|
|
}
|
|
// Process ON DUPLICATE KEY clause
|
|
for x in x.iter() {
|
|
let v = x.2.compute(&ctx, opt, txn, Some(&self.current)).await?;
|
|
match x.1 {
|
|
Operator::Equal => match v {
|
|
Value::None => {
|
|
self.current.doc.to_mut().del(&ctx, opt, txn, &x.0).await?
|
|
}
|
|
_ => self.current.doc.to_mut().set(&ctx, opt, txn, &x.0, v).await?,
|
|
},
|
|
Operator::Inc => {
|
|
self.current.doc.to_mut().increment(&ctx, opt, txn, &x.0, v).await?
|
|
}
|
|
Operator::Dec => {
|
|
self.current.doc.to_mut().decrement(&ctx, opt, txn, &x.0, v).await?
|
|
}
|
|
Operator::Ext => {
|
|
self.current.doc.to_mut().extend(&ctx, opt, txn, &x.0, v).await?
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
}
|
|
_ => unreachable!(),
|
|
};
|
|
};
|
|
// Set default field values
|
|
self.current.doc.to_mut().def(rid);
|
|
// Carry on
|
|
Ok(())
|
|
}
|
|
}
|