35 lines
822 B
JavaScript
35 lines
822 B
JavaScript
'use strict';
|
|
|
|
module.exports = function (req, time) {
|
|
if (req.timeoutTimer) { return req; }
|
|
|
|
var host = req._headers ? (' to ' + req._headers.host) : '';
|
|
|
|
req.timeoutTimer = setTimeout(function timeoutHandler() {
|
|
req.abort();
|
|
var e = new Error('Connection timed out on request' + host);
|
|
e.code = 'ETIMEDOUT';
|
|
req.emit('error', e);
|
|
}, time);
|
|
|
|
// Set additional timeout on socket - in case if remote
|
|
// server freeze after sending headers
|
|
req.setTimeout(time, function socketTimeoutHandler() {
|
|
req.abort();
|
|
var e = new Error('Socket timed out on request' + host);
|
|
e.code = 'ESOCKETTIMEDOUT';
|
|
req.emit('error', e);
|
|
});
|
|
|
|
function clear() {
|
|
if (req.timeoutTimer) {
|
|
clearTimeout(req.timeoutTimer);
|
|
req.timeoutTimer = null;
|
|
}
|
|
}
|
|
|
|
return req
|
|
.on('response', clear)
|
|
.on('error', clear);
|
|
};
|