express和koa中的超时处理

来源:互联网 发布:java数组删除指定元素 编辑:程序博客网 时间:2024/06/18 14:21

从某个话题中得到的灵感,这里做一个分享,
http://cnodejs.org/topic/592fdc2f03dba3510d8a62a0#59364c3b538dbcaa6bc7dd48
楼主遇到一个问题,如何管理express中的超时?

express中的超时处理

现成的模块有connect-time,有一百五十个star,在中间件中已经算不少了,下面是一段使用connect-time的代码

var express = require('express')var timeout = require('connect-timeout')var app = express()app.use(timeout('5s'))//....一些中间件app.use(haltOnTimedout)// Add your routes here, etc.function haltOnTimedout (req, res, next) {if (!req.timedout) next()}app.listen(3000)

看起来很美好,然而实际上是有缺点的。

假设timeout的作用范围里中包含了一个名为query的中间件,长这个样子

function query(req, res, next) {setTimeout(function() {    res.send('respond with a resource');}, 10000);};

然后这样调用timeout

app.use(timeout('5s'))app.use(query)app.use(haltOnTimedout)

当然实际中可能是一个数据库操作或者别的,而且在某些情况下会因为阻塞让执行时间变得很慢,上面的代码中会在10秒后将数据返回前端,而这时timeout已经返回了一个超时相应,于是程序就gg了,报出一个Can’t set headers after they are sent的错误。


这个其实不能怪中间件,要怪express本身的机制,query中间件虽然加载了,然而里面的异步却不归timeout管。

这个问题有办法解决吗?一个思路是当timeout触发后,中断后面的异步操作,这个可以通过事件监听实现,然而很麻烦。

Koa中的timeout

所以说还是老老实实地用koa,能用同步写法就别写回调。
koa也有一些管理timeout的中间件例如koa-timeout,然而两三年没更新了,用的还是koa1.x的版本,好在也不是很难写,我就自己写了一个

var koa2-timeout = async function(ctx ,next){var tmr = null;const timeout = 5000;//设置超时时间await Promise.race([new Promise(function(resolve, reject) {tmr = setTimeout(function() {var e = new Error('Request timeout');e.status = 408;reject(e);}, timeout);}),new Promise(function(resolve, reject) {//使用一个闭包来执行下面的中间件(async function() {await next();clearTimeout(tmr);resolve();})();})])}

核心思路是用promise.race来看定时器和后面全部的中间件哪个先跑完,
* 如果定时器先跑完,返回错误信息,而且没有调用后续的next()
* 如果后面的中间件先跑完….那就是正常执行没毛病

下面是个调用的例子

var koa = require("koa");var app = new koa();app.use(koa2-timeout);app.use(async (ctx,next)=>{await myTimeout(1000);//这里如果改成6000就会返回超时信息ctx.body = "end";})function myTimeout(ms) {return new Promise((resolve, reject) => {setTimeout(resolve, ms, 'done');});}app.listen(3000);

其他的解决方案

仔细想想,还是有别的方案的,例如我们可以直接hack底层的res.end方法,只要在里面增加一个是否已经响应过的flag判断就好了。这里不再讲述

总结

Koa真是太好用了
比起express来koa的优点数不胜数

原创粉丝点击