在Nodejs中Promisify事件处理程序和tiemout

时间:2016-11-01 00:52:27

标签: node.js event-handling promise

我有一个回调处理程序来处理websocket事件,我正在尝试编写一个返回promise的函数。如果在未执行处理程序的情况下调用处理程序或拒绝处理程序,则应解析promise。有更好的方法吗?

var callbackCalled = false;
var message = null;

waitForMessage = function() {
  return new Promise(function(resolve, reject) {
    setTimeout(function() {
      if (callbackCalled) {
        callbackCalled = false;
        message = null;
        resolve();
      } else {
        reject();
      }
    }, 5000);
  });
}

onEvent = function(event) {
  console.log("Process the event");
  //set the message
  callbackCalled = true;
};

1 个答案:

答案 0 :(得分:2)

你可以使用这样的东西。这将拒绝是在传入事件之前发生的超时。如果事件发生在超时之前,它将解析数据。

function waitForEventWithTimeout(socket, event, t) {
    return new Promise(function(resolve, reject) {
        var timer;

        function listener(data) {
            clearTimeout(timer);
            socket.removeListener(event, listener);
            resolve(data);
        }

        socket.on(event, listener);
        timer = setTimeout(function() {
            socket.removeListener(event, listener);
            reject(new Error("timeout"));
        }, t);
    });
}

您可以这样使用它:

waitForEventWithTimeout(socket, "myEvent", 1000).then(function(data) {
    // got data here
}, function() {
    // timeout here
});

注意,我在这里添加和删除事件监听器以确保它没有处理太多次或添加太多次,因为事件处理程序是多次侦听器,但承诺是一次性的。