index.js

'use strict'
/**
 * Copyright (c) 2010-2017 Brian Carlson (brian.m.carlson@gmail.com)
 * All rights reserved.
 *
 * This source code is licensed under the MIT license found in the
 * README.md file in the root directory of this source tree.
 */

const Client = require('./client')
const defaults = require('./defaults')
const Connection = require('./connection')
const Pool = require('pg-pool')

const poolFactory = (Client) => {
  return class BoundPool extends Pool {
    constructor(options) {
      const config = Object.assign({ Client: Client }, options)
      super(config)
    }
  }
}

class PG {
  constructor(clientConstructor) {
    this.defaults = defaults
    this.Client = clientConstructor
    this.Query = this.Client.Query
    this.Pool = poolFactory(this.Client)
    this._pools = []
    this.Connection = Connection
    this.types = require('pg-types')
  }
}

if (typeof process.env.NODE_PG_FORCE_NATIVE !== 'undefined') {
  module.exports = new PG(require('./native'))
} else {
  module.exports = new PG(Client)

  // lazy require native module...the native module may not have installed
  module.exports.__defineGetter__('native', function () {
    delete module.exports.native
    let native = null
    try {
      native = new PG(require('./native'))
    } catch (err) {
      if (err.code !== 'MODULE_NOT_FOUND') {
        throw err
      }
      /* eslint-disable no-console */
      console.error(err.message)
      /* eslint-enable no-console */
    }
    module.exports.native = native
    return native
  })
}