use reifydb_core::interface::catalog::{
change::CatalogTrackNamespaceChangeOperations, id::NamespaceId, namespace::Namespace,
};
use reifydb_type::Result;
use crate::{
change::{
Change,
OperationType::{Create, Delete, Update},
TransactionalNamespaceChanges,
},
transaction::{admin::AdminTransaction, subscription::SubscriptionTransaction},
};
impl CatalogTrackNamespaceChangeOperations for AdminTransaction {
fn track_namespace_created(&mut self, namespace: Namespace) -> Result<()> {
let change = Change {
pre: None,
post: Some(namespace),
op: Create,
};
self.changes.add_namespace_change(change);
Ok(())
}
fn track_namespace_updated(&mut self, pre: Namespace, post: Namespace) -> Result<()> {
let change = Change {
pre: Some(pre),
post: Some(post),
op: Update,
};
self.changes.add_namespace_change(change);
Ok(())
}
fn track_namespace_deleted(&mut self, namespace: Namespace) -> Result<()> {
let change = Change {
pre: Some(namespace),
post: None,
op: Delete,
};
self.changes.add_namespace_change(change);
Ok(())
}
}
impl TransactionalNamespaceChanges for AdminTransaction {
fn find_namespace(&self, id: NamespaceId) -> Option<&Namespace> {
for change in self.changes.namespace.iter().rev() {
if let Some(namespace) = &change.post {
if namespace.id() == id {
return Some(namespace);
}
} else if let Some(namespace) = &change.pre {
if namespace.id() == id && change.op == Delete {
return None;
}
}
}
None
}
fn find_namespace_by_name(&self, name: &str) -> Option<&Namespace> {
self.changes.namespace.iter().rev().find_map(|change| change.post.as_ref().filter(|s| s.name() == name))
}
fn is_namespace_deleted(&self, id: NamespaceId) -> bool {
self.changes
.namespace
.iter()
.rev()
.any(|change| change.op == Delete && change.pre.as_ref().map(|s| s.id()) == Some(id))
}
fn is_namespace_deleted_by_name(&self, name: &str) -> bool {
self.changes
.namespace
.iter()
.rev()
.any(|change| change.op == Delete && change.pre.as_ref().map(|s| s.name()) == Some(name))
}
}
impl CatalogTrackNamespaceChangeOperations for SubscriptionTransaction {
fn track_namespace_created(&mut self, namespace: Namespace) -> Result<()> {
self.inner.track_namespace_created(namespace)
}
fn track_namespace_updated(&mut self, pre: Namespace, post: Namespace) -> Result<()> {
self.inner.track_namespace_updated(pre, post)
}
fn track_namespace_deleted(&mut self, namespace: Namespace) -> Result<()> {
self.inner.track_namespace_deleted(namespace)
}
}
impl TransactionalNamespaceChanges for SubscriptionTransaction {
fn find_namespace(&self, id: NamespaceId) -> Option<&Namespace> {
self.inner.find_namespace(id)
}
fn find_namespace_by_name(&self, name: &str) -> Option<&Namespace> {
self.inner.find_namespace_by_name(name)
}
fn is_namespace_deleted(&self, id: NamespaceId) -> bool {
self.inner.is_namespace_deleted(id)
}
fn is_namespace_deleted_by_name(&self, name: &str) -> bool {
self.inner.is_namespace_deleted_by_name(name)
}
}