hacky work around socket hangups on http requests

This commit is contained in:
Ryan Dahl 2011-01-20 17:07:44 -08:00
parent d89454e5d4
commit f465949516
2 changed files with 41 additions and 49 deletions

View File

@ -965,15 +965,18 @@ Agent.prototype._establishNewConnection = function() {
socket.on('error', function(err) { socket.on('error', function(err) {
debug("AGENT SOCKET ERROR: " + err.message); debug("AGENT SOCKET ERROR: " + err.message);
var req;
if (socket._httpMessage) { if (socket._httpMessage) {
socket._httpMessage.emit('error', err); req = socket._httpMessage
} else if (self.queue.length) { } else if (self.queue.length) {
var req = self.queue.shift(); req = self.queue.shift();
if (req) req.emit('error', err);
} else { } else {
// No requests on queue? Where is the request // No requests on queue? Where is the request
assert(0); assert(0);
} }
req.emit('error', err);
req._hadError = true; // hacky
}); });
socket.ondata = function(d, start, end) { socket.ondata = function(d, start, end) {
@ -987,6 +990,8 @@ Agent.prototype._establishNewConnection = function() {
socket.onend = null; socket.onend = null;
var res = parser.incoming; var res = parser.incoming;
assert(socket._httpMessage);
socket._httpMessage.res = res;
// This is start + byteParsed + 1 due to the error of getting \n // This is start + byteParsed + 1 due to the error of getting \n
// in the upgradeHead from the closing lines of the headers // in the upgradeHead from the closing lines of the headers
@ -1019,6 +1024,20 @@ Agent.prototype._establishNewConnection = function() {
// When the socket closes remove it from the list of available sockets. // When the socket closes remove it from the list of available sockets.
socket.on('close', function() { socket.on('close', function() {
// This is really hacky: What if someone issues a request, the server
// accepts, but then terminates the connection. There is no parse error,
// there is no socket-level error. How does the user get informed?
// We check to see if the socket has a request, if so if it has a
// response (meaning that it emitted a 'response' event). If the socket
// has a request but no response and it never emitted an error event:
// THEN we need to trigger it manually.
// There must be a better way to do this.
if (socket._httpMessage &&
!socket._httpMessage.res &&
!socket._httpMessage._hadError) {
socket._httpMessage.emit('error', new Error('socket hang up'));
}
self._removeSocket(socket); self._removeSocket(socket);
// unref the parser for easy gc // unref the parser for easy gc
parsers.free(parser); parsers.free(parser);
@ -1254,49 +1273,26 @@ exports.cat = function(url, encoding_, headers_) {
var url = require('url').parse(url); var url = require('url').parse(url);
var hasHost = false; var options = {
if (Array.isArray(headers)) { method: 'GET',
for (var i = 0, l = headers.length; i < l; i++) { port: url.port || 80,
if (headers[i][0].toLowerCase() === 'host') { host: url.hostname,
hasHost = true; headers: headers,
break; path: (url.pathname || '/') + (url.search || '') + (url.hash || '')
} };
}
} else if (typeof headers === 'Object') {
var keys = Object.keys(headers);
for (var i = 0, l = keys.length; i < l; i++) {
var key = keys[i];
if (key.toLowerCase() == 'host') {
hasHost = true;
break;
}
}
}
if (!hasHost) headers['Host'] = url.hostname;
var content = ''; var content = '';
var client = exports.createClient(url.port || 80, url.hostname);
var req = client.request((url.pathname || '/') +
(url.search || '') +
(url.hash || ''),
headers);
if (url.protocol == 'https:') {
client.https = true;
}
var callbackSent = false; var callbackSent = false;
req.addListener('response', function(res) { var req = exports.request(options, function(res) {
if (res.statusCode < 200 || res.statusCode >= 300) { if (res.statusCode < 200 || res.statusCode >= 300) {
if (callback && !callbackSent) { if (callback && !callbackSent) {
callback(res.statusCode); callback(res.statusCode);
callbackSent = true; callbackSent = true;
} }
client.end();
return; return;
} }
res.setEncoding(encoding); res.setEncoding(encoding);
res.addListener('data', function(chunk) { content += chunk; }); res.addListener('data', function(chunk) { content += chunk; });
res.addListener('end', function() { res.addListener('end', function() {
@ -1306,19 +1302,13 @@ exports.cat = function(url, encoding_, headers_) {
} }
}); });
}); });
req.end();
client.addListener('error', function(err) {
req.on('error', function(err) {
if (callback && !callbackSent) { if (callback && !callbackSent) {
callback(err); callback(err);
callbackSent = true; callbackSent = true;
} }
}); });
client.addListener('close', function() {
if (callback && !callbackSent) {
callback(new Error('Connection closed unexpectedly'));
callbackSent = true;
}
});
req.end();
}; };

View File

@ -7,6 +7,7 @@ var server = http.createServer(function(req, res) {
req.connection.setTimeout(500); req.connection.setTimeout(500);
req.connection.addListener('timeout', function() { req.connection.addListener('timeout', function() {
req.connection.destroy();
common.debug('TIMEOUT'); common.debug('TIMEOUT');
server.close(); server.close();
}); });
@ -19,9 +20,10 @@ server.listen(common.PORT, function() {
throw new Error('Timeout was not sucessful'); throw new Error('Timeout was not sucessful');
}, 2000); }, 2000);
http.cat('http://localhost:' + common.PORT + '/', 'utf8', var url = 'http://localhost:' + common.PORT + '/';
function(err, content) {
clearTimeout(errorTimer); http.cat(url, 'utf8', function(err, content) {
console.log('HTTP REQUEST COMPLETE (this is good)'); clearTimeout(errorTimer);
}); console.log('HTTP REQUEST COMPLETE (this is good)');
});
}); });