NodeJS中Promise链式调用实现事务化任务删除方案咨询

嘿，这个需求很典型，咱们把 Promise 链式调用和数据库事务结合起来就能完美解决！你的初始思路方向是对的，但需要把事务的提交/回滚逻辑和 Promise 链的成功/失败回调绑定起来，避免回调嵌套混乱。下面给你一套完整的实现方案：

核心思路是把所有回调式的数据库操作封装成 Promise，然后将事务的开启、提交、回滚和业务操作的 Promise 链串联起来，确保任一环节失败就触发回滚。

1. 先封装事务相关的 Promise 方法 #

把数据库自带的回调式事务方法转成 Promise，方便后续链式调用：

// 封装事务开启
function beginTransaction(connection) {
  return new Promise((resolve, reject) => {
    connection.beginTransaction(err => {
      if (err) reject(err);
      else resolve(connection);
    });
  });
}

// 封装事务提交
function commitTransaction(connection) {
  return new Promise((resolve, reject) => {
    connection.commit(err => {
      if (err) reject(err);
      else resolve();
    });
  });
}

// 封装事务回滚
function rollbackTransaction(connection) {
  return new Promise(resolve => {
    // 回滚操作一般不抛错，确保流程能收尾
    connection.rollback(() => resolve());
  });
}

2. 定义各业务步骤的 Promise 函数 #

每个业务操作都返回 Promise，确保能链式传递数据：

// 根据ID获取任务（没找到任务直接抛出错误）
function getTaskById(connection, taskId) {
  return new Promise((resolve, reject) => {
    const query = 'SELECT * FROM tasks WHERE id = ?';
    connection.query(query, [taskId], (err, results) => {
      if (err) reject(err);
      else if (results.length === 0) reject(new Error('目标任务不存在'));
      else resolve(results[0]);
    });
  });
}

// 获取任务关联的文档（把任务和文档一起传递给下一个步骤）
function getTaskDocuments(connection, task) {
  return new Promise((resolve, reject) => {
    const query = 'SELECT * FROM documents WHERE task_id = ?';
    connection.query(query, [task.id], (err, results) => {
      if (err) reject(err);
      else resolve({ task, documents: results });
    });
  });
}

// 删除关联的文档（无文档时直接跳过）
function deleteTaskDocuments(connection, data) {
  const { documents } = data;
  if (documents.length === 0) return Promise.resolve(data);

  return new Promise((resolve, reject) => {
    const docIds = documents.map(doc => doc.id);
    const query = 'DELETE FROM documents WHERE id IN (?)';
    connection.query(query, [docIds], (err) => {
      if (err) reject(err);
      else resolve(data);
    });
  });
}

// 删除任务本身
function deleteTask(connection, data) {
  const { task } = data;
  return new Promise((resolve, reject) => {
    const query = 'DELETE FROM tasks WHERE id = ?';
    connection.query(query, [task.id], (err) => {
      if (err) reject(err);
      else resolve();
    });
  });
}

3. 串联完整流程，绑定事务逻辑 #

把所有步骤串起来，统一处理成功提交和失败回滚：

// 主执行函数
function deleteTaskWithDocuments(connection, taskId) {
  let txConnection; // 保存事务连接，确保所有操作在同一个事务里
  return beginTransaction(connection)
    .then(conn => {
      txConnection = conn;
      return getTaskById(txConnection, taskId);
    })
    .then(task => getTaskDocuments(txConnection, task))
    .then(data => deleteTaskDocuments(txConnection, data))
    .then(data => deleteTask(txConnection, data))
    .then(() => commitTransaction(txConnection)) // 所有步骤成功，提交事务
    .catch(async (err) => {
      console.error('操作失败，触发回滚:', err);
      if (txConnection) await rollbackTransaction(txConnection);
      throw err; // 重新抛出错误，让上层调用方处理
    });
}

// 使用示例
deleteTaskWithDocuments(yourDatabaseConnection, 1)
  .then(() => console.log('任务及关联文档删除成功'))
  .catch(err => console.error('删除失败:', err.message));

关键注意点 #

• 原子性保障：所有操作都复用同一个事务连接，确保要么全部成功，要么全部回滚
• 错误传递：每个步骤的错误都会被catch捕获，触发回滚后再抛出，不隐藏错误
• 边界处理：考虑了任务不存在、无关联文档的情况，避免无效操作导致的异常

内容的提问来源于stack exchange，提问作者David