mockgres 0.0.22

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

use crate::catalog::SchemaName;
use crate::engine::{ObjName, Plan, fe};
use pg_query::protobuf::DeleteStmt;
use pgwire::error::PgWireResult;

use super::expr::parse_bool_expr;
use super::returning::parse_returning_clause;

pub fn plan_delete(mut del: DeleteStmt) -> PgWireResult<Plan> {
    let with_clause = del.with_clause.take();
    let rv = del.relation.ok_or_else(|| fe("missing target table"))?;
    let schema = if rv.schemaname.is_empty() {
        None
    } else {
        Some(SchemaName::new(rv.schemaname))
    };
    let table = ObjName {
        schema,
        name: rv.relname,
    };
    let table_alias = rv.alias.and_then(|a| {
        if a.aliasname.is_empty() {
            None
        } else {
            Some(a.aliasname)
        }
    });
    let filter = if let Some(w) = del.where_clause.as_ref().and_then(|n| n.node.as_ref()) {
        Some(parse_bool_expr(w)?)
    } else {
        None
    };
    let returning = parse_returning_clause(&del.returning_list)?;
    super::cte::wrap_with_clause(
        with_clause,
        Plan::Delete {
            table,
            table_alias,
            filter,
            returning,
            returning_schema: None,
        },
    )
}