reifydb-rql 0.4.6

ReifyDB Query Language (RQL) parser and AST
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

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

use reifydb_catalog::catalog::subscription::SubscriptionColumnToCreate;
use reifydb_transaction::transaction::Transaction;

use crate::{
	Result,
	ast::ast::AstCreateSubscription,
	bump::BumpVec,
	convert_data_type_with_constraints,
	plan::logical::{Compiler, CreateSubscriptionNode, LogicalPlan},
};

impl<'bump> Compiler<'bump> {
	pub(crate) fn compile_create_subscription(
		&self,
		ast: AstCreateSubscription<'bump>,
		tx: &mut Transaction<'_>,
	) -> Result<LogicalPlan<'bump>> {
		let mut columns = Vec::with_capacity(ast.columns.len());

		for col in ast.columns.iter() {
			let constraint = convert_data_type_with_constraints(&col.ty)?;
			columns.push(SubscriptionColumnToCreate {
				name: col.name.text().to_string(),
				ty: constraint.get_type(),
			});
		}

		// Compile the AS clause to logical plans
		let as_clause = if let Some(as_statement) = ast.as_clause {
			self.compile(as_statement, tx)?
		} else {
			BumpVec::new_in(self.bump)
		};

		Ok(LogicalPlan::CreateSubscription(CreateSubscriptionNode {
			columns,
			as_clause,
		}))
	}
}