Merge pull request #59 from Horusiath/close-oneshot-requests

Send close request when HTTP is executed only once
tursodatabase · Nov 20, 2023 · 4e715e1 · 4e715e1
2 parents 9e0421b + 2ea86e8
commit 4e715e1
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/http.rs b/src/http.rs
@@ -170,11 +170,19 @@ impl Client {
         } else {
             Cookie::default()
         };
+        let requests = if tx_id != 0 {
+            vec![pipeline::StreamRequest::Execute(
+                pipeline::StreamExecuteReq { stmt },
+            )]
+        } else {
+            vec![
+                pipeline::StreamRequest::Execute(pipeline::StreamExecuteReq { stmt }),
+                pipeline::StreamRequest::Close,
+            ]
+        };
         let msg = pipeline::ClientMsg {
             baton: cookie.baton,
-            requests: vec![pipeline::StreamRequest::Execute(
-                pipeline::StreamExecuteReq { stmt },
-            )],
+            requests,
         };
         let body = serde_json::to_string(&msg)?;
         let url = cookie