Skip to main content

reifydb_transaction/transaction/catalog/
dictionary.rs

1// SPDX-License-Identifier: Apache-2.0
2// Copyright (c) 2025 ReifyDB
3
4use reifydb_core::interface::catalog::{
5	change::CatalogTrackDictionaryChangeOperations, dictionary::DictionaryDef, id::NamespaceId,
6};
7use reifydb_type::{Result, value::dictionary::DictionaryId};
8
9use crate::{
10	change::{
11		Change,
12		OperationType::{Create, Delete, Update},
13		TransactionalDictionaryChanges,
14	},
15	transaction::{admin::AdminTransaction, subscription::SubscriptionTransaction},
16};
17
18impl CatalogTrackDictionaryChangeOperations for AdminTransaction {
19	fn track_dictionary_def_created(&mut self, dictionary: DictionaryDef) -> Result<()> {
20		let change = Change {
21			pre: None,
22			post: Some(dictionary),
23			op: Create,
24		};
25		self.changes.add_dictionary_def_change(change);
26		Ok(())
27	}
28
29	fn track_dictionary_def_updated(&mut self, pre: DictionaryDef, post: DictionaryDef) -> Result<()> {
30		let change = Change {
31			pre: Some(pre),
32			post: Some(post),
33			op: Update,
34		};
35		self.changes.add_dictionary_def_change(change);
36		Ok(())
37	}
38
39	fn track_dictionary_def_deleted(&mut self, dictionary: DictionaryDef) -> Result<()> {
40		let change = Change {
41			pre: Some(dictionary),
42			post: None,
43			op: Delete,
44		};
45		self.changes.add_dictionary_def_change(change);
46		Ok(())
47	}
48}
49
50impl TransactionalDictionaryChanges for AdminTransaction {
51	fn find_dictionary(&self, id: DictionaryId) -> Option<&DictionaryDef> {
52		for change in self.changes.dictionary_def.iter().rev() {
53			if let Some(dictionary) = &change.post {
54				if dictionary.id == id {
55					return Some(dictionary);
56				}
57			} else if let Some(dictionary) = &change.pre {
58				if dictionary.id == id && change.op == Delete {
59					return None;
60				}
61			}
62		}
63		None
64	}
65
66	fn find_dictionary_by_name(&self, namespace: NamespaceId, name: &str) -> Option<&DictionaryDef> {
67		self.changes
68			.dictionary_def
69			.iter()
70			.rev()
71			.find_map(|change| change.post.as_ref().filter(|d| d.namespace == namespace && d.name == name))
72	}
73
74	fn is_dictionary_deleted(&self, id: DictionaryId) -> bool {
75		self.changes
76			.dictionary_def
77			.iter()
78			.rev()
79			.any(|change| change.op == Delete && change.pre.as_ref().map(|d| d.id) == Some(id))
80	}
81
82	fn is_dictionary_deleted_by_name(&self, namespace: NamespaceId, name: &str) -> bool {
83		self.changes.dictionary_def.iter().rev().any(|change| {
84			change.op == Delete
85				&& change
86					.pre
87					.as_ref()
88					.map(|d| d.namespace == namespace && d.name == name)
89					.unwrap_or(false)
90		})
91	}
92}
93
94impl CatalogTrackDictionaryChangeOperations for SubscriptionTransaction {
95	fn track_dictionary_def_created(&mut self, dictionary: DictionaryDef) -> Result<()> {
96		self.inner.track_dictionary_def_created(dictionary)
97	}
98
99	fn track_dictionary_def_updated(&mut self, pre: DictionaryDef, post: DictionaryDef) -> Result<()> {
100		self.inner.track_dictionary_def_updated(pre, post)
101	}
102
103	fn track_dictionary_def_deleted(&mut self, dictionary: DictionaryDef) -> Result<()> {
104		self.inner.track_dictionary_def_deleted(dictionary)
105	}
106}
107
108impl TransactionalDictionaryChanges for SubscriptionTransaction {
109	fn find_dictionary(&self, id: DictionaryId) -> Option<&DictionaryDef> {
110		self.inner.find_dictionary(id)
111	}
112
113	fn find_dictionary_by_name(&self, namespace: NamespaceId, name: &str) -> Option<&DictionaryDef> {
114		self.inner.find_dictionary_by_name(namespace, name)
115	}
116
117	fn is_dictionary_deleted(&self, id: DictionaryId) -> bool {
118		self.inner.is_dictionary_deleted(id)
119	}
120
121	fn is_dictionary_deleted_by_name(&self, namespace: NamespaceId, name: &str) -> bool {
122		self.inner.is_dictionary_deleted_by_name(namespace, name)
123	}
124}