RethinkDB: использование пула соединений

Не так давно в slack’е ресинковцев обсуждение было о том, какие ORM кто использует в node.js. Я лично не использую их. Вообще, для работы с RethinkDB только пул соединений использую.

Если у вас поболее, чем полтора юзера, то гонять данные по одному соединению вы не сможете. Поэтому, вы захотите использовать какой-то пул. Я лично этот использую. Сама инициализация примерно такая:

var rethink = require('rethinkdb');
var Pool = require('rethinkdb-pool');

pool = Pool(rethink, {
    host:rHost,
    port:rPort,
    db:'my_db',
    min: 5,
    max: 20
});
pool.r = rethink;

var rethink = require('rethinkdb');

var Pool = require('rethinkdb-pool');

pool = Pool(rethink, {

host:rHost,

port:rPort,

db:'my_db',

min: 5,

max: 20

});

pool.r = rethink;

Теперь в коде вы можете использовать pool.r для создание запросов. Например, простой запрос на получение записи:

var getDoc = pool.r.table("documents").get(1);
pool.run(query, function (error, ret) {
        if (error !== null) {
            // ошибочка
        }
});

var getDoc = pool.r.table("documents").get(1);

pool.run(query, function (error, ret) {

if (error !== null) {

// ошибочка

}

});

Вообще, код либы совсем небольшой:

'use strict'

var debug = require('debug')('rethinkdb:pool')
var Pool = require('generic-pool').Pool

function toArray (cursorOrResult) {
  if (cursorOrResult && typeof cursorOrResult.toArray === 'function') {
    return cursorOrResult.toArray()
  } else {
    return cursorOrResult
  }
}

module.exports = function (r, options) {
  function create (done) {
    return r.connect(options, done)
  }

  function destroy (connection) {
    connection.close()
  }

  var pool = new Pool({
    name: 'rethinkdb',
    create: create,
    destroy: destroy,
    log: options.log || debug,
    max: options.max || 10,
    min: options.min || 1,
    idleTimeoutMillis: options.idleTimeoutMillis || 30 * 1000
  })

  var Promise = r._bluebird

  function acquire () {
    return new Promise(function (resolve, reject) {
      pool.acquire(function (e, conn) {
        e ? reject(e) : resolve(conn)
      })
    }).disposer(function (conn) { pool.release(conn) })
  }

  pool.run = function (query, opt, done) {
    if (typeof opt === 'function') {
      done = opt
      opt = null
    }

    var p = Promise.using(acquire(), function (conn) {
      return query.run(conn, opt)
    }).then(toArray)

    if (done) {
      p.then(function (d) {
        done(null, d)
      }).catch(done)
    } else {
      return p
    }
  }

  return pool
}

'use strict'

var debug = require('debug')('rethinkdb:pool')

var Pool = require('generic-pool').Pool

function toArray (cursorOrResult) {

if (cursorOrResult && typeof cursorOrResult.toArray === 'function') {

return cursorOrResult.toArray()

} else {

return cursorOrResult

}

module.exports = function (r, options) {

function create (done) {

return r.connect(options, done)

}

function destroy (connection) {

connection.close()

}

var pool = new Pool({

name: 'rethinkdb',

create: create,

destroy: destroy,

log: options.log || debug,

max: options.max || 10,

min: options.min || 1,

idleTimeoutMillis: options.idleTimeoutMillis || 30 * 1000

})

var Promise = r._bluebird

function acquire () {

return new Promise(function (resolve, reject) {

pool.acquire(function (e, conn) {

e ? reject(e) : resolve(conn)

})

}).disposer(function (conn) { pool.release(conn) })

}

pool.run = function (query, opt, done) {

if (typeof opt === 'function') {

done = opt

opt = null

}

var p = Promise.using(acquire(), function (conn) {

return query.run(conn, opt)

}).then(toArray)

if (done) {

p.then(function (d) {

done(null, d)

}).catch(done)

} else {

return p

}

return pool

}

Gamedev suffering

Блог о разработке игр и серверных технологиях

RethinkDB: использование пула соединений