Question

我编写了一个脚本，以递归方式异步修改目录中的js文件。它由一个search(dirname)函数组成，该函数在目录中搜索js文件，还有modify(filename)进行修改。

let totalFilesSearched = 0;

const search = (dir) => {
    fs.readdir(dir, (err, list) => {
        if (err) return;
        list.forEach((filename) => {
            const filepath = path.join(dir, filename);
            if (filename.endsWith('.js')) {
                modify(filepath);
            } else if (fs.lstatSync(filepath).isDirectory()) {
                search(filepath);
            }
        })
    });
}

const modify = (filename) => {
    fs.readFile(filename, 'utf8', (err, data) => {
        if (err) console.log(err);
        // ... my modification code ...
        totalFilesSearched++;
    });
}

search(args[0])
console.log(`Total files searched: ${totalFilesSearched}`);

我想打印出脚本末尾的totalFilesSearched，但因为我的代码是异步的，所以它只是立即打印Total files searched: 0。

有谁知道我要等到脚本即将结束打印出来之后怎么样？我遇到了麻烦，因为我的search()和modify()函数都是异步的。

Answer 1

改为使用Promise，然后在解决所有问题后调用console.log。使用promisify将回调转换为承诺：

const { promisify } = require('util');
const readFile = promisify(fs.readFile);
const readDir = util.promisify(fs.readdir);

const search = (dir) => (
  readDir(dir).then((list) => (
    Promise.all(list.map((filename) => {
      const filepath = path.join(dir, filename);
      if (filename.endsWith('.js')) {
        return modify(filepath); // recursively return the promise
      } else if (fs.lstatSync(filepath).isDirectory()) {
        return search(filepath); // recursively return the promise
      }
    }))
  ))
  .catch(err => void 0)
);

const modify = (filename) => (
  readFile(filename, 'utf8')
    .then((data) => {
      // other code
      totalFilesSearched++;
    }).catch(err => console.log(err))
)

search(args[0])
  .then(() => {
    console.log(`Total files searched: ${totalFilesSearched}`);
  });

Answer 2

自我回答：

只需使用process.on('exit', callback_function_to_execute_at_end)

即可

它内置于节点中，您的回调将在进程退出之前立即执行。

在节点脚本结束时运行行？

2 个答案: