From b1943976453366ae43668e25766c95627bbf8419 Mon Sep 17 00:00:00 2001
From: luaneko <lumi@lua.re>
Date: Sat, 11 Jan 2025 00:46:17 +1100
Subject: [PATCH] Optimise query request write

---
 README.md |  4 ++--
 deno.json |  2 +-
 wire.ts   | 45 ++++++++++++++++++++++++++++++++-------------
 3 files changed, 35 insertions(+), 16 deletions(-)

diff --git a/README.md b/README.md
index 4cd33f5..af95aa5 100644
--- a/README.md
+++ b/README.md
@@ -14,9 +14,9 @@ The glue for TypeScript to PostgreSQL.
 ## Installation
 
 ```ts
-import pglue from "https://git.lua.re/luaneko/pglue/raw/tag/0.1.1/mod.ts";
+import pglue from "https://git.lua.re/luaneko/pglue/raw/tag/0.1.2/mod.ts";
 // ...or from github:
-import pglue from "https://raw.githubusercontent.com/luaneko/pglue/refs/tags/0.1.1/mod.ts";
+import pglue from "https://raw.githubusercontent.com/luaneko/pglue/refs/tags/0.1.2/mod.ts";
 ```
 
 ## Documentation
diff --git a/deno.json b/deno.json
index fb8ca00..be61147 100644
--- a/deno.json
+++ b/deno.json
@@ -1,5 +1,5 @@
 {
   "name": "@luaneko/pglue",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "exports": "./mod.ts"
 }
diff --git a/wire.ts b/wire.ts
index 762068b..9f3f4ff 100644
--- a/wire.ts
+++ b/wire.ts
@@ -579,6 +579,11 @@ export class Wire extends TypedEmitter<WireEvents> implements Disposable {
   }
 }
 
+const msg_PD = object({ P: Parse, D: Describe });
+const msg_BE = object({ B: Bind, E: Execute });
+const msg_BEc = object({ B: Bind, E: Execute, c: CopyDone });
+const msg_BEcC = object({ B: Bind, E: Execute, c: CopyDone, C: Close });
+
 function wire_impl(
   wire: Wire,
   socket: Deno.Conn,
@@ -904,10 +909,11 @@ function wire_impl(
       try {
         const { name, query } = this;
         return await pipeline(
-          async () => {
-            await write(Parse, { statement: name, query, param_types: [] });
-            await write(Describe, { which: "S", name });
-          },
+          () =>
+            write(msg_PD, {
+              P: { statement: name, query, param_types: [] },
+              D: { which: "S", name },
+            }),
           async () => {
             await read(ParseComplete);
             const param_desc = await read(ParameterDescription);
@@ -1078,16 +1084,23 @@ function wire_impl(
     try {
       const { rows, tag } = await pipeline(
         async () => {
-          await write(Bind, {
+          const B = {
             portal,
             statement: st.name,
             param_formats: [],
             param_values,
             column_formats: [],
-          });
-          await write(Execute, { portal, row_limit: 0 });
-          await write_copy_in(stdin);
-          await write(Close, { which: "P" as const, name: portal });
+          };
+          const E = { portal, row_limit: 0 };
+          const C = { which: "P" as const, name: portal };
+
+          if (stdin !== null) {
+            await write(msg_BE, { B, E });
+            await write_copy_in(stdin);
+            await write(Close, C);
+          } else {
+            return write(msg_BEcC, { B, E, c: {}, C });
+          }
         },
         async () => {
           await read(BindComplete);
@@ -1131,15 +1144,21 @@ function wire_impl(
     try {
       let { done, rows, tag } = await pipeline(
         async () => {
-          await write(Bind, {
+          const B = {
             portal,
             statement: st.name,
             param_formats: [],
             param_values,
             column_formats: [],
-          });
-          await write(Execute, { portal, row_limit: chunk_size });
-          await write_copy_in(stdin);
+          };
+          const E = { portal, row_limit: chunk_size };
+
+          if (stdin !== null) {
+            await write(msg_BE, { B, E });
+            await write_copy_in(stdin);
+          } else {
+            return write(msg_BEc, { B, E, c: {} });
+          }
         },
         async () => {
           await read(BindComplete);