mockgres 0.0.29

An in-memory database that replicates a reasonable subset of Postgres functionality to make unit tests that rely on a database to run.
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

use crate::catalog::SchemaId;
use crate::db::Db;
use crate::engine::{
    ColumnRefName, DataType, Field, ReturningClause, ReturningExpr, ScalarExpr, Schema,
};
use pgwire::error::PgWireResult;

use super::BindTimeContext;
use super::expr::{bind_scalar_expr, scalar_expr_type};

pub(crate) fn bind_returning_clause(
    clause: &mut ReturningClause,
    schema: &Schema,
    db: &Db,
    search_path: &[SchemaId],
    current_database: Option<&str>,
    time_ctx: BindTimeContext,
) -> PgWireResult<Schema> {
    let mut expanded = Vec::new();
    for item in clause.exprs.drain(..) {
        match item {
            ReturningExpr::Star => {
                for field in &schema.fields {
                    expanded.push(ReturningExpr::Expr {
                        expr: ScalarExpr::Column(ColumnRefName {
                            schema: None,
                            relation: None,
                            column: field.name.clone(),
                            location: None,
                        }),
                        alias: field.name.clone(),
                    });
                }
            }
            ReturningExpr::Expr { expr, alias } => {
                expanded.push(ReturningExpr::Expr { expr, alias });
            }
        }
    }
    clause.exprs = expanded;
    let mut fields = Vec::with_capacity(clause.exprs.len());
    for item in clause.exprs.iter_mut() {
        if let ReturningExpr::Expr { expr, alias } = item {
            let bound = bind_scalar_expr(
                expr,
                schema,
                None,
                db,
                search_path,
                current_database,
                time_ctx,
            )?;
            let dt = scalar_expr_type(&bound, schema).unwrap_or(DataType::Text);
            fields.push(Field {
                name: alias.clone(),
                data_type: dt,
                origin: None,
            });
            *expr = bound;
        }
    }
    Ok(Schema { fields })
}