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

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

use reifydb_core::{
	encoded::{key::EncodedKey, row::EncodedRow},
	interface::catalog::shape::ShapeId,
	key::{EncodableKey, column::ColumnKey, columns::ColumnsKey},
};
use reifydb_transaction::transaction::Transaction;

use super::CatalogChangeApplier;
use crate::{CatalogStore, Result, catalog::Catalog};

pub(super) struct ColumnApplier;

impl CatalogChangeApplier for ColumnApplier {
	fn set(catalog: &Catalog, txn: &mut Transaction<'_>, key: &EncodedKey, row: &EncodedRow) -> Result<()> {
		txn.set(key, row.clone())?;
		reload_parent_columns(catalog, txn, key)
	}

	fn remove(catalog: &Catalog, txn: &mut Transaction<'_>, key: &EncodedKey) -> Result<()> {
		txn.remove(key)?;
		reload_parent_columns(catalog, txn, key)
	}
}

fn reload_parent_columns(catalog: &Catalog, txn: &mut Transaction<'_>, key: &EncodedKey) -> Result<()> {
	let shape_id = if let Some(ck) = ColumnKey::decode(key) {
		Some(ck.shape)
	} else if let Some(_ck) = ColumnsKey::decode(key) {
		// ColumnsKey only has column_id, no parent shape - cannot determine parent
		return Ok(());
	} else {
		None
	};

	let shape_id = match shape_id {
		Some(id) => id,
		None => return Ok(()),
	};

	let version = txn.version();
	let columns = CatalogStore::list_columns(txn, shape_id)?;

	match shape_id {
		ShapeId::Table(id) => {
			if let Some(mut table) = catalog.materialized.find_table_at(id, version) {
				table.columns = columns;
				catalog.materialized.set_table(id, version, Some(table));
			}
		}
		ShapeId::View(id) => {
			if let Some(mut view) = catalog.materialized.find_view_at(id, version) {
				*view.columns_mut() = columns;
				catalog.materialized.set_view(id, version, Some(view));
			}
		}
		ShapeId::RingBuffer(id) => {
			if let Some(mut rb) = catalog.materialized.find_ringbuffer_at(id, version) {
				rb.columns = columns;
				catalog.materialized.set_ringbuffer(id, version, Some(rb));
			}
		}
		ShapeId::Series(id) => {
			if let Some(mut s) = catalog.materialized.find_series_at(id, version) {
				s.columns = columns;
				catalog.materialized.set_series(id, version, Some(s));
			}
		}
		_ => {}
	}

	Ok(())
}