surrealpatch/core/src/doc/store.rs

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(())
	}
}