Improve write (chunk if needed)

porsager · porsager · commit 1021a36ef980 · 2019-12-27T12:37:04.000+01:00
diff --git a/lib/connection.js b/lib/connection.js
@@ -181,6 +181,8 @@ function postgresSocket(options, {
 }) {
   let socket
   let closed = true
+  let next = null
+  let buffer
 
   function onclose() {
     socket.removeListener('data', data)
@@ -227,7 +229,12 @@ function postgresSocket(options, {
 
   const x = {
     ready: false,
-    write: x => socket.write(x),
+    write: x => {
+      buffer = buffer ? Buffer.concat([buffer, x]) : Buffer.from(x)
+      if (buffer.length >= 1024)
+        return write()
+      next === null && (next = setImmediate(write))
+    },
     destroy: () => {
       socket && socket.destroy()
       return Promise.resolve()
@@ -238,6 +245,12 @@ function postgresSocket(options, {
     connect
   }
 
+  function write() {
+    socket.write(buffer)
+    next !== null && clearImmediate(next)
+    buffer = next = null
+  }
+
   /* c8 ignore next */
   return x
 }
