libpiggy/lib/close-connection.js

Use a PostgreSQL database like a JSON document store.

"use strict";

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

var _promise = require("babel-runtime/core-js/promise");

var _promise2 = _interopRequireDefault(_promise);

var _keys = require("babel-runtime/core-js/object/keys");

var _keys2 = _interopRequireDefault(_keys);

var _asyncToGenerator2 = require("babel-runtime/helpers/asyncToGenerator");

var _asyncToGenerator3 = _interopRequireDefault(_asyncToGenerator2);

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

const closeConnection = (() => {
  var _ref = (0, _asyncToGenerator3.default)(function* (options, globals) {
    const { connectionName } = options || {};
    const { state } = globals;

    const closing = [];

    if (connectionName) {
      closing.push(state.openPools[connectionName].end());
    } else {
      for (const poolName of (0, _keys2.default)(state.openPools)) {
        closing.push(state.openPools[poolName].end());
      }
    }

    yield _promise2.default.all(closing);

    if (connectionName) {
      delete state.openPools[connectionName];
    } else {
      state.openPools = {};
    }

    return state;
  });

  return function closeConnection(_x, _x2) {
    return _ref.apply(this, arguments);
  };
})();

exports.default = closeConnection;