Add support for connection parameters

porsager · porsager · commit 972e8f9a35f6 · 2019-11-26T10:19:04.000+01:00
diff --git a/lib/connection.js b/lib/connection.js
@@ -8,8 +8,6 @@ import { errors } from './types.js'
 export default function Connection(options = {}) {
   const {
     transform,
-    database,
-    user,
     timeout,
     onnotify,
     onnotice,
@@ -152,7 +150,7 @@ export default function Connection(options = {}) {
   }
 
   function ready() {
-    socket.write(frontend.connect({ user, database }))
+    socket.write(frontend.connect(options))
   }
 
   function close() {
diff --git a/lib/frontend.js b/lib/frontend.js
@@ -74,12 +74,17 @@ export default {
 
 */
 
-function connect({ user, database }) {
+function connect({ user, database, connection }) {
   return bytes
     .inc(4)
     .i16(3)
     .z(2)
-    .str(['user', user, 'database', database, 'client_encoding', '\'utf-8\''].join(N))
+    .str(Object.entries({
+      user,
+      database,
+      client_encoding: '\'utf-8\'',
+      ...connection
+    }).filter(([k, v]) => v).map(p).map(([k, v]) => k + N + v).join(N))
     .z(2)
     .end(0)
 }
