reifydb-engine 0.10.0

Query execution and processing engine 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

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

use arrow_array::RecordBatch;
use reifydb_catalog::error::{CatalogError, CatalogObjectKind};
use reifydb_core::{
	flow::operator::{LookupObject, OperatorDef},
	value::batch::single_row,
};
use reifydb_rql::nodes::DropRingBufferNode;
use reifydb_transaction::transaction::{Transaction, admin::AdminTransaction};
use reifydb_value::value::Value;

use super::dependent::find_flow_dependents;
use crate::{Result, vm::services::Services};

pub(crate) fn drop_ringbuffer(
	services: &Services,
	txn: &mut AdminTransaction,
	plan: DropRingBufferNode,
) -> Result<RecordBatch> {
	let Some(ringbuffer_id) = plan.ringbuffer_id else {
		return single_row([
			("namespace", Value::Utf8(plan.namespace_name.text().to_string())),
			("ringbuffer", Value::Utf8(plan.ringbuffer_name.text().to_string())),
			("dropped", Value::Boolean(false)),
		]);
	};

	let def = services.catalog.get_ringbuffer(&mut Transaction::Admin(txn), ringbuffer_id)?;

	let dags = services.catalog.list_flow_dags_asc(&mut Transaction::Admin(txn))?;
	let flows = services.catalog.list_flows_all(&mut Transaction::Admin(txn))?;
	let dependents = find_flow_dependents(&services.catalog, txn, &dags, &flows, |node_type| {
		matches!(node_type, OperatorDef::SourceRingBuffer { ringbuffer, .. } if *ringbuffer == ringbuffer_id)
			|| matches!(node_type, OperatorDef::Lookup { right: LookupObject::RingBuffer(ringbuffer), .. } if *ringbuffer == ringbuffer_id)
	})?;
	if !dependents.is_empty() {
		let dependents_str = dependents.join(", ");
		return Err(CatalogError::InUse {
			kind: CatalogObjectKind::RingBuffer,
			namespace: plan.namespace_name.text().to_string(),
			name: Some(plan.ringbuffer_name.text().to_string()),
			dependents: dependents_str,
			fragment: plan.ringbuffer_name.clone(),
		}
		.into());
	}

	services.catalog.drop_ringbuffer(txn, def)?;

	single_row([
		("namespace", Value::Utf8(plan.namespace_name.text().to_string())),
		("ringbuffer", Value::Utf8(plan.ringbuffer_name.text().to_string())),
		("dropped", Value::Boolean(true)),
	])
}