使用蓝鸟承诺的 ms sql 资源管理

Question

我正在使用 mssql 模块通过节点连接到 sql server 数据库。 Bluebird 有一个类似于 c# 中的资源管理的功能。它有一个“使用”方法来避免必须使用 try/catch/finall 来处理资源。他们有 pg 和 mysql 的示例，但他们没有 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);
});