send close request when HTTP is executed only once

tursodatabase · Nov 20, 2023 · 2ae48cb · 2ae48cb
1 parent 9e0421b
commit 2ae48cb
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/http.rs b/src/http.rs
@@ -170,11 +170,16 @@ impl Client {
         } else {
             Cookie::default()
         };
+        let mut requests = vec![pipeline::StreamRequest::Execute(
+            pipeline::StreamExecuteReq { stmt },
+        )];
+        if tx_id == 0 {
+            // automatically close one-shot requests
+            requests.push(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