时间戳

首页 > 解决方案 > 在执行下一个函数之前等待 writestream 完成

node.js - 在执行下一个函数之前等待 writestream 完成

问题描述

我有两个功能。第一个函数读取文件夹中的所有文件并将其数据写入新文件。第二个函数将该新文件（函数 1 的输出）作为输入并创建另一个文件。因此它必须等到函数 1 的写入流完成。

const fs = require('fs');
const path = require('path');

function f1(inputDir, outputFile) {
    let stream = fs.createWriteStream(outputFile, {flags:'a'}); // new data should be appended to outputFile piece by piece (hence flag a)
    let files = await fs.promises.readdir(inputDir);

    for(let file of files) {
        let pathOfCurrentFile = path.join(inputDir, file);
        let stat = fs.statSync(pathOfCurrentFile);
        if(stat.isFile()) {
            data = await fs.readFileSync(pathOfCurrentFile, 'utf8');

            // now the data is being modified for output
            let result = data + 'other stuff';
            stream.write(result);
        }
    }
    stream.end();
}

function f2(inputFile, outputFile) {
    let newData = doStuffWithMy(inputFile);
    let stream = fs.createWriteStream(outputFile);
    stream.write(newData);
    stream.end();
}

f1('myFiles', 'myNewFile.txt');
f2('myNewFile.txt', 'myNewestFile.txt');

这是发生的事情：

当我在执行 f2 之前使用超时时，它工作正常，但我不能使用超时，因为 inputDir 中可能有数千个输入文件，因此我需要一种动态执行它的方法。

我已经尝试过异步/等待、回调、承诺等，但这些东西对我来说似乎有点高级，我无法让它工作。

还有什么我可以尝试的吗？

标签： node.jsfssynchronous

解决方案

既然您询问了同步版本，这就是它的样子。这应该只在单个用户脚本或启动代码中使用，而不是在正在运行的服务器中。服务器应该只使用异步文件 I/O。

// synchronous version
function f1(inputDir, outputFile) {
    let outputHandle = fs.openSync(outputFile, "a");
    try {
        let files = fs.readdirSync(inputDir, {withFileTypes: true});
        for (let f of files) {
            if (f.isFile()) {
                let pathOfCurrentFile = path.join(inputDir, f.name);
                let data = fs.readFileSync(pathOfCurrentFile, 'utf8');
                fs.writeSync(outputHandle, data);
            }
        }
    } finally {
        fs.closeSync(outputHandle);
    }
}

function f2(inputFile, outputFile) {
    let newData = doStuffWithMy(inputFile);
    fs.writeFileSync(outputFile, newData);
}

f1('myFiles', 'myNewFile.txt');
f2('myNewFile.txt', 'myNewestFile.txt');

推荐阅读