reifydb_transaction/transaction/catalog/
handler.rs

1// SPDX-License-Identifier: Apache-2.0
2// Copyright (c) 2025 ReifyDB
3
4use reifydb_core::interface::catalog::{
5	change::CatalogTrackHandlerChangeOperations,
6	handler::HandlerDef,
7	id::{HandlerId, NamespaceId},
8};
9use reifydb_type::Result;
10
11use crate::{
12	change::{
13		Change,
14		OperationType::{Create, Delete},
15		TransactionalHandlerChanges,
16	},
17	transaction::{admin::AdminTransaction, subscription::SubscriptionTransaction},
18};
19
20impl CatalogTrackHandlerChangeOperations for AdminTransaction {
21	fn track_handler_def_created(&mut self, handler: HandlerDef) -> Result<()> {
22		let change = Change {
23			pre: None,
24			post: Some(handler),
25			op: Create,
26		};
27		self.changes.add_handler_def_change(change);
28		Ok(())
29	}
30
31	fn track_handler_def_deleted(&mut self, handler: HandlerDef) -> Result<()> {
32		let change = Change {
33			pre: Some(handler),
34			post: None,
35			op: Delete,
36		};
37		self.changes.add_handler_def_change(change);
38		Ok(())
39	}
40}
41
42impl TransactionalHandlerChanges for AdminTransaction {
43	fn find_handler_by_id(&self, id: HandlerId) -> Option<&HandlerDef> {
44		for change in self.changes.handler_def.iter().rev() {
45			if let Some(handler) = &change.post {
46				if handler.id == id {
47					return Some(handler);
48				}
49			} else if let Some(handler) = &change.pre {
50				if handler.id == id && change.op == Delete {
51					return None;
52				}
53			}
54		}
55		None
56	}
57
58	fn find_handler_by_name(&self, namespace: NamespaceId, name: &str) -> Option<&HandlerDef> {
59		self.changes
60			.handler_def
61			.iter()
62			.rev()
63			.find_map(|change| change.post.as_ref().filter(|h| h.namespace == namespace && h.name == name))
64	}
65
66	fn is_handler_deleted_by_name(&self, namespace: NamespaceId, name: &str) -> bool {
67		self.changes.handler_def.iter().rev().any(|change| {
68			change.op == Delete
69				&& change
70					.pre
71					.as_ref()
72					.map(|h| h.namespace == namespace && h.name == name)
73					.unwrap_or(false)
74		})
75	}
76}
77
78impl CatalogTrackHandlerChangeOperations for SubscriptionTransaction {
79	fn track_handler_def_created(&mut self, handler: HandlerDef) -> Result<()> {
80		self.inner.track_handler_def_created(handler)
81	}
82
83	fn track_handler_def_deleted(&mut self, handler: HandlerDef) -> Result<()> {
84		self.inner.track_handler_def_deleted(handler)
85	}
86}
87
88impl TransactionalHandlerChanges for SubscriptionTransaction {
89	fn find_handler_by_id(&self, id: HandlerId) -> Option<&HandlerDef> {
90		self.inner.find_handler_by_id(id)
91	}
92
93	fn find_handler_by_name(&self, namespace: NamespaceId, name: &str) -> Option<&HandlerDef> {
94		self.inner.find_handler_by_name(namespace, name)
95	}
96
97	fn is_handler_deleted_by_name(&self, namespace: NamespaceId, name: &str) -> bool {
98		self.inner.is_handler_deleted_by_name(namespace, name)
99	}
100}
reifydb_transaction/transaction/catalog/handler.rs

reifydb_transaction/transaction/catalog/
handler.rs
