add savepoint and release stmt categories

Author: MarinPostma

sqld/src/query_analysis.rs

@@ -28,6 +28,8 @@ pub enum StmtKind {
     TxnEnd,
     Read,
     Write,
+    Savepoint,
+    Release,
     Other,
 }
 
@@ -50,7 +52,7 @@ impl StmtKind {
             Cmd::Explain(_) => Some(Self::Other),
             Cmd::ExplainQueryPlan(_) => Some(Self::Other),
             Cmd::Stmt(Stmt::Begin { .. }) => Some(Self::TxnBegin),
-            Cmd::Stmt(Stmt::Commit { .. } | Stmt::Rollback { .. }) => Some(Self::TxnEnd),
+            Cmd::Stmt(Stmt::Commit { .. } | Stmt::Rollback { savepoint_name: None, .. }) => Some(Self::TxnEnd),
             Cmd::Stmt(
                 Stmt::CreateVirtualTable { tbl_name, .. }
                 | Stmt::CreateTable {
@@ -84,14 +86,15 @@ impl StmtKind {
             Cmd::Stmt(Stmt::AlterTable(tbl_name, _)) => write_if_not_reserved(tbl_name),
             Cmd::Stmt(
                 Stmt::DropIndex { .. }
-                | Stmt::Savepoint(_)
                 | Stmt::CreateTrigger {
                     temporary: false, ..
                 }
                 | Stmt::CreateIndex { .. },
             ) => Some(Self::Write),
             Cmd::Stmt(Stmt::Select { .. }) => Some(Self::Read),
             Cmd::Stmt(Stmt::Pragma(name, body)) => Self::pragma_kind(name, body.as_ref()),
+            Cmd::Stmt(Stmt::Savepoint(_)) => Some(Self::Savepoint),
+            Cmd::Stmt(Stmt::Release(_)) | Cmd::Stmt(Stmt::Rollback { savepoint_name: Some(_) , ..}) => Some(Self::Release),
             _ => None,
         }
     }
@@ -181,7 +184,7 @@ impl State {
             (State::Txn, StmtKind::TxnEnd) => State::Init,
             (state, StmtKind::Other | StmtKind::Write | StmtKind::Read) => state,
             (State::Invalid, _) => State::Invalid,
-            (State::Init, StmtKind::TxnBegin) => State::Txn,
+            (State::Init, StmtKind::TxnBegin | ) => State::Txn,
         };
     }