Как использовать mysql транзакции вместе с async/await?

Question

[vetsmen]

Доброго времени суток. Есть библиотека mysql для node.js.
Вот код для mysql транзакций:

spoiler

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

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

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

Как можно видеть, это cb hell. Как можно переписать данный код с использованием async/await?

Answer 1

[Владлен Хеллсайт]

Если я всё правильно понял, то так:

const { promisify } = require('util');

/* ... */

const query = promisify(connection.query).bind(connection);
const commit = promisify(connection.commit).bind(connection);
const rollback = promisify(connection.rollback).bind(connection);
const beginTransaction = promisify(connection.beginTransaction).bind(connection);

/* ... */

async function run(title) {
	try {
		await beginTransaction();

		const { insertId } = await query('INSERT INTO posts SET title=?', title);

		const log = `Post ${insertId} added`;

		await query('INSERT INTO log SET data=?', log);

		await commit();

		console.log('Success!');
	} catch (e) {
		await rollback();
	}
}

Comments

[RidgeA]

+ рекомендую использовать https://www.npmjs.com/package/mysql2 - там уже все промиссифицировано.

(транзакции на коллбеках - это боль еще та....)

[vetsmen]

RidgeA, чем плох выше указанный способ?

[RidgeA]

vetsmen, ничем. Я не говорил, что он плох.

[vetsmen]

А как можно переписать данный код для использования с пулом коннектов?
Текущий промис для query у меня выглядит так:

exports.query = (sql, props) => {
    return new Promise((resolve, reject) => {
        pool.getConnection((err, connection) => {
            if (err) throw err;
            connection.query(
                sql, props,
                (err, res) => {
                    if (err) reject(err);
                    else resolve(res);
                }
            );
            connection.release();
        });
    });
};

Не совсем представляю, как можно транзакции сюда вписать для того же пула.

[Владлен Хеллсайт]

vetsmen, я не работал с модулем mysql напрямю, но там же просто нужно получить соединение. А по хорошему лучше всё таки использовать модуль с пулом и Promise по умолчанию, а то так и продолжится :)