手动宣传与Bluebird的pg.connect

lui*_*ati 6 node.js promise node-postgres bluebird

我想宣传node-postgres的pg.connect方法以及connection.query回调中提供的内部方法.

我可以.promisify后者,但我需要手动实现第一个(如果我在这里遗漏了一些东西,请解释).

问题是,我不确定这段代码是否正确或是否应该改进?代码正在运行,我只是想知道我是否正在使用Bluebird.

// aliases
var asPromise = Promise.promisify;

// save reference to original method
var connect = pg.connect.bind(pg);

// promisify method
pg.connect = function (data) {
  var deferred = Promise.defer();

  connect(data, function promisify(err, connection, release) {
    if (err) return deferred.reject(err);

    // promisify query factory
    connection.query = asPromise(connection.query, connection);

    // resolve promised connection
    deferred.resolve([connection,release]);
  });

  return deferred.promise;
};
Run Code Online (Sandbox Code Playgroud)

Esa*_*ija 4

扔掉所有可怕的回调代码,然后在应用程序初始化的某个地方执行此操作:

\n\n
var pg = require("pg");\nvar Promise = require("bluebird");\n\nObject.keys(pg).forEach(function(key) {\n    var Class = pg[key];\n    if (typeof Class === "function") {\n        Promise.promisifyAll(Class.prototype);\n        Promise.promisifyAll(Class);\n    }\n})\nPromise.promisifyAll(pg);\n
Run Code Online (Sandbox Code Playgroud)\n\n

稍后,您可以在任何地方使用 pg 模块,就好像它被设计为使用 Promise 一样:

\n\n
// Later\n// Don\'t even need to require bluebird here\nvar pg = require("pg");\n// Note how it\'s the pg API but with *Async suffix\npg.connectAsync(...).spread(function(connection, release) {\n     return connection.queryAsync("...")\n         .then(function(result) {\n            console.log("rows", result.rows);\n         })\n         .finally(function()\xc2\xa0{\n            // Creating a superfluous anonymous function cos I am\n            // unsure of your JS skill level\n            release();\n         });\n});\n
Run Code Online (Sandbox Code Playgroud)\n