You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.3 KiB
56 lines
1.3 KiB
3 years ago
|
'use strict'
|
||
|
|
|
||
|
|
var Client = require('./client')
|
||
|
|
var defaults = require('./defaults')
|
||
|
|
var Connection = require('./connection')
|
||
|
|
var Pool = require('pg-pool')
|
||
|
|
const { DatabaseError } = require('pg-protocol')
|
||
|
|
|
||
|
|
const poolFactory = (Client) => {
|
||
|
|
return class BoundPool extends Pool {
|
||
|
|
constructor(options) {
|
||
|
|
super(options, Client)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
var PG = function (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')
|
||
|
|
this.DatabaseError = DatabaseError
|
||
|
|
}
|
||
|
|
|
||
|
|
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
|
||
|
|
Object.defineProperty(module.exports, 'native', {
|
||
|
|
configurable: true,
|
||
|
|
enumerable: false,
|
||
|
|
get() {
|
||
|
|
var native = null
|
||
|
|
try {
|
||
|
|
native = new PG(require('./native'))
|
||
|
|
} catch (err) {
|
||
|
|
if (err.code !== 'MODULE_NOT_FOUND') {
|
||
|
|
throw err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// overwrite module.exports.native so that getter is never called again
|
||
|
|
Object.defineProperty(module.exports, 'native', {
|
||
|
|
value: native,
|
||
|
|
})
|
||
|
|
|
||
|
|
return native
|
||
|
|
},
|
||
|
|
})
|
||
|
|
}
|