javascript - 使用 bluebird promises 的 ms sql 资源管理

Question

我正在使用 mssql 模块连接到使用 Node 的 sql server 数据库。 Bluebird 具有类似于 c# 中的资源管理的功能。它有一个“使用”方法来避免必须使用 try/catch/finall 来处理资源。他们有 pg 和 mysql 的示例，但没有 mssql 的示例，mssql 的创建方式与 pg 和 mysql 不同。以下是如何使用它的示例:

using(getConnection(),
  fs.readFileAsync("file.sql", "utf8"), function(connection, fileContents) {
  return connection.query(fileContents);
}).then(function() {
  console.log("query successful and connection closed");
});

但是为了能够使用这个方法，你需要创建一个描述如何关闭连接的连接方法。这是 pg 的示例:

function getSqlConnection(connectionString) {
   var close;
   return pg.connectAsync(connectionString).spread(function(client, done) {
      close = done;
      return client;
   }).disposer(function(client) {
      if (close) close(client);
   });
}

我在使用 mssql 模块时遇到的问题是 connect 方法不返回像 pg 甚至 mysql 模块这样的连接对象。有没有人能够用 mssql 做到这一点？

更新 1:

这是我制作交易处理器的方式:

function getTransaction(connection) {
   return new Promise(function(resolve, reject) {
       var tx = sql.Transaction(connection);
       tx.beginAsync().then(function(err) {
           if(err) {
             tx = null;
             return reject(err);
           }
           return resolve(tx);
       });
   }).disposer(function(tx, promise) {
       if(promise.isFulfilled()) {
          return tx.commitAsync();
       }
       else {
          return tx.rollbackAsync();
       }
   });
}

这似乎是有效的，但不确定这是否有效。现在我需要弄清楚如何捕获查询中的错误。

这就是我进行交易的方式:

using(getConnection(), function(connection) {
   return using(getTransaction(connection), function(tx) {
          return query(queryString, tx).then(function() {
               console.log('first query in transaction completed.');
               console.log('starting second query in transaction.');
               return query(anotherQueryString, tx);
          });
   });
});

如果我将单个捕获标记为外部“使用”，是否会捕获整个事务中的所有错误？

Answer 1

好问题，mssql 有非常棘手的 API(构造函数接受回调!)所以这是对文档的很好补充。

var Promise = require("bluebird");
var sql = Promise.promisifyAll(require("mssql"));
global.using = Promise.using;

function getConnection(config) {
    var connection;
    return new Promise(function(resolve, reject) 
        connection = new sql.Connection(config, function(err) {
            if (err) {
                connection = null;
                return reject(err);
            }
            resolve(connection);
        });
    }).disposer(function() {
      if (connection) connection.close();
   });
}

var config = {
    user: '...',
    password: '...',
    server: 'localhost',
    database: '...',
};

using(getConnection(config), function(connection) {
    var request = new sql.Request(connection);
    return request.queryAsync("select 1 as number").then(function(recordSet) {
        console.log("got record set", recordSet);
        return request.queryAsync("select 10 as number");
    });
}).then(function(recordSet) {
    console.log("got record set", recordSet);
})

要使用事务，请尝试像这样实现 getTransaction:

function getTransaction(connection) {
  var tx = new sql.Transaction(connection);
  return tx.beginAsync().thenReturn(tx).disposer(function(tx, promise) {
    return promise.isFulfilled() ? tx.commitAsync() : tx.rollbackAsync();
  });
}

像这样使用它:

using(getConnection(), function(connection) {
  return using(getTransaction(connection), function(tx) {
    var request = new sql.Request(tx);
    return request.queryAsync("INSERT 1...").then(function() {
      return request.queryAsync("INSERT 2...");
    }).then(function() {
      return request.queryAsync("INSERT 3...");
    });
  });
});

错误处理:

using(getConnection(), function(connection) {
  return using(getTransaction(connection), function(tx) {
    var request = new sql.Request(tx);
    return request.queryAsync("INSERT...");
  });
}).catch(sql.TransactionError, function(e) {
  console.log("transaction failed", e);
}).catch(sql.ConnectionError, function(e) {
  console.log("connection failed", e);
}).catch(sql.RequestError, function(e) {
  console.log("invalid query", e);
});