uqa-sql 0.3.5

PostgreSQL-compatible SQL compiler built on libpg_query
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
64
65
66
67
68
69
70

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

//! Transaction commands that require an explicit SQL block.

use crate::SQLError;

pub fn transaction_requires_explicit_block(transaction: &crate::ast::TransactionStmt) -> bool {
    matches!(
        transaction,
        crate::ast::TransactionStmt::Savepoint(_)
            | crate::ast::TransactionStmt::ReleaseSavepoint(_)
            | crate::ast::TransactionStmt::RollbackToSavepoint(_)
            | crate::ast::TransactionStmt::CommitAndChain
            | crate::ast::TransactionStmt::RollbackAndChain
    )
}

pub fn no_active_transaction_error(transaction: &crate::ast::TransactionStmt) -> SQLError {
    let command = match transaction {
        crate::ast::TransactionStmt::Savepoint(_) => "SAVEPOINT",
        crate::ast::TransactionStmt::ReleaseSavepoint(_) => "RELEASE SAVEPOINT",
        crate::ast::TransactionStmt::RollbackToSavepoint(_) => "ROLLBACK TO SAVEPOINT",
        crate::ast::TransactionStmt::CommitAndChain => "COMMIT AND CHAIN",
        crate::ast::TransactionStmt::RollbackAndChain => "ROLLBACK AND CHAIN",
        _ => unreachable!("only explicit-block transaction commands use this error"),
    };
    SQLError::Routine {
        sqlstate: "25P01".into(),
        message: format!("{command} can only be used in transaction blocks"),
    }
}

#[cfg(test)]
mod tests {
    use super::{no_active_transaction_error, transaction_requires_explicit_block};
    use crate::{ast::TransactionStmt, SQLError};

    #[test]
    fn explicit_block_commands_keep_their_individual_diagnostics() {
        for (command, spelling) in [
            (TransactionStmt::Savepoint("s".into()), "SAVEPOINT"),
            (
                TransactionStmt::ReleaseSavepoint("s".into()),
                "RELEASE SAVEPOINT",
            ),
            (
                TransactionStmt::RollbackToSavepoint("s".into()),
                "ROLLBACK TO SAVEPOINT",
            ),
            (TransactionStmt::CommitAndChain, "COMMIT AND CHAIN"),
            (TransactionStmt::RollbackAndChain, "ROLLBACK AND CHAIN"),
        ] {
            assert!(transaction_requires_explicit_block(&command));
            assert!(
                matches!(no_active_transaction_error(&command), SQLError::Routine { sqlstate, message } if sqlstate == "25P01" && message == format!("{spelling} can only be used in transaction blocks"))
            );
        }
        for command in [
            TransactionStmt::Begin,
            TransactionStmt::Commit,
            TransactionStmt::Rollback,
        ] {
            assert!(!transaction_requires_explicit_block(&command));
        }
    }
}