Nodejs回调加超时限制两种实现方法

Nodejs下的IO操作都是异步的，有时候异步请求返回太慢，不想无限等待回调怎么办呢？我们可以给回调函数加一个超时限制，到一定时间还没有回调就表示失败，继续后面的步骤。

方案1：用async的parallel实现

在这里我用到了async的parallel方法，parallel可以让若干个异步回调的方法并行执行，正常情况下，所有回调函数都完成后会汇集到parallel最终的回调函数里，但是有一种例外，如果其中一个步骤有错误抛出并直接调用parallel最终的回调函数，利用这个特点就可以实现我们要的功能。

我封装了一个函数asyncWrapper， 逻辑比较简单，直接看代码：

async-timer.js

const async = require('async');const asyncWrapper = (fn, interval, ...args) =>{ let final_callback = args[args.length-1]; async.parallel([ function(callback){ args[args.length - 1] = callback; fn.apply(this, args); }, function(callback){ setTimeout(function(){ callback(408); }, interval); } ], function(err, results){ if(err==408 && results[0])err = null; final_callback.apply(this,[err].concat([results[0]])); });}if(module.parent){ exports.asyncWrapper = asyncWrapper;}else{ let myfn = (arg_1, arg_2, callback) => { setTimeout(function(){ callback(null,'value 1: '+arg_1,'value 2: '+arg_2); }, 1000); } asyncWrapper(myfn, 2000, 10, 20, (err, values)=>{ console.log(`${err}, ${values}`); });}

你可以直接运行这个js看看效果：

node async-timer.js

模块调用方法：

const asyncWrapper = require('./async-timer.js').asyncWrapperconst fn = (arg1, arg2, callback) => { //...假设这里过程很漫长，有可能超时 callback(null, result_1, result_2);}asyncWrapper( fn,//异步函数 10000,//超时时间 'arg1_value',//异步函数的参数1 'arg2_value',//异步函数的参数2，有多个参数就在后面继续加 (err, results)=>{ //results : [result_1, result_2] //最后的回调,results比较特殊，fn有多个返回值时，results会以数组的形式返回给你 });

这种方案的好处是对node没有最低版本要求，引入async库就可以用，缺陷是最终返回的值都是在一个参数里面以数组的形式呈现。

方案2：用Promise实现

在promise里面加settimeout，超过时间就reject。

promise-timer.js

const promiseWrapper = (fn, interval, ...args) =>{ let final_callback = args[args.length-1]; new Promise((resolve, reject)=>{ args[args.length - 1] = (err, ...vals)=>{ if(err)reject(err); else resolve(vals); }; fn.apply(this, args); setTimeout(_=>{ reject('Promise time out'); }, interval); }) .then( result => { final_callback.apply(this, [null].concat(result)); } ) .catch(err=>{ final_callback(err); })}if(module.parent){ exports.promiseWrapper = promiseWrapper;}else{ let myfn = (arg_1, arg_2, callback) => { setTimeout(function(){ callback(null,'value 1: '+arg_1,'value 2: '+arg_2); }, 1000); } promiseWrapper(myfn, 1000, 10, 20, (err, value_1, value_2)=>{ console.log(`${err}, value 1: ${value_1} ... value 2: ${value_2}`); });}

模块调用方法：

const asyncWrapper = require('./promise-timer.js').promiseWrapperconst fn = (arg1, arg2, callback) => { //...假设这里过程很漫长，有可能超时 callback(null, result_1, result_2);}promiseWrapper( fn,//异步函数 10000,//超时时间 'arg1_value',//异步函数的参数1 'arg2_value',//异步函数的参数2，有多个参数就在后面继续加 (err, result_1, result_2)=>{ //最后的回调 });

这种方案要求nodejs版本高于5.0，支持Promise。

感谢阅读，希望能帮助到大家，谢谢大家对本站的支持！