proof_of_sql/sql/proof_plans/
dyn_proof_plan.rs

1use super::{
2    EmptyExec, FilterExec, GroupByExec, ProjectionExec, SliceExec, SortMergeJoinExec, TableExec,
3    UnionExec,
4};
5use crate::{
6    base::{
7        database::{
8            ColumnField, ColumnRef, LiteralValue, OwnedTable, Table, TableEvaluation, TableRef,
9        },
10        map::{IndexMap, IndexSet},
11        proof::{PlaceholderResult, ProofError},
12        scalar::Scalar,
13    },
14    sql::{
15        proof::{
16            FinalRoundBuilder, FirstRoundBuilder, ProofPlan, ProverEvaluate, VerificationBuilder,
17        },
18        proof_exprs::{AliasedDynProofExpr, ColumnExpr, DynProofExpr, TableExpr},
19        AnalyzeResult,
20    },
21};
22use alloc::{boxed::Box, vec::Vec};
23use bumpalo::Bump;
24use serde::{Deserialize, Serialize};
25use sqlparser::ast::Ident;
26
27/// The query plan for proving a query
28#[derive(Debug, PartialEq, Serialize, Deserialize, Clone)]
29#[enum_dispatch::enum_dispatch]
30pub enum DynProofPlan {
31    /// Source [`ProofPlan`] for (sub)queries without table source such as `SELECT "No table here" as msg;`
32    Empty(EmptyExec),
33    /// Source [`ProofPlan`] for (sub)queries with table source such as `SELECT col from tab;`
34    Table(TableExec),
35    /// Provable expressions for queries of the form
36    /// ```ignore
37    ///     SELECT <result_expr1>, ..., <result_exprN> FROM <table>
38    /// ```
39    Projection(ProjectionExec),
40    /// Provable expressions for queries of the form
41    /// ```ignore
42    ///     SELECT <group_by_expr1>, ..., <group_by_exprM>,
43    ///         SUM(<sum_expr1>.0) as <sum_expr1>.1, ..., SUM(<sum_exprN>.0) as <sum_exprN>.1,
44    ///         COUNT(*) as count_alias
45    ///     FROM <table>
46    ///     WHERE <where_clause>
47    ///     GROUP BY <group_by_expr1>, ..., <group_by_exprM>
48    /// ```
49    GroupBy(GroupByExec),
50    /// Provable expressions for queries of the form, where the result is sent in a dense form
51    /// ```ignore
52    ///     SELECT <result_expr1>, ..., <result_exprN> FROM <table> WHERE <where_clause>
53    /// ```
54    Filter(FilterExec),
55    /// `ProofPlan` for queries of the form
56    /// ```ignore
57    ///     <ProofPlan> LIMIT <fetch> [OFFSET <skip>]
58    /// ```
59    Slice(SliceExec),
60    /// `ProofPlan` for queries of the form
61    /// ```ignore
62    ///     <ProofPlan>
63    ///     UNION ALL
64    ///     <ProofPlan>
65    ///     ...
66    ///     UNION ALL
67    ///     <ProofPlan>
68    /// ```
69    Union(UnionExec),
70    /// `ProofPlan` for queries of the form
71    /// ```ignore
72    ///     <ProofPlan> INNER JOIN <ProofPlan>
73    ///     ON col1 = col2
74    /// ```
75    SortMergeJoin(SortMergeJoinExec),
76}
77
78impl DynProofPlan {
79    /// Creates a new empty plan.
80    #[must_use]
81    pub fn new_empty() -> Self {
82        Self::Empty(EmptyExec::new())
83    }
84
85    /// Creates a new table plan.
86    #[must_use]
87    pub fn new_table(table_ref: TableRef, schema: Vec<ColumnField>) -> Self {
88        Self::Table(TableExec::new(table_ref, schema))
89    }
90
91    /// Creates a new projection plan.
92    #[must_use]
93    pub fn new_projection(aliased_results: Vec<AliasedDynProofExpr>, input: DynProofPlan) -> Self {
94        Self::Projection(ProjectionExec::new(aliased_results, Box::new(input)))
95    }
96
97    /// Creates a new filter plan.
98    #[must_use]
99    pub fn new_filter(
100        aliased_results: Vec<AliasedDynProofExpr>,
101        input: TableExpr,
102        filter_expr: DynProofExpr,
103    ) -> Self {
104        Self::Filter(FilterExec::new(aliased_results, input, filter_expr))
105    }
106
107    /// Creates a new group by plan.
108    #[must_use]
109    pub fn new_group_by(
110        group_by_exprs: Vec<ColumnExpr>,
111        sum_expr: Vec<AliasedDynProofExpr>,
112        count_alias: Ident,
113        table: TableExpr,
114        where_clause: DynProofExpr,
115    ) -> Self {
116        Self::GroupBy(GroupByExec::new(
117            group_by_exprs,
118            sum_expr,
119            count_alias,
120            table,
121            where_clause,
122        ))
123    }
124
125    /// Creates a new slice plan.
126    #[must_use]
127    pub fn new_slice(input: DynProofPlan, skip: usize, fetch: Option<usize>) -> Self {
128        Self::Slice(SliceExec::new(Box::new(input), skip, fetch))
129    }
130
131    /// Creates a new union plan.
132    pub fn try_new_union(inputs: Vec<DynProofPlan>) -> AnalyzeResult<Self> {
133        UnionExec::try_new(inputs).map(Self::Union)
134    }
135}
proof_of_sql/sql/proof_plans/dyn_proof_plan.rs

proof_of_sql/sql/proof_plans/
dyn_proof_plan.rs
