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

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

use crate::{
	Result,
	ast::ast::{AstTake, AstTakeValue},
	nodes::TakeLimit,
	plan::logical::{Compiler, LogicalPlan, TakeNode},
};

impl<'bump> Compiler<'bump> {
	pub(crate) fn compile_take(&self, ast: AstTake<'bump>) -> Result<LogicalPlan<'bump>> {
		let take = match ast.take {
			AstTakeValue::Literal(n) => TakeLimit::Literal(n),
			AstTakeValue::Variable(tok) => {
				let name = tok.fragment.text().trim_start_matches('$').to_string();
				TakeLimit::Variable(name)
			}
		};
		Ok(LogicalPlan::Take(TakeNode {
			take,
		}))
	}
}