better buffer utilization in pg

Author: splitice

packages/pg-protocol/src/buffer-reader.ts

@@ -1,4 +1,4 @@
-const emptyBuffer = Buffer.allocUnsafe(0)
+const emptyBuffer = Buffer.allocUnsafeSlow(0)
 
 export class BufferReader {
   private buffer: Buffer = emptyBuffer

packages/pg-protocol/src/buffer-writer.ts

@@ -5,7 +5,7 @@ export class Writer {
   private offset: number = 5
   private headerPosition: number = 0
   constructor(private size = 256) {
-    this.buffer = Buffer.allocUnsafe(size)
+    this.buffer = Buffer.allocUnsafeSlow(size)
   }
 
   private ensure(size: number): void {
@@ -15,7 +15,7 @@ export class Writer {
       // exponential growth factor of around ~ 1.5
       // https://stackoverflow.com/questions/2269063/buffer-growth-strategy
       var newSize = oldBuffer.length + (oldBuffer.length >> 1) + size
-      this.buffer = Buffer.allocUnsafe(newSize)
+      this.buffer = Buffer.allocUnsafeSlow(newSize)
       oldBuffer.copy(this.buffer)
     }
   }
@@ -79,7 +79,7 @@ export class Writer {
     var result = this.join(code)
     this.offset = 5
     this.headerPosition = 0
-    this.buffer = Buffer.allocUnsafe(this.size)
+    this.buffer = Buffer.allocUnsafeSlow(this.size)
     return result
   }
 }

packages/pg-protocol/src/parser.ts

@@ -134,7 +134,7 @@ export class Parser {
           while (newLength >= newBufferLength) {
             newBufferLength *= 2
           }
-          newBuffer = Buffer.allocUnsafe(newBufferLength)
+          newBuffer = Buffer.allocUnsafeSlow(newBufferLength)
         }
         // Move the remaining buffer to the new one
         this.buffer.copy(newBuffer, 0, this.bufferOffset, this.bufferOffset + this.bufferLength)

packages/pg-protocol/src/serializer.ts

@@ -36,7 +36,7 @@ const startup = (opts: Record<string, string>): Buffer => {
 }
 
 const requestSsl = (): Buffer => {
-  const response = Buffer.allocUnsafe(8)
+  const response = Buffer.allocUnsafeSlow(8)
   response.writeInt32BE(8, 0)
   response.writeInt32BE(80877103, 4)
   return response
@@ -183,7 +183,7 @@ const execute = (config?: ExecOpts): Buffer => {
   const portalLength = Buffer.byteLength(portal)
   const len = 4 + portalLength + 1 + 4
   // one extra bit for code
-  const buff = Buffer.allocUnsafe(1 + len)
+  const buff = Buffer.allocUnsafeSlow(1 + len)
   buff[0] = code.execute
   buff.writeInt32BE(len, 1)
   buff.write(portal, 5, 'utf-8')
@@ -193,7 +193,7 @@ const execute = (config?: ExecOpts): Buffer => {
 }
 
 const cancel = (processID: number, secretKey: number): Buffer => {
-  const buffer = Buffer.allocUnsafe(16)
+  const buffer = Buffer.allocUnsafeSlow(16)
   buffer.writeInt32BE(16, 0)
   buffer.writeInt16BE(1234, 4)
   buffer.writeInt16BE(5678, 6)
@@ -211,7 +211,7 @@ const cstringMessage = (code: code, string: string): Buffer => {
   const stringLen = Buffer.byteLength(string)
   const len = 4 + stringLen + 1
   // one extra bit for code
-  const buffer = Buffer.allocUnsafe(1 + len)
+  const buffer = Buffer.allocUnsafeSlow(1 + len)
   buffer[0] = code
   buffer.writeInt32BE(len, 1)
   buffer.write(string, 5, 'utf-8')

packages/pg/lib/connection.js

@@ -40,7 +40,7 @@ class Connection extends EventEmitter {
 
     this._connecting = true
     this.stream.setNoDelay(true)
-    this.stream.connect(port, host)
+    this.stream.connect({port, host})
 
     this.stream.once('connect', function () {
       if (self._keepAlive) {

packages/pg/lib/stream.js

@@ -20,7 +20,8 @@ module.exports = {
 function getNodejsStreamFuncs() {
   function getStream(ssl) {
     const net = require('net')
-    return new net.Socket()
+    const socket = new net.Socket({onread:{buffer: Buffer.allocUnsafeSlow(2048), callback: (b, n)=>socket.emit('data', b.slice(0, n))}})
+    return socket
   }
 
   function getSecureStream(options) {