Busboy 在解析所有数据之前完成

问题描述

我有一个 express 应用程序并且有一个函数，它使用 busboy 来解析表单数据，该数据返回表单的字段值，但在返回调用之前不解析整个字段。


module.exports = async function (headers,invalidMime,res,req) {
    let fields = {};
    const fileWrites = []
    let filesToUpload = [];
    let finished = false;
    const busboy = new Busboy({
        headers: headers,limits: { ileSize: 10 * 1024 * 1024 }
    });

    await busboy.on("field",(fieldname,val) => {
        console.log(fieldname); // Log 1
        fields[fieldname] = val;
    });

    await busboy.on("file",file,filename,encoding,mimetype) => {
        
        if (invalidMime(mimetype)) return res.status(404).json({ [fieldname]: "Wrong file type submitted" });
        
        file.on("limit",() => { return res.status(404).json({ general: "File size too large!" }); });

        const randomizedFileName = createFileName(filename);
        const filePath = path.join(os.tmpdir(),randomizedFileName);

        fileWrites.push(createImagePromise(file,filePath,randomizedFileName,mimetype,fieldname));
    });

    busboy.on("finish",() => {
      console.log(fields); // Log 2
    });
      
    busboy.end(req.rawBody);

    return {filesToUpload: fileWrites,fields: fields}
}

这会返回我的字段，但缺少最后一个。当我用控制台日志调试它时，我可以看到 busboy.on("finish") 在我返回我的值后执行，这导致缺少变量。

const formData = await  FormParser(req.headers,req);

console.log(formData); // Log 3
const { fields,filesToUpload } = formData;

  var1                     // Log 1 - Before return
  var2                     // Log 1 - Before return
  var3                     // log 1 - Before return
  { filesToUpload: [],// Log 3 - After return
    fields:
     { var1: 'var1',var2: 'var2',var3: 'var3' } }          
  var4                     // Log 1 - After return
  { var1: 'var1',// Log 2 - After return
    var2: 'var2',var3: 'var3',var4: 'var4' }

如何让 busboy 在解析完成后返回值？

解决方法

当您在这里和那里散布几个 await 时，异步函数不会神奇地变成同步。

例如，这毫无意义：

await busboy.on("field",(fieldname,val) => {
    console.log(fieldname); // Log 1
    fields[fieldname] = val;
});

你认为它以某种方式等待 fields 对象被填充，但它真正做的是：它等待 .on() 函数返回，而这个函数立即返回。它唯一的工作是分配一个事件处理程序。 field 事件根本没有发生。

异步编程中的解决方案始终是：在表示任务已完成的事件处理程序中执行工作。您正在尝试在函数的最后一行中完成工作 (return {filesToUpload: fileWrites,fields: fields})，就好像最后一行将是最后运行的内容。事实并非如此。

一旦您将需要对事件做出反应的位移动到内部事件处理程序中，您会发现整个函数根本不需要async。

免责声明：以下代码未经测试，我之前没有使用过busboy，做它意味着，不一定是它所说的。

module.exports = function (headers,invalidMime,res,req) {
    let fields = {};
    let pendingFileWrites = [];

    const busboy = new Busboy({
        headers: headers,limits: { fileSize: 10 * 1024 * 1024 }
    });

    busboy.on("filesLimit",() => {
        res.status(400).json({ error: "File size too large!" });
    });
    busboy.on("error",() => {
        res.status(500).json({ error: "Error parsing data" });
    });
    busboy.on("field",val) => {
        fields[fieldname] = val;
    });
    busboy.on("finish",() => {
        Promise.all(pendingFileWrites).then((fileWrites) => {
            // NOW we're done
            res.json({
                filesToUpload: fileWrites,fields: fields
            });
        });
    });

    busboy.on("file",file,filename,encoding,mimetype) => {
        console.log(`Processing [{filename}] ({mimetype})`);
        if (invalidMime(mimetype)) return res.status(404).json({ [fieldname]: "Wrong file type submitted" });
        file.on("limit",() => { return res.status(404).json({ general: "File size too large!" }); });
        file.on("end",() => { console.log(`Done processing [{filename}] ({mimetype})`); });

        const randomizedFileName = createFileName(filename);
        const filePath = path.join(os.tmpdir(),randomizedFileName);

        pendingFileWrites.push(createImagePromise(file,filePath,randomizedFileName,mimetype,fieldname));
    });
};

请注意，您可以将其设为 async：

busboy.on("finish",() => {
    Promise.all(pendingFileWrites).then((fileWrites) => {
        // NOW we're done
        res.json({
            filesToUpload: fileWrites,fields: fields
        });
    });
});

喜欢

busboy.on("finish",async () => {
    var fileWrites = await Promise.all(pendingFileWrites);
    // NOW we're done
    res.json({
        filesToUpload: fileWrites,fields: fields
    });
});

甚至

busboy.on("finish",async () => {
    res.json({
        filesToUpload: await Promise.all(pendingFileWrites),fields: fields
    });
});

如果你愿意。无论哪种情况，您都需要添加错误处理（前者通过 .catch()，后者通过 try/catch 块）。

busboy express