uqa-execution 0.3.5

Volcano physical operators with row-batch pipelines
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

//
// Unified Query Algebra
//
// Copyright (c) 2023-2026 Cognica, Inc.
//

//! Save and restore materialized, deferred, and non-returning CTE bindings.

use crate::query::CteScope;
use std::collections::BTreeSet;
use uqa_sql::plan::CtePlan;

pub fn save_and_remove_cte_names<S: Clone>(
    ctes: &mut CteScope<S>,
    names: &BTreeSet<String>,
) -> Vec<SavedCteBinding> {
    names
        .iter()
        .map(|name| SavedCteBinding {
            name: name.clone(),
            non_returning: ctes.non_returning_ctes.contains(name),
            rows: ctes.remove_materialized(name),
            deferred: ctes.remove_deferred(name),
        })
        .collect()
}

pub fn restore_cte_names<S: Clone>(ctes: &mut CteScope<S>, saved: Vec<SavedCteBinding>) {
    for binding in saved {
        let name = binding.name.clone();
        ctes.remove_materialized(&binding.name);
        ctes.remove_deferred(&binding.name);
        if let Some(rows) = binding.rows {
            ctes.insert_shared(binding.name, rows);
        } else if let Some(plan) = binding.deferred {
            ctes.insert_deferred(plan);
        }
        if binding.non_returning {
            ctes.non_returning_ctes.insert(name);
        }
    }
}

pub struct SavedCteBinding {
    name: String,
    non_returning: bool,
    rows: Option<crate::SharedSpill>,
    deferred: Option<CtePlan>,
}