h4ckx0r/engine.io-client-java
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improve test server

Browse Source
This commit is contained in:
Naoyuki Kanezawa
2014-08-17 05:34:58 +09:00
parent 21fb11f0c2
commit 9e24f9d922
1 changed files with 21 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
src/test/resources/server.js
View File

@@ -32,28 +32,27 @@ server.on('connection', function(socket) {
console.error(err); console.error(err);
}); });
var handleRequest = server.handleRequest;
server.handleRequest = function(req, res) { function before(context, name, fn) {
var method = context[name];
context[name] = function() {
fn.apply(this, arguments);
return method.apply(this, arguments);
};
}
before(server, 'handleRequest', function(req, res) {
// echo a header value // echo a header value
var value = req.headers['x-engineio']; var value = req.headers['x-engineio'];
if (value) { if (!value) return;
res.setHeader('X-EngineIO', value); res.setHeader('X-EngineIO', value);
} });
handleRequest.call(this, req, res); before(server, 'handleUpgrade', function(req, socket, head) {
}; // echo a header value for websocket handshake
var value = req.headers['x-engineio'];
var headerValue; if (!value) return;
var handleUpgrade = server.handleUpgrade; this.ws.once('headers', function(headers) {
server.handleUpgrade = function(req, socket, head) { headers.push('X-EngineIO: ' + value);
// echo a header value for websocket handshake });
headerValue = req.headers['x-engineio'];
handleUpgrade.call(this, req, socket, head);
};
// FIXME: support parallel requests
server.ws.on('headers', function(headers) {
if (headerValue) {
headers.push('X-EngineIO: ' + headerValue);
}
}); });