callback promisify 化初体验：22 行 Node.js 源码实现 + 源码共读总结

2023-12-30 16:47:34

回调函数的 Promise 化：Node.js 源码解析

1. 回调函数 Promise 化简介

在 Node.js 中，回调函数是一种常见的异步编程模式。然而，使用回调函数会带来一些不便，例如代码的可读性较差、难以处理错误等。为了解决这些问题，Node.js 提供了一个名为 promisify 的工具，可以将回调函数转换为 Promise 函数。

2. 源码解析

promisify 函数位于 util 模块中，其源码如下：

function promisify(original) {
  if (typeof original !== 'function') {
    throw new TypeError('The "original" argument must be of type Function.');
  }

  if (typeof original.promise === 'function') {
    return original.promise();
  }

  if (!original.length) {
    return function promiseified() {
      const args = Array.prototype.slice.call(arguments);
      return new Promise((resolve, reject) => {
        original.call(this, ...args, (err, value) => {
          if (err) {
            reject(err);
          } else {
            resolve(value);
          }
        });
      });
    };
  } else {
    return function promiseified(...args) {
      return new Promise((resolve, reject) => {
        original.call(this, ...args, (err, value) => {
          if (err) {
            reject(err);
          } else {
            resolve.apply(this, arguments.slice(1));
          }
        });
      });
    };
  }
}

核心思路

promisify 函数的核心思路是：将回调函数的参数包装成一个 Promise 对象，并将回调函数的第一个参数（错误对象）绑定到 Promise 对象的 reject 方法，将回调函数的第二个参数（返回值）绑定到 Promise 对象的 resolve 方法。

3. 使用示例

下面是一个使用 promisify 将回调函数转换为 Promise 函数的示例：

const fs = require('fs');
const promisifiedReadfile = promisify(fs.readFile);

promisifiedReadfile('package.json').then((data) => {
  console.log(data.toString());
});