49 lines
1.3 KiB
Rust
49 lines
1.3 KiB
Rust
use crate::ctx::Context;
|
|
use crate::dbs::Options;
|
|
use crate::dbs::Statement;
|
|
use crate::doc::Document;
|
|
use crate::err::Error;
|
|
|
|
impl Document {
|
|
pub async fn store(
|
|
&self,
|
|
ctx: &Context,
|
|
opt: &Options,
|
|
stm: &Statement<'_>,
|
|
) -> Result<(), Error> {
|
|
// Check if changed
|
|
if !self.changed() {
|
|
return Ok(());
|
|
}
|
|
// Check if the table is a view
|
|
if self.tb(ctx, opt).await?.drop {
|
|
return Ok(());
|
|
}
|
|
// Get the transaction
|
|
let txn = ctx.tx();
|
|
// Get the record id
|
|
let rid = self.id.as_ref().unwrap();
|
|
// Store the record data
|
|
let key = crate::key::thing::new(opt.ns()?, opt.db()?, &rid.tb, &rid.id);
|
|
// Match the statement type
|
|
match stm {
|
|
// This is a CREATE statement so try to insert the key
|
|
Statement::Create(_) => match txn.put(key, self, opt.version).await {
|
|
// The key already exists, so return an error
|
|
Err(Error::TxKeyAlreadyExists) => Err(Error::RecordExists {
|
|
thing: rid.to_string(),
|
|
}),
|
|
// Return any other received error
|
|
Err(e) => Err(e),
|
|
// Record creation worked fine
|
|
Ok(v) => Ok(v),
|
|
},
|
|
// INSERT can be versioned
|
|
Statement::Insert(_) => txn.set(key, self, opt.version).await,
|
|
// This is not a CREATE statement, so update the key
|
|
_ => txn.set(key, self, None).await,
|
|
}?;
|
|
// Carry on
|
|
Ok(())
|
|
}
|
|
}
|