reifydb_transaction/transaction/catalog/
dictionary.rs1use 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}