87 lines
2.4 KiB
Rust
87 lines
2.4 KiB
Rust
|
use crate::ctx::Context;
|
||
|
|
use crate::dbs::Options;
|
||
|
|
use crate::dbs::Statement;
|
||
|
|
use crate::dbs::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.to_mut().def(ctx, opt, txn, rid).await?;
|
||
|
|
// The statement has a data clause
|
||
|
|
if let Some(v) = stm.data() {
|
||
|
|
match v {
|
||
|
|
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::Void => {
|
||
|
|
self.current.to_mut().del(ctx, opt, txn, &x.0).await?
|
||
|
|
}
|
||
|
|
_ => self.current.to_mut().set(ctx, opt, txn, &x.0, v).await?,
|
||
|
|
},
|
||
|
|
Operator::Inc => {
|
||
|
|
self.current.to_mut().increment(ctx, opt, txn, &x.0, v).await?
|
||
|
|
}
|
||
|
|
Operator::Dec => {
|
||
|
|
self.current.to_mut().decrement(ctx, opt, txn, &x.0, v).await?
|
||
|
|
}
|
||
|
|
_ => unreachable!(),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
Data::UpdateExpression(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::Void => {
|
||
|
|
self.current.to_mut().del(ctx, opt, txn, &x.0).await?
|
||
|
|
}
|
||
|
|
_ => self.current.to_mut().set(ctx, opt, txn, &x.0, v).await?,
|
||
|
|
},
|
||
|
|
Operator::Inc => {
|
||
|
|
self.current.to_mut().increment(ctx, opt, txn, &x.0, v).await?
|
||
|
|
}
|
||
|
|
Operator::Dec => {
|
||
|
|
self.current.to_mut().decrement(ctx, opt, txn, &x.0, v).await?
|
||
|
|
}
|
||
|
|
_ => unreachable!(),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
Data::PatchExpression(data) => {
|
||
|
|
self.current.to_mut().patch(ctx, opt, txn, data).await?
|
||
|
|
}
|
||
|
|
Data::MergeExpression(data) => {
|
||
|
|
self.current.to_mut().merge(ctx, opt, txn, data).await?
|
||
|
|
}
|
||
|
|
Data::ReplaceExpression(data) => {
|
||
|
|
self.current.to_mut().replace(ctx, opt, txn, data).await?
|
||
|
|
}
|
||
|
|
Data::ContentExpression(data) => {
|
||
|
|
self.current.to_mut().replace(ctx, opt, txn, data).await?
|
||
|
|
}
|
||
|
|
_ => unreachable!(),
|
||
|
|
};
|
||
|
|
};
|
||
|
|
// Set default field values
|
||
|
|
self.current.to_mut().def(ctx, opt, txn, rid).await?;
|
||
|
|
// Carry on
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
}
|