Node.js自动重试装饰器

开发node应用时，尤其是有爬虫类功能的，经常会需要发出HTTP请求。而由于网络波动等因素，请求有可能失败。
为了保证请求成功，通常会在请求失败时重试几次。我们可以封装一个装饰器函数，用它装饰请求函数，达到简单实现操作重试的目的。

装饰器函数：

autoRetry(fn, maxTryTimes, errHandler)
● fn Function 被装饰函数
● maxTryTimes Number 最大尝试次数
● errHandler Function 异常回调函数
 ○ err Error 失败产生的异常

errHandler是一个回调函数，唯一参数err，类似function (err) {}。在这里可以根据err决定接下来如何处理。如果抛出异常function (err) { throw err; }，则中止重试。如果返回truefunction (err) { return true; }，则无视尝试次数，继续重试。

function autoRetry(fn, maxTryTimes, errHandler) {
  return async function (...args) {
    let tryTimes = 0;
    async function inner() {
      try {
        tryTimes++;
        return await fn(...args);
      } catch (err) {
        if (errHandler && errHandler(err)) return inner();
        if (tryTimes === maxTryTimes) throw err; 
        return inner();
      }
    }
    return inner();
  };
}

使用实例：

// 目标：包装request()，在其抛出超时异常时重试。最多尝试3次。抛出其他异常则不重试。

// 先包装一下request，使其由回调形式变为返回Promise对象。
// 这里将返回状态码非200也视为异常
function request(opts) {
  console.log('我执行了一次');
  return new Promise((resolve, reject) => {
    require('request')(opts, (err, res) => {
      if (err) return reject(err);
      if (res.statusCode !== 200) return reject(res.statusCode);
      resolve(res.body);
    });
  });
}

// 使用装饰器
request = autoRetry(request, 3, (err) => {
  if (!err.code || (err.code !== 'ETIMEDOUT' && err.code !== 'ESOCKETTIMEDOUT')) {
    throw(err);
  }
});

// 测试不同的url
(async function () {
    try {
      await request({ url: '...' });
    } catch (err) {
      console.error('最终抛出错误', err);
    }  
})();

使用上述实例测试，会发现，除超时外的错误都会直接抛出异常；超时则会尝试3次后，在第3次抛出异常。