nodedb-sql 0.6.2

SQL parser, planner, and optimizer for NodeDB
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

// SPDX-License-Identifier: Apache-2.0

//! Catalog wrapper that resolves CTE and derived-alias names as relations.

use nodedb_types::DatabaseId;

use crate::resolver::derived::open_subquery_relation;
use crate::types::{CollectionInfo, SqlCatalog, SqlCatalogError};

/// Catalog wrapper that answers for a synthesized relation before delegating
/// to the stored catalog.
pub(crate) struct CteCatalog<'a> {
    pub(crate) inner: &'a dyn SqlCatalog,
    /// Each synthesized relation name paired with the shape its body exposes.
    pub(crate) relations: Vec<(String, CollectionInfo)>,
}

impl<'a> CteCatalog<'a> {
    /// A catalog exposing one relation whose shape is not inferable.
    ///
    /// The recursive arm of a `WITH RECURSIVE` names the working table while
    /// planning its own body, so that arm's shape is not known yet.
    pub(crate) fn open(inner: &'a dyn SqlCatalog, name: &str) -> Self {
        Self {
            inner,
            relations: vec![(name.to_string(), open_subquery_relation(name))],
        }
    }
}

impl SqlCatalog for CteCatalog<'_> {
    fn get_collection(
        &self,
        database_id: DatabaseId,
        name: &str,
    ) -> std::result::Result<Option<CollectionInfo>, SqlCatalogError> {
        if let Some((_, info)) = self.relations.iter().find(|(key, _)| key == name) {
            return Ok(Some(info.clone()));
        }
        self.inner.get_collection(database_id, name)
    }
}