pgwire/lib/index.js

PostgreSQL client library for Node.js which exposes all features of wire protocol

const Pool = require('./pool.js');
const Connection = require('./connection.js');
const { pgliteral, pgident } = require('./escape.js');

module.exports = {
  connect,
  connectRetry,
  pool,
  pgliteral,
  pgident,
};

async function connectRetry(url, options) {
  // TODO respect connect_timeout
  for (;;) {
    try {
      return await connect(url, options);
    } catch (err) {
      if (!(
        err.code == 'ENOTFOUND' ||
        err.code == 'ECONNREFUSED' ||
        err.code == 'ECONNRESET' ||
        err.code == 'PGERR_57P03' // cannot_connect_now
      )) {
        throw err;
      }
      await new Promise(resolve => setTimeout(resolve, 1e3));
    }
  }
}

function connect(...optionsArray) {
  return new Promise((resolve, reject) => {
    const conn = new Connection(mergeConnectOptions(optionsArray));
    conn._connectIfNotConnected();
    conn.on('close', reject);
    conn.on('ready', _ => {
      conn.off('close', reject);
      resolve(conn);
    });
  });
}

function pool(...optionsArray) {
  return new Pool(mergeConnectOptions(optionsArray));
}

function mergeConnectOptions(optionsArray) {
  return (
    optionsArray
    .map(optionsOrUrl => {
      if (typeof optionsOrUrl == 'string') {
        return url2conf(optionsOrUrl);
      }
      return optionsOrUrl;
    })
    .reduceRight((acc, it) => Object.assign(acc, it), {})
  );
}

function url2conf(urlstr) {
  const url = new URL(urlstr);
  return {
    hostname: url.hostname,
    port: url.port,
    user: url.username,
    password: url.password,
    database: url.pathname.replace(/^\//, '') || 'postgres',
    ...Array.from(url.searchParams).reduce(
      (obj, [key, val]) => (obj[key] = val, obj),
      {},
    ),
  };
}