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

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

use reifydb_core::{
	interface::catalog::migration::{Migration, MigrationAction, MigrationEvent},
	key::{migration::MigrationKey, migration_event::MigrationEventKey},
};
use reifydb_runtime::hash::Hash128;
use reifydb_transaction::transaction::admin::AdminTransaction;

use crate::{
	CatalogStore, Result,
	store::{
		migration::shape::{migration as migration_shape, migration_event as event_shape},
		sequence::system::SystemSequence,
	},
};

pub struct MigrationToCreate {
	pub name: String,
	pub body: String,
	pub rollback_body: Option<String>,
	pub hash: Hash128,
}

impl CatalogStore {
	pub(crate) fn create_migration(txn: &mut AdminTransaction, to_create: MigrationToCreate) -> Result<Migration> {
		let migration_id = SystemSequence::next_migration_id(txn)?;

		let mut row = migration_shape::SHAPE.allocate();
		migration_shape::SHAPE.set_u64(&mut row, migration_shape::ID, migration_id);
		migration_shape::SHAPE.set_utf8(&mut row, migration_shape::NAME, &to_create.name);
		migration_shape::SHAPE.set_utf8(&mut row, migration_shape::BODY, &to_create.body);
		migration_shape::SHAPE.set_utf8(
			&mut row,
			migration_shape::ROLLBACK_BODY,
			to_create.rollback_body.as_deref().unwrap_or(""),
		);
		migration_shape::SHAPE.set_u128(&mut row, migration_shape::HASH, to_create.hash.0);

		txn.set(&MigrationKey::encoded(migration_id), row)?;

		Ok(Migration {
			id: migration_id,
			name: to_create.name,
			body: to_create.body,
			rollback_body: to_create.rollback_body,
			hash: to_create.hash,
		})
	}

	pub(crate) fn create_migration_event(
		txn: &mut AdminTransaction,
		migration: &Migration,
		action: MigrationAction,
	) -> Result<MigrationEvent> {
		let event_id = SystemSequence::next_migration_event_id(txn)?;

		let mut row = event_shape::SHAPE.allocate();
		event_shape::SHAPE.set_u64(&mut row, event_shape::ID, event_id);
		event_shape::SHAPE.set_u64(&mut row, event_shape::MIGRATION_ID, migration.id);
		event_shape::SHAPE.set_u8(
			&mut row,
			event_shape::ACTION,
			match action {
				MigrationAction::Applied => 0,
				MigrationAction::Rollback => 1,
			},
		);

		txn.set(&MigrationEventKey::encoded(event_id), row)?;

		Ok(MigrationEvent {
			id: event_id,
			migration_id: migration.id,
			action,
		})
	}
}