bluebird节点-mysql2事务

Question

我一直在尝试终止node-mysql2事务，但是不能让它工作，有什么建议吗？

我看过http://bluebirdjs.com/docs/api/disposer.html的文档

我也有非事务相关代码的工作版本，但让事务工作，以及适当的资源处理和错误处理一直很困难。下面是取自node-mysql api文档的基于回调的实现

connection.beginTransaction(function(err) {
  if (err) { throw err; }
  connection.query('INSERT INTO posts SET title=?', title, function(err, result) {
    if (err) {
      return connection.rollback(function() {
        throw err;
      });
    }

    var log = 'Post ' + result.insertId + ' added';

    connection.query('INSERT INTO log SET data=?', log, function(err, result) {
      if (err) {
        return connection.rollback(function() {
          throw err;
        });
      }  
      connection.commit(function(err) {
        if (err) {
          return connection.rollback(function() {
            throw err;
          });
        }
        console.log('success!');
      });
    });
  });
});

下面是我的查询管理实现

import { mysqldb } from '../config';
import { createPool } from 'mysql2';
import Pool from 'mysql2/lib/pool';
import Connection from 'mysql2/lib/connection';
import { using, promisifyAll, try as promiseTry } from 'bluebird';

promisifyAll([Pool, Connection]);
const pool = createPool(mysqldb);

export const getConnection = () =>
  pool.getConnectionAsync().disposer(connection =>
      connection.release());

export const withTransaction = (fn) =>
  using(getConnection(), tx =>
    promiseTry(fn(tx), tx.beginTransaction()).then(
      res => tx.commitAsync().catch(e => e).thenReturn(res),
      err => tx.rollbackAsync().catch(e => e).thenThrow(err)
    )
);

// withTransaction(tx =>
//   tx.executeAsync('sql1')
//     .then(tx.executeAsync('sql2'))
//     .then(tx.executeAsync('sql3'))
//     .then(tx.executeAsync('sql4'))
// );

export const query = (sql, params) =>
  using(getConnection(), connection =>
    connection.executeAsync(sql, params)
  );

然而，事务为我提供了

Unhandled rejection TypeError: expecting a function but got [object Object]
    at apiRejection (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:10:27)
    at Promise.attempt.Promise.try (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/method.js:26:16)
    at /Users/willh/workspace/ChenPin/graphql/lib/services/mysql.js:35:30
    at tryCatcher (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/util.js:16:23)
    at /Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/using.js:184:26
    at tryCatcher (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/util.js:16:23)
    at Promise._settlePromiseFromHandler (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:503:31)
    at Promise._settlePromise (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:560:18)
    at Promise._settlePromise0 (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:605:10)
    at Promise._settlePromises (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:684:18)
    at Promise._fulfill (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:629:18)
    at PromiseArray._resolve (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise_array.js:125:19)
    at PromiseArray._promiseFulfilled (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise_array.js:143:14)
    at Promise._settlePromise (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:565:26)
    at Promise._settlePromise0 (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:605:10)
    at Promise._settlePromises (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:684:18)
    at Promise._fulfill (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:629:18)
    at Promise._resolveCallback (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:424:57)
    at Promise._settlePromiseFromHandler (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:515:17)
    at Promise._settlePromise (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:560:18)
    at Promise._settlePromise0 (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:605:10)
    at Promise._settlePromises (/Users/willh/workspace/ChenPin/graphql/node_modules/bluebird/js/release/promise.js:684:18)

正确的方式是什么？

更新：

通过查看源代码让它工作，结果发现它只是一个简单的帮助器，真的不值得使用它们的帮助器，下面是最终的工作版本。无论如何，感谢你的帮助:D

export const withTransaction = (fn) =>
  using(getConnection(), tx =>
    tx.queryAsync('START TRANSACTION').then(fn(tx)).then(
      res => tx.queryAsync('COMMIT').thenReturn(res),
      err => tx.queryAsync('ROLLBACK').catch(e => e).thenThrow(err)
    )
  );

// withTransaction(tx =>
//   tx.executeAsync('select :x + :y as z', { x: 1, y: 2 }).then(res1 =>
//   tx.executeAsync('select :x + :y as z', { x: res1[0].z, y: 1 })).then(res2 =>
//   tx.executeAsync('select :x + :y as z', { x: res2[0].z, y: 1 })).then(res3 =>
//   tx.executeAsync('select :x + :y as z', { x: res3[0].z, y: 1 })).then(res4 => res4)
// );

顺便说一句，原来的promisify方法是有效的，没有区别，但谢谢你帮我解决了这个问题！

Answer 1

出现这个错误是因为promisifyAll使用的是函数，而不是对象。在大多数情况下，函数位于类的prototype中。尝尝这个

var Promise = require('bluebird');
var mysql = require('mysql2');
Promise.promisifyAll(mysql.Connection.prototype);
Promise.promisifyAll(require('mysql2/lib/pool').prototype)

在我的实践中，上面的例子工作得很好。

如果您想要在这方面做更多的实验，您也可以手动遍历所有元素并仅选择函数，就像我个人对Postgres使用的以下示例一样

var Promise = require('bluebird');
var pg = require('pg');
Object.keys(pg).forEach(function (key) {
  var Cls = null;
  try {
    Cls = pg[key];
    if (typeof Cls === 'function') {
      Promise.promisifyAll(Cls.prototype);
      Promise.promisifyAll(Cls);
    }
  } catch (e) {
    console.log(e);
  }
});
Promise.promisifyAll(pg);