reifydb-catalog 0.4.13

Database catalog and metadata management for ReifyDB
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2025 ReifyDB

use reifydb_core::{
	encoded::{key::EncodedKey, row::EncodedRow},
	interface::catalog::policy::{Policy, PolicyTargetType},
	key::{EncodableKey, kind::KeyKind, policy::PolicyKey},
};
use reifydb_transaction::transaction::Transaction;

use super::CatalogChangeApplier;
use crate::{Result, catalog::Catalog, error::CatalogChangeError, store::policy::shape::policy};

pub(super) struct PolicyApplier;

impl CatalogChangeApplier for PolicyApplier {
	fn set(catalog: &Catalog, txn: &mut Transaction<'_>, key: &EncodedKey, row: &EncodedRow) -> Result<()> {
		txn.set(key, row.clone())?;
		let p = decode_policy(row);
		catalog.materialized.set_policy(p.id, txn.version(), Some(p));
		Ok(())
	}

	fn remove(catalog: &Catalog, txn: &mut Transaction<'_>, key: &EncodedKey) -> Result<()> {
		txn.remove(key)?;
		let id = PolicyKey::decode(key).map(|k| k.policy).ok_or(CatalogChangeError::KeyDecodeFailed {
			kind: KeyKind::Policy,
		})?;
		catalog.materialized.set_policy(id, txn.version(), None);
		Ok(())
	}
}

fn decode_policy(row: &EncodedRow) -> Policy {
	let id = policy::SHAPE.get_u64(row, policy::ID);
	let name_str = policy::SHAPE.get_utf8(row, policy::NAME).to_string();
	let name = if name_str.is_empty() {
		None
	} else {
		Some(name_str)
	};
	let target_type_str = policy::SHAPE.get_utf8(row, policy::TARGET_TYPE);
	let target_type = match target_type_str {
		"table" => PolicyTargetType::Table,
		"column" => PolicyTargetType::Column,
		"namespace" => PolicyTargetType::Namespace,
		"procedure" => PolicyTargetType::Procedure,
		"function" => PolicyTargetType::Function,
		"subscription" => PolicyTargetType::Subscription,
		"series" => PolicyTargetType::Series,
		"dictionary" => PolicyTargetType::Dictionary,
		"session" => PolicyTargetType::Session,
		"feature" => PolicyTargetType::Feature,
		"view" => PolicyTargetType::View,
		"ringbuffer" => PolicyTargetType::RingBuffer,
		_ => PolicyTargetType::Table,
	};
	let target_ns_str = policy::SHAPE.get_utf8(row, policy::TARGET_NAMESPACE).to_string();
	let target_namespace = if target_ns_str.is_empty() {
		None
	} else {
		Some(target_ns_str)
	};
	let target_shape_str = policy::SHAPE.get_utf8(row, policy::TARGET_SHAPE).to_string();
	let target_shape = if target_shape_str.is_empty() {
		None
	} else {
		Some(target_shape_str)
	};
	let enabled = policy::SHAPE.get_bool(row, policy::ENABLED);

	Policy {
		id,
		name,
		target_type,
		target_namespace,
		target_shape,
		enabled,
	}
}