use reifydb_core::{
encoded::{key::EncodedKey, row::EncodedRow},
interface::catalog::identity::Identity,
key::{EncodableKey, identity::IdentityKey, kind::KeyKind},
};
use reifydb_transaction::transaction::Transaction;
use super::CatalogChangeApplier;
use crate::{Result, catalog::Catalog, error::CatalogChangeError, store::identity::shape::identity};
pub(super) struct IdentityApplier;
impl CatalogChangeApplier for IdentityApplier {
fn set(catalog: &Catalog, txn: &mut Transaction<'_>, key: &EncodedKey, row: &EncodedRow) -> Result<()> {
txn.set(key, row.clone())?;
let id_entity = decode_identity(row);
catalog.materialized.set_identity(id_entity.id, txn.version(), Some(id_entity));
Ok(())
}
fn remove(catalog: &Catalog, txn: &mut Transaction<'_>, key: &EncodedKey) -> Result<()> {
txn.remove(key)?;
let id = IdentityKey::decode(key).map(|k| k.identity).ok_or(CatalogChangeError::KeyDecodeFailed {
kind: KeyKind::Identity,
})?;
catalog.materialized.set_identity(id, txn.version(), None);
Ok(())
}
}
fn decode_identity(row: &EncodedRow) -> Identity {
let id = identity::SHAPE.get_identity_id(row, identity::IDENTITY);
let name = identity::SHAPE.get_utf8(row, identity::NAME).to_string();
let enabled = identity::SHAPE.get_bool(row, identity::ENABLED);
Identity {
id,
name,
enabled,
}
}