surrealdb-core 3.3.1

A scalable, distributed, collaborative, document-graph database, for the realtime web
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use anyhow::Result;

use crate::catalog::Error;
use crate::catalog::providers::DatabaseProvider;
use crate::ctx::FrozenContext;
use crate::dbs::Options;
use crate::expr::Base;
use crate::expr::model::get_model_path;
use crate::expr::statements::remove::model::RemoveModelStatement;
use crate::iam::{Action, ResourceKind};
use crate::key::schema::MlModelKey;
use crate::val::Value;

/// Process this type returning a computed simple Value
pub(crate) async fn remove_model_statement_compute(
	this: &RemoveModelStatement,
	ctx: &FrozenContext,
	opt: &Options,
) -> Result<Value> {
	// Allowed to run?
	ctx.is_allowed(opt, Action::Edit, ResourceKind::Model, Base::Db)?;
	// Get the transaction
	let txn = ctx.tx();
	// Get the defined model
	let (ns, db) = ctx.expect_ns_db_ids(opt).await?;
	let ml = match txn.get_db_model(ns, db, &this.name, &this.version, None).await? {
		Some(x) => x,
		None => {
			if this.if_exists {
				return Ok(Value::None);
			}
			return Err(Error::MlNotFound {
				name: format!("{}<{}>", this.name.as_str(), this.version.as_str()),
			}
			.into());
		}
	};
	// Delete the definition
	let key = MlModelKey {
		ns,
		db,
		ml: std::borrow::Cow::Borrowed(&ml.name),
		vn: std::borrow::Cow::Borrowed(&ml.version),
	};
	txn.del_key(&key).await?;
	// Clear the cache
	txn.clear_cache();
	// `obs::del` is idempotent, so this is safe even for definitions
	// registered without an uploaded artifact (e.g. via import).
	let (ns_name, db_name) = opt.ns_db()?;
	let path = get_model_path(ns_name, db_name, &ml.name, &ml.version, &ml.hash);
	crate::obs::del(&path).await?;
	// Ok all good
	Ok(Value::None)
}