如何等到回调被调用

Question

我有一个执行长时间计算的函数，一旦完成，就会调用一个作为参数传递的回调函数。 我在 Koa 路由器中查询这个函数，需要将长时间计算的结果返回给浏览器。函数来自库，我无法更改其接口（即，我可以更改回调的代码，但无法更改 someLongComputation 以返回承诺）

当前代码集ctx.body 立即返回，someLongComputation 立即返回。知道如何等到回调被调用，然后才使用回调的结果设置ctx.body。

router.post(`/abc`, async (ctx) => {
  try {
      someLongComputation(function(err, res) {
          if(err) {
              console.log(err);
          }
      }
      ctx.body = {
        status: 'success',
        data: {'res' : ""}, 
        errors: []
      };
  } catch (err) {
      console.log(err)
  }
})

Answer 1

你有两个选择。

1。从您的回调中发送响应

router.post(`/abc`, async (ctx) => {
  try {
    someLongComputation(function(err, res) {
      if (err) {
        console.log(err);
        // you might want to send error response here
      }
      // send response from callback
      ctx.body = {
        status: 'success',
        data: { res: '' },
        errors: []
      };
    });
  } catch (err) {
    console.log(err);
    // you might want to send error response here as well
  }
});

2。从您的库函数中创建一个承诺并使用async/await

const doLongComputation = () => {
  return new Promise((resolve, reject) => {
    try {
      someLongComputation(function(err, res) {
        if (err) {
          console.log(err);
          reject(err);
        }
        resolve(res);
      });
    } catch (err) {
      console.log(err);
      reject(err);
    }
  });
};

router.post(`/abc`, async (ctx) => {
  try {
    const res = await doLongComputation();
    ctx.body = {
      status: 'success',
      data: { res: '' },
      errors: []
    };
  } catch (err) {
    console.log(err);
    // send error response here
  }
});

Answer 2

如果您想使用 async/await 并假设 longcomputation 函数可以返回一个 Promise，您可以更改您的代码以等待该函数的返回。

router.post(`/abc`, async (ctx) => {
  try {
    const result = await someLongComputation();
    // do whatever you want with result

  } catch (err) {
    console.log(err)
  }
})

另一种选择是在函数的回调中使用变量

router.post(`/abc`, async (ctx) => {
  try {
    someLongComputation(function(err, res) {
      if(err) {
          console.log(err);
      }
      //moved setting the body to inside the callback for the computation
      ctx.body = {
        status: 'success',
        data: {'res' : ""}, 
        errors: []
      };
    }
  } catch (err) {
    console.log(err)
  }
})