feat: add prepare statement

src/pod/babashka/sql.clj

@@ -71,18 +71,26 @@
 (defn execute!
   ([db-spec sql-params]
    (execute! db-spec sql-params nil))
-  ([db-spec sql-params opts]
+  ([db-spec-or-stmt sql-params opts]
    ;; (.println System/err (str sql-params))
-   (let [conn (->connectable db-spec)
-         res (jdbc/execute! conn sql-params opts)]
+   (let [conn (->connectable db-spec-or-stmt)
+         res (if (some? conn)
+               (jdbc/execute! conn sql-params opts)
+               ;; a statement is provide
+               (let [stmt db-spec-or-stmt]
+                 (jdbc/execute! stmt sql-params opts)))]
      res)))
 
 (defn execute-one!
   ([db-spec sql-params]
    (execute-one! db-spec sql-params nil))
-  ([db-spec sql-params opts]
-   (let [conn (->connectable db-spec)
-         res (jdbc/execute-one! conn sql-params opts)]
+  ([db-spec-or-stmt sql-params opts]
+   (let [conn (->connectable db-spec-or-stmt)
+         res (if (some? conn)
+               (jdbc/execute-one! conn sql-params opts)
+               ;; a statement is provide
+               (let [stmt db-spec-or-stmt]
+                 (jdbc/execute-one! stmt sql-params opts)))]
      res)))
 
 (defn insert-multi!
@@ -98,6 +106,12 @@
     (when-let [conn (get old connection)]
       (.close ^java.lang.AutoCloseable conn))))
 
+(defn prepare
+  ([conn-map statement] (prepare conn-map statement nil))
+  ([conn-map statement opts]
+   (let [conn (->connectable conn-map)]
+     (jdbc/prepare conn statement opts))))
+
 (def transact @#'t/transact*)
 
 (defn transaction-begin
@@ -153,6 +167,7 @@
            'execute-one! execute-one!
            'get-connection get-connection
            'close-connection close-connection
+           'prepare prepare
            'transaction/begin transaction-begin
            'transaction/rollback transaction-rollback
            'transaction/commit transaction-commit
@@ -215,6 +230,7 @@
                           {:name execute-one!}
                           {:name get-connection}
                           {:name close-connection}
+                          {:name prepare}
                           {:name with-transaction
                            :code ~with-transaction}]}
                   {:name ~(symbol (str sql-ns ".transaction"))

test/pod/babashka/hsqldb_test.clj

@@ -25,6 +25,11 @@
           (is (= [#:FOO{:FOO 1} #:FOO{:FOO 2} #:FOO{:FOO 3}]
                  (db/execute! conn  ["select * from foo;"])))
           (db/close-connection conn)))
+      (testing "prepared statements"
+        (let [conn (db/get-connection db)]
+          (with-open [ps (db/prepare conn ["select * from foo where foo = ?" 1])]
+            (let [result (db/execute-one! ps)]
+              (is (= result #:foo{:foo 1}))))))
       (testing "transaction"
         (let [conn (db/get-connection db)]
           (transaction/begin conn)

test/pod/babashka/mysql_test.clj

@@ -53,6 +53,11 @@
         (is (= [#:foo{:foo 1} #:foo{:foo 2} #:foo{:foo 3}]
               (db/execute! conn  ["select * from foo;"])))
         (db/close-connection conn)))
+    (testing "prepared statements"
+      (let [conn (db/get-connection db)]
+        (with-open [ps (db/prepare conn ["select * from foo where foo = ?" 1])]
+          (let [result (db/execute-one! ps)]
+            (is (= result #:foo{:foo 1}))))))
     (testing "input parameters"
       (try (db/execute! db ["drop table foo_timed;"])
            (catch Exception _ nil))

test/pod/babashka/postgresql_test.clj

@@ -47,6 +47,11 @@
           (db/execute! x ["insert into foo_timed values (?, ?)" 1 start-date])
           (let [result (db/execute! x ["select foo from foo_timed where created <= ?" start-date])]
             (is (= result [{:foo_timed/foo 1}]))))))
+    (testing "prepared statements"
+      (let [conn (db/get-connection db)]
+        (with-open [ps (db/prepare conn ["select * from foo where foo = ?" 1])]
+          (let [result (db/execute-one! ps nil)]
+            (is (= result #:foo{:foo 1}))))))
     (testing "transaction"
       (let [conn (db/get-connection db)]
         (transaction/begin conn)