pg-async/lib/index.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _bluebird = require('bluebird');

var _bluebird2 = _interopRequireDefault(_bluebird);

var _pg = require('pg');

var _pg2 = _interopRequireDefault(_pg);

var _debug = require('./debug');

var _debug2 = _interopRequireDefault(_debug);

var _makeAsyncApi = require('./makeAsyncApi');

var _makeAsyncApi2 = _interopRequireDefault(_makeAsyncApi);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function _asyncToGenerator(fn) { return function () { var gen = fn.apply(this, arguments); return new _bluebird2.default(function (resolve, reject) { function step(key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return _bluebird2.default.resolve(value).then(function (value) { return step("next", value); }, function (err) { return step("throw", err); }); } } return step("next"); }); }; }

function checkAsyncFunction(asyncFunc) {
  if (typeof asyncFunc !== 'function') throw new TypeError('async function expected');
}

class PgAsync {
  constructor(connectionOptions, driver) {
    this.closeConnections = () => this.getDriver().end();

    this.setConnectionOptions(connectionOptions);
    this.setDriver(driver);

    const wrap = name => {
      var _this = this;

      this[name] = function (sql) {
        for (var _len = arguments.length, values = Array(_len > 1 ? _len - 1 : 0), _key = 1; _key < _len; _key++) {
          values[_key - 1] = arguments[_key];
        }

        return _this.connect(client => client[`${ name }Args`](sql, values));
      };
    };
    const wrapArgs = name => {
      this[name] = (sql, values) => this.connect(client => client[name](sql, values));
    };

    wrap('query');
    wrapArgs('queryArgs');

    wrap('rows');
    wrapArgs('rowsArgs');

    wrap('row');
    wrapArgs('rowArgs');

    wrap('value');
    wrapArgs('valueArgs');
  }

  setConnectionOptions(options) {
    this._connectionOptions = options;
    return this;
  }

  getConnectionOptions() {
    return this._connectionOptions || this.getDriver().defaults;
  }

  getDriver() {
    return this._driver;
  }

  setDriver(driver) {
    if (typeof driver === 'string') switch (driver) {
      case '':case 'pg':
        driver = _pg2.default;
        break;
      case 'native':case 'pg.native':
        driver = _pg2.default.native;
        break;
      default:
        throw new Error(`Unrecognized driver name: ${ driver }`);
    }
    this._driver = driver || _pg2.default;
    return this;
  }

  getClient() {
    var _this2 = this;

    return _asyncToGenerator(function* () {
      return new _bluebird2.default(function (resolve, reject) {
        _this2.getDriver().connect(_this2.getConnectionOptions(), function (err, client, done) {
          if (err) {
            (0, _debug2.default)('%s getClient(%j)', err, _this2.getConnectionOptions());
            if (done) done(err);
            return reject(err);
          }
          return resolve({ client, done });
        });
      });
    })();
  }

  connect(asyncFunc) {
    var _this3 = this;

    return _asyncToGenerator(function* () {
      checkAsyncFunction(asyncFunc);

      var _ref = yield _this3.getClient();

      const client = _ref.client;
      const done = _ref.done;

      try {
        const api = (0, _makeAsyncApi2.default)(client);
        const result = yield asyncFunc(api);
        yield api.end();
        done();
        return result;
      } catch (err) {
        done(err);
        throw err;
      }
    })();
  }

  transaction(asyncFunc) {
    var _this4 = this;

    return _asyncToGenerator(function* () {
      checkAsyncFunction(asyncFunc);

      return yield _this4.connect((() => {
        var ref = _asyncToGenerator(function* (client) {
          yield client.startTransaction();
          try {
            const result = yield asyncFunc(client);
            yield client.commit();
            return result;
          } catch (err) {
            try {
              yield client.rollback();
            } catch (_) {
              // client disconnected?
            }
            throw err;
          }
        }),
            _this = this;

        return function (_x) {
          return ref.apply(_this, arguments);
        };
      })());
    })();
  }

}
exports.default = PgAsync;
module.exports = exports['default'];