开发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次抛出异常。