use reifydb_core::{
interface::catalog::{change::CatalogTrackRowSettingsChangeOperations, shape::ShapeId},
row::RowSettings,
};
use reifydb_value::Result;
use crate::{
change::{
Change,
OperationType::{Create, Delete, Update},
TransactionalRowSettingsChanges,
},
transaction::admin::AdminTransaction,
};
impl CatalogTrackRowSettingsChangeOperations for AdminTransaction {
fn track_row_settings_created(&mut self, shape: ShapeId, settings: RowSettings) -> Result<()> {
let change = Change {
pre: None,
post: Some((shape, settings)),
op: Create,
};
self.changes.add_row_settings_change(change);
Ok(())
}
fn track_row_settings_updated(&mut self, shape: ShapeId, pre: RowSettings, post: RowSettings) -> Result<()> {
let change = Change {
pre: Some((shape, pre)),
post: Some((shape, post)),
op: Update,
};
self.changes.add_row_settings_change(change);
Ok(())
}
fn track_row_settings_deleted(&mut self, shape: ShapeId, settings: RowSettings) -> Result<()> {
let change = Change {
pre: Some((shape, settings)),
post: None,
op: Delete,
};
self.changes.add_row_settings_change(change);
Ok(())
}
}
impl TransactionalRowSettingsChanges for AdminTransaction {
fn find_row_settings(&self, shape: ShapeId) -> Option<&RowSettings> {
for change in self.changes.row_settings.iter().rev() {
if let Some((s, settings)) = &change.post {
if *s == shape {
return Some(settings);
}
} else if let Some((s, _)) = &change.pre
&& *s == shape && change.op == Delete
{
return None;
}
}
None
}
fn is_row_settings_deleted(&self, shape: ShapeId) -> bool {
self.changes.row_settings.iter().rev().any(|change| {
change.op == Delete && change.pre.as_ref().map(|(s, _)| *s == shape).unwrap_or(false)
})
}
}