Node JS async/await with multiple fs.writeFile using through

Question

我正在使用 through2 从 Gulp 流生成多个文件。我正在使用 NodeJS 10.6.0，所以以为我会充分利用 async/await，但还没有完全理解机制。目前，在写入所有文件之前触发 through2done()回调。

这是我所拥有的(简化) - 请注意，我不会在最后返回流，因为没有必要。

async function createDirectory(pathDir) {
return new Promise((resolve, reject) => {
mkdirp(pathDir, (err) => {
if (err) reject(err);
else resolve();
});
});
}
async function writeFile(outputFilePath, outputFileContent) {
return new Promise((resolve, reject) => {
fs.writeFile(outputFilePath, outputFileContent, (err) => {
if (err) reject(err);
else resolve();
});
});
}
async function doWriteFile(outputFolderPath, outputFilePath, outputContent) {
await createDirectory(outputFolderPath);
await writeFile(outputFilePath, outputContent, outputContent);
}
async function doGenerateVariant(data, variantArr) {
for (const variant of variantArr) {
/* Do a load of stuff */
const variantOutputFolderPath = blah;
const variantOutputFilePath = blah;
const variantOutputContent = blah;
await doWriteFile(variantOutputFolderPath, variantOutputFilePath, variantOutputContent);
}
}
const generateVariant = () => {
return through.obj((file, enc, done) => {
const data = JSON.parse(file.contents.toString());
*/ Do a load of stuff */
const { variant } = data;
const variantArr = Object.values(variant);
doGenerateVariant(data, variantArr);
return done();
});
};

这不起作用，因为done()在所有文件写入之前返回。我猜我错过了一两个回报，但我所做的一切似乎都没有奏效。

如果我done()传递到doGenerateVariant并在doWriteFile之后调用它，一切都按预期工作，但我知道这是不正确的。

Answer 1

您需要等待doGenerateVariant完成其工作才能调用done。请记住async函数始终返回一个承诺。所以你可以这样做

const generateVariant = () => {
return through.obj((file, enc, done) => {
const data = JSON.parse(file.contents.toString());
*/ Do a load of stuff */
const { variant } = data;
const variantArr = Object.values(variant);
doGenerateVariant(data, variantArr).then(() => done());
});
};

或使用async/await

const generateVariant = () => {
return through.obj(async (file, enc, done) => {
const data = JSON.parse(file.contents.toString());
*/ Do a load of stuff */
const { variant } = data;
const variantArr = Object.values(variant);
await doGenerateVariant(data, variantArr);
done();
});
};