深入了解Node.js util模块的promisify()方法

Node.js 内置的 util 模块有一个 promisify() 方法，该方法将基于回调的函数转换为基于 Promise 的函数。这使您可以将 Promise 链和 async/await 与基于回调的 API 结合使用。

例如，Node.js 的 fs 模块在读取文件时，需要使用回调：

const fs = require('fs')

fs.readFile('./package.json', function callback(err, buf) {
  const obj = JSON.parse(buf.toString('utf8'))
  console.log(obj.name) // 'Example' -> package.json 包名
})

我们可以使用 util.promisify() 将 fs.readFile() 的回调函数转换为返回 Promise 函数：

const fs = require('fs')
const util = require('util')

// 将 fs.readFile() 转换为一个接受相同参数但返回 Promise 的函数。
const readFile = util.promisify(fs.readFile)

// 现在可以将 readFile() 与 await 一起使用！
const buf = await readFile('./package.json')

const obj = JSON.parse(buf.toString('utf8'))
console.log(obj.name) // 'Example'

promisify 是如何工作的？

util.promisify() 在后台是如何工作的？npm 上有一个 polyfill，您可以在这里阅读完整的实现。您也可以在这里找到 Node.js 的实现，不过为了便于理解，polyfill 更易于阅读。【推荐学习：《nodejs 教程》】

util.promisify() 背后的关键思想是向传入的参数添加回调函数。该回调函数解析或拒绝 promisified 函数返回的 Promise。

为了便于理解，下面是一个非常简化的 util.promisify() 自定义实现示例：

const fs = require('fs')

// util.promisify() 的简化实现。不包括所有情况，不要在 prod 环境中使用此选项！
function promisify(fn) {
  return function() {
    const args = Array.prototype.slice.call(arguments)
    return new Promise((resolve, reject) => {
      fn.apply(this, [].concat(args).concat([(err, res) => {
        if (err != null) {
          return reject(err)
        }
        resolve(res)
      }]))
    })
  }
}

// 将 fs.readFile() 转换为一个接受相同参数但返回 Promise 的函数。
const readFile = promisify(fs.readFile)

// 现在可以将 readFile() 与 await 一起使用！
const buf = await readFile('./package.json')

const obj = JSON.parse(buf.toString('utf8'))
console.log(obj.name) // 'Example'

那么这是什么意思呢？首先，util.promisify()向传入的参数添加 1 个额外参数，然后使用这些新参数调用原始函数。这意味着底层函数需要支持该数量的参数。因此，如果您要调用myFn()具有 2 个类型参数的promisified 函数 [String, Object]，请确保原始函数支持[String, Object, Function]。

那么这意味着什么呢？首先，util.promisify() 向传入的参数添加一个额外参数，然后使用这些新参数调用原始函数。这意味着基础函数需要支持该数量的参数。因此，如果您使用 [String, Object] 类型的 2 个参数调用 promisified 函数 myFn()，请确保原始函数支持 [String, Object, Function]。

其次，util.promisify() 对函数上下文（this）有影响。

丢失上下文

丢失上下文（this）意味着函数调用以错误的值结束。丢失上下文是转换函数的常见问题：

class MyClass {
  myCallbackFn(cb) {
    cb(null, this)
  }
}

const obj = new MyClass()
const promisified = require('util').promisify(obj.myCallbackFn)

const context = await promisified()
console.log(context) // 打印 undefined 而不是 MyClass 实例！

请记住，this 包含函数被调用时的属性的任何对象。因此，您可以通过将 promisified 函数设置为同一对象的属性来保留上下文：

class MyClass {
  myCallbackFn(cb) {
    cb(null, this)
  }
}

const obj = new MyClass()
// 保留上下文，因为 promisified 是 obj 的属性
obj.promisified = require('util').promisify(obj.myCallbackFn)

const context = await obj.promisified()
console.log(context === obj) // true

更多编程相关知识，请访问：编程视频！！

以上就是深入了解Node.js util模块的promisify()方法的详细内容，更多请关注其它相关文章！