reifydb-catalog 0.4.8

Database catalog and metadata management for ReifyDB
Documentation 
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2025 ReifyDB

use reifydb_core::{
	common::CommitVersion,
	interface::catalog::{
		id::{NamespaceId, ProcedureId},
		procedure::{Procedure, ProcedureTrigger},
	},
};
use reifydb_type::value::sumtype::VariantRef;

use crate::materialized::{MaterializedCatalog, MultiVersionProcedure};

impl MaterializedCatalog {
	/// Find a procedure by ID at a specific version
	pub fn find_procedure_at(&self, procedure: ProcedureId, version: CommitVersion) -> Option<Procedure> {
		self.procedures.get(&procedure).and_then(|entry| {
			let multi = entry.value();
			multi.get(version)
		})
	}

	/// Find a procedure by name in a namespace at a specific version
	pub fn find_procedure_by_name_at(
		&self,
		namespace: NamespaceId,
		name: &str,
		version: CommitVersion,
	) -> Option<Procedure> {
		self.procedures_by_name.get(&(namespace, name.to_string())).and_then(|entry| {
			let procedure_id = *entry.value();
			self.find_procedure_at(procedure_id, version)
		})
	}

	/// Find a procedure by ID (returns latest version)
	pub fn find_procedure(&self, procedure: ProcedureId) -> Option<Procedure> {
		self.procedures.get(&procedure).and_then(|entry| {
			let multi = entry.value();
			multi.get_latest()
		})
	}

	/// Find a procedure by name in a namespace (returns latest version)
	pub fn find_procedure_by_name(&self, namespace: NamespaceId, name: &str) -> Option<Procedure> {
		self.procedures_by_name.get(&(namespace, name.to_string())).and_then(|entry| {
			let procedure_id = *entry.value();
			self.find_procedure(procedure_id)
		})
	}

	/// List all procedures for a specific event variant at a specific version
	pub fn list_procedures_for_variant_at(&self, variant: VariantRef, version: CommitVersion) -> Vec<Procedure> {
		if let Some(entry) = self.procedures_by_variant.get(&variant) {
			entry.value().iter().filter_map(|id| self.find_procedure_at(*id, version)).collect()
		} else {
			vec![]
		}
	}

	pub fn set_procedure(&self, id: ProcedureId, version: CommitVersion, procedure: Option<Procedure>) {
		if let Some(entry) = self.procedures.get(&id)
			&& let Some(pre) = entry.value().get_latest()
		{
			// Remove old name from index
			self.procedures_by_name.remove(&(pre.namespace, pre.name.clone()));

			// Remove from variant index if it had an event binding
			if let ProcedureTrigger::Event {
				variant,
			} = &pre.trigger && let Some(ids_entry) = self.procedures_by_variant.get(variant)
			{
				let mut ids = ids_entry.value().clone();
				ids.retain(|existing| *existing != id);
				drop(ids_entry);
				self.procedures_by_variant.insert(*variant, ids);
			}
		}

		let multi = self.procedures.get_or_insert_with(id, MultiVersionProcedure::new);
		if let Some(new) = procedure {
			self.procedures_by_name.insert((new.namespace, new.name.clone()), id);

			// Add to variant index if it has an event binding
			if let ProcedureTrigger::Event {
				variant,
			} = &new.trigger
			{
				if let Some(entry) = self.procedures_by_variant.get(variant) {
					let mut ids = entry.value().clone();
					if !ids.contains(&id) {
						ids.push(id);
					}
					drop(entry);
					self.procedures_by_variant.insert(*variant, ids);
				} else {
					self.procedures_by_variant.insert(*variant, vec![id]);
				}
			}

			multi.value().insert(version, new);
		} else {
			multi.value().remove(version);
		}
	}
}