add missing TransactionStmtKind::Start check for transactions

sbdchd · Oct 1, 2024 · fe3a965 · fe3a965
1 parent 0019f41
commit fe3a965
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/linter/src/rules/ban_concurrent_index_creation_in_transaction.rs b/linter/src/rules/ban_concurrent_index_creation_in_transaction.rs
@@ -14,7 +14,7 @@ pub fn ban_concurrent_index_creation_in_transaction(
     for raw_stmt in tree {
         match &raw_stmt.stmt {
             Stmt::TransactionStmt(stmt) => {
-                if stmt.kind == TransactionStmtKind::Begin && !in_transaction {
+                if (stmt.kind == TransactionStmtKind::Begin || stmt.kind == TransactionStmtKind::Start) && !in_transaction {
                     in_transaction = true;
                 }
                 if stmt.kind == TransactionStmtKind::Commit {

diff --git a/linter/src/rules/constraint_missing_not_valid.rs b/linter/src/rules/constraint_missing_not_valid.rs
@@ -16,7 +16,7 @@ fn not_valid_validate_in_transaction(tree: &[RawStmt], assume_in_transaction: bo
     for raw_stmt in tree {
         match &raw_stmt.stmt {
             Stmt::TransactionStmt(stmt) => {
-                if stmt.kind == TransactionStmtKind::Begin && !in_transaction {
+                if (stmt.kind == TransactionStmtKind::Begin || stmt.kind == TransactionStmtKind::Start) && !in_transaction {
                     in_transaction = true;
                     not_valid_names.clear();
                 }

diff --git a/linter/src/rules/prefer_robust_stmts.rs b/linter/src/rules/prefer_robust_stmts.rs
@@ -37,7 +37,7 @@ pub fn prefer_robust_stmts(
     for raw_stmt in tree {
         match &raw_stmt.stmt {
             Stmt::TransactionStmt(stmt) => match stmt.kind {
-                TransactionStmtKind::Begin => inside_transaction = true,
+                TransactionStmtKind::Begin | TransactionStmtKind::Start => inside_transaction = true,
                 TransactionStmtKind::Commit => inside_transaction = false,
                 _ => continue,
             },

diff --git a/linter/src/rules/utils.rs b/linter/src/rules/utils.rs
@@ -13,7 +13,7 @@ pub fn tables_created_in_transaction(
     for raw_stmt in tree {
         match &raw_stmt.stmt {
             Stmt::TransactionStmt(stmt) => match stmt.kind {
-                TransactionStmtKind::Begin => inside_transaction = true,
+                TransactionStmtKind::Begin | TransactionStmtKind::Start => inside_transaction = true,
                 TransactionStmtKind::Commit => inside_transaction = false,
                 _ => continue,
             },