uqa-execution 0.3.0

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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

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

//! Current-session prepared statement definitions and plan selection counters.

use super::helpers::rows::{catalog_array, row};
use crate::catalog::services::CatalogSession;
use uqa_core::{TemporalValue, Value};
use uqa_sql::{ColumnType, ResultRow, SQLError};

pub fn rows(session: &dyn CatalogSession) -> Result<Vec<ResultRow>, SQLError> {
    session
        .prepared_statements()
        .into_iter()
        .map(|entry| {
            let parameters = type_array(entry.parameter_types.iter())?;
            let results = entry
                .result_types
                .as_ref()
                .map(|types| type_array(types.iter()))
                .transpose()?
                .unwrap_or(Value::Null);
            Ok(row([
                ("name", Value::Str(entry.name)),
                (
                    "statement",
                    entry
                        .source_sql
                        .map_or(Value::Null, |source| Value::Str(source.to_string())),
                ),
                (
                    "prepare_time",
                    Value::Temporal(TemporalValue::TimestampTz {
                        micros: entry.prepared_at_micros,
                    }),
                ),
                ("parameter_types", parameters),
                ("result_types", results),
                ("from_sql", Value::Bool(entry.from_sql)),
                ("generic_plans", Value::Int(entry.generic_plans)),
                ("custom_plans", Value::Int(entry.custom_plans)),
            ]))
        })
        .collect()
}

fn type_array<'a>(types: impl Iterator<Item = &'a Option<ColumnType>>) -> Result<Value, SQLError> {
    catalog_array(
        types
            .map(|ty| {
                ty.as_ref().map_or(Value::Null, |ty| {
                    let oid = match ty {
                        ColumnType::Domain { oid, .. } => *oid,
                        _ => super::postgres_result_type(ty).type_oid,
                    };
                    Value::Int(i64::from(oid))
                })
            })
            .collect(),
        "prepared statement types",
    )
}