在异步函数内生成 NodeJS 工作线程不尊重承诺解析

问题描述

我有一个异步函数,可以像这样启动 NodeJS 工作线程:

    encode : async (config) => {

      if (isMainThread) {

            const encode_worker = new Worker(`./service-encode.js`,{ workerData: config });

            encode_worker.on('message',(transcode_data) => { 
              log.info("%o",transcode_data);
              return transcode_data;
            });

            encode_worker.on('error',(err) => { log.error(err)});
            encode_worker.on('exit',(code) => {
              if (code !== 0)
                throw new Error(`Encoding stopped with exit code [ ${code} ]`);
                console.log("* * * EXITED ENCODER WORKER * * *")

            });

      }

    },

serivce-encode.js 文件中,我有以下使用异步函数代码。请注意,我使用 postMessage 表示已完成。

var transcoder        = require('./transcoder');

const {Worker,isMainThread,parentPort,workerData} = require('worker_threads');

console.log("* * * STARTING ENCODE THREAD * * *\n");
console.log(workerData);

transcoder.run(workerData)
          .then((results) => {
              transcode_data = results;
              parentPort.postMessage(transcode_data);
          })
          .catch(err => { throw err });

然后,我使用以下示例代码,但上面的 'message' 事件中的代码会立即触发。也就是说,它似乎没有等到完成:

encode(conf).then((encode_data) => { console.log("Encode Data :",encode_data); 

encode 函数工作正常,但 console.log 语句在调用 encode() 函数时会立即执行 - encode_data 变量也是 undefined。既然encode中的return语句在message事件中,那异步函数的promise不应该在那个时候解析吗?

解决方法

因此,您的 async 函数内的代码没有任何内容支持 Promise。您不能只是在 async 函数中抛出随机的异步(但不是基于 Promise 的)代码并期望任何东西都能正常工作。 async 函数可以很好地与您 await 的基于 Promise 的异步函数配合使用。否则,它对那里的异步操作一无所知。这就是调用 encode() 会立即返回而不等待任何事情完成的原因。

此外,return transcode_data 在异步回调中。返回该回调只会返回调用该回调的系统代码,并被尽职尽责地忽略。您不会从 async 函数本身返回任何内容。您将返回到那个回调。

由于您的操作不是基于承诺的,为了解决这个问题,您必须通过将其包装在承诺中来使其基于承诺,然后在需要时使用适当的值手动解决或拒绝该承诺。你可以这样做:

encode: (config) => {

    if (isMainThread) {

        return new Promise((resolve,reject) => {
            const encode_worker = new Worker(`./service-encode.js`,{ workerData: config });

            encode_worker.on('message',(transcode_data) => {
                log.info("%o",transcode_data);
                resolve(transcode_data);
            });

            encode_worker.on('error',(err) => {
                log.error(err)
                reject(err);
            });

            encode_worker.on('exit',(code) => {
                if (code !== 0) {
                    reject(new Error(`Encoding stopped with exit code [ ${code} ]`));
                }
                console.log("* * * EXITED ENCODER WORKER * * *")
            });

        });


    } else {
        // should return a promise from all calling paths
        return Promise.reject(new Error("Can only call encode() from main thread"));
    }

},

仅供参考,此代码假定您在此处从 Promise 中寻找的“结果”是您从该工作人员处获得的第一个 transcode_datamessage