icydb-core 0.257.3

IcyDB — A schema-first typed query engine and persistence runtime for Internet Computer canisters
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

//! Copy admitted expression operands without normalization or a sizing walk.

use crate::db::{
    QueryError,
    query::{
        plan::expr::{CaseWhenArm, Expr, FieldId, FieldPath},
        preparation::PreparationWork,
    },
};
use icydb_diagnostic_code::DiagnosticExecutionBudgetResource as Resource;

impl PreparationWork<'_> {
    /// Copy an input-admitted tree under the current request. Callers must
    /// validate input depth before entering this recursive construction path;
    /// partial results cannot exceed that depth and are discarded on failure.
    pub(in crate::db) fn copy_expr(&self, expr: &Expr) -> Result<Expr, QueryError> {
        self.charge(Resource::PredicateExpressionSteps, 1)?;
        Ok(match expr {
            Expr::Field(field) => Expr::Field(FieldId::new(self.copy_text(field.as_str())?)),
            Expr::FieldPath(path) => Expr::FieldPath(FieldPath::new(
                self.copy_text(path.root().as_str())?,
                self.copy_slice(path.segments(), |segment| self.copy_text(segment))?,
            )),
            Expr::Literal(value) => Expr::Literal(self.copy_value(value)?),
            Expr::FunctionCall { function, args } => Expr::FunctionCall {
                function: *function,
                args: self.copy_slice(args, |arg| self.copy_expr(arg))?,
            },
            Expr::Unary { op, expr } => Expr::Unary {
                op: *op,
                expr: self.copy_boxed_expr(expr)?,
            },
            Expr::Binary { op, left, right } => Expr::Binary {
                op: *op,
                left: self.copy_boxed_expr(left)?,
                right: self.copy_boxed_expr(right)?,
            },
            Expr::Case {
                when_then_arms,
                else_expr,
            } => Expr::Case {
                when_then_arms: self.copy_slice(when_then_arms, |arm| {
                    Ok(CaseWhenArm::new(
                        self.copy_expr(arm.condition())?,
                        self.copy_expr(arm.result())?,
                    ))
                })?,
                else_expr: self.copy_boxed_expr(else_expr)?,
            },
            Expr::Aggregate(aggregate) => Expr::Aggregate(aggregate.copy_for_preparation(self)?),
            #[cfg(test)]
            Expr::Alias { expr, name } => Expr::Alias {
                expr: self.copy_boxed_expr(expr)?,
                name: self.copy_text(name.as_str())?.into(),
            },
        })
    }

    /// Charge the retained operand box before copying its admitted child.
    pub(in crate::db) fn copy_boxed_expr(&self, expr: &Expr) -> Result<Box<Expr>, QueryError> {
        self.charge(Resource::TemporaryBytes, size_of::<Expr>() as u64)?;
        Ok(Box::new(self.copy_expr(expr)?))
    }
}

#[cfg(test)]
mod tests;