forbid savepoints in legacy http

Author: MarinPostma

sqld/src/http/mod.rs

@@ -100,6 +100,12 @@ fn parse_queries(queries: Vec<QueryObject>) -> anyhow::Result<Vec<Query>> {
         out.push(query);
     }
 
+    // The state of a batch is unpredictable with savepoints, it's easier to just disable them in
+    // legacy API
+    if out.iter().any(|q| q.stmt.kind.is_savepoint() || q.stmt.kind.is_release()) {
+        anyhow::bail!("savepoints not supported in legacy HTTP API, use hrana endpoint instead");
+    }
+
     match predict_final_state(State::Init, out.iter().map(|q| &q.stmt)) {
         State::Txn => anyhow::bail!("interactive transaction not allowed in HTTP queries"),
         State::Init => (),

sqld/src/query_analysis.rs

@@ -164,6 +164,22 @@ impl StmtKind {
             },
         }
     }
+
+    /// Returns `true` if the stmt kind is [`Savepoint`].
+    ///
+    /// [`Savepoint`]: StmtKind::Savepoint
+    #[must_use]
+    pub fn is_savepoint(&self) -> bool {
+        matches!(self, Self::Savepoint)
+    }
+
+    /// Returns `true` if the stmt kind is [`Release`].
+    ///
+    /// [`Release`]: StmtKind::Release
+    #[must_use]
+    pub fn is_release(&self) -> bool {
+        matches!(self, Self::Release)
+    }
 }
 
 /// The state of a transaction for a series of statement
@@ -184,7 +200,8 @@ 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,
+            _ => State::Invalid,
         };
     }
 
@@ -273,7 +290,7 @@ impl Statement {
     pub fn is_read_only(&self) -> bool {
         matches!(
             self.kind,
-            StmtKind::Read | StmtKind::TxnEnd | StmtKind::TxnBegin
+            StmtKind::Read | StmtKind::TxnEnd | StmtKind::TxnBegin | StmtKind::Release | StmtKind::Savepoint
         )
     }
 }