mongoose-connection-ready/index.js

Promisified mongoose connection state.

module.exports = function connectionReady (conn) {
  var STATES = conn.constructor.STATES

  if (STATES.connecting === conn.readyState) {
    return new Promise(function (_accept, _reject) {
      conn.once('connected', accept)
      conn.once('error', reject)
      conn.once('disconnected', reject)

      function off () {
        conn.removeListener('connected', accept)
        conn.removeListener('error', reject)
        conn.removeListener('disconnected', reject)
      }

      function accept () {
        off()
        _accept()
      }

      function reject () {
        off()
        _reject()
      }
    })
  }

  if (STATES.connected === conn.readyState) {
    return Promise.resolve()
  }

  return Promise.reject()
}