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

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

use reifydb_core::{
	encoded::{key::EncodedKey, row::EncodedRow},
	interface::catalog::{
		flow::FlowStatus,
		id::{NamespaceId, SinkId},
		sink::Sink,
	},
	key::{EncodableKey, kind::KeyKind, sink::SinkKey},
};
use reifydb_transaction::transaction::Transaction;
use serde_json::from_str;

use super::CatalogChangeApplier;
use crate::{
	Result,
	catalog::Catalog,
	error::CatalogChangeError,
	store::sink::shape::sink::{
		self, CONFIG, CONNECTOR, ID, NAME, NAMESPACE, SOURCE_NAME, SOURCE_NAMESPACE, STATUS,
	},
};

pub(super) struct SinkApplier;

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

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

fn decode_sink(row: &EncodedRow) -> Sink {
	let id = SinkId(sink::SHAPE.get_u64(row, ID));
	let namespace = NamespaceId(sink::SHAPE.get_u64(row, NAMESPACE));
	let name = sink::SHAPE.get_utf8(row, NAME).to_string();
	let source_namespace = NamespaceId(sink::SHAPE.get_u64(row, SOURCE_NAMESPACE));
	let source_name = sink::SHAPE.get_utf8(row, SOURCE_NAME).to_string();
	let connector = sink::SHAPE.get_utf8(row, CONNECTOR).to_string();
	let config_json = sink::SHAPE.get_utf8(row, CONFIG);
	let config: Vec<(String, String)> = from_str(config_json).unwrap_or_default();
	let status = FlowStatus::from_u8(sink::SHAPE.get_u8(row, STATUS));

	Sink {
		id,
		namespace,
		name,
		source_namespace,
		source_name,
		connector,
		config,
		status,
	}
}