mirror of https://github.com/lukechilds/node.git
Ben Noordhuis
13 years ago
1 changed files with 109 additions and 0 deletions
@ -0,0 +1,109 @@ |
|||||
|
//
|
||||
|
// Usage:
|
||||
|
// node benchmark/http_simple_auto.js <args> <target>
|
||||
|
//
|
||||
|
// Where:
|
||||
|
// <args> Arguments to pass to `ab`.
|
||||
|
// <target> Target to benchmark, e.g. `bytes/1024` or `buffer/8192`.
|
||||
|
//
|
||||
|
|
||||
|
var path = require("path"); |
||||
|
var http = require("http"); |
||||
|
var spawn = require("child_process").spawn; |
||||
|
|
||||
|
var port = parseInt(process.env.PORT || 8000); |
||||
|
|
||||
|
var fixed = "" |
||||
|
for (var i = 0; i < 20*1024; i++) { |
||||
|
fixed += "C"; |
||||
|
} |
||||
|
|
||||
|
var stored = {}; |
||||
|
var storedBuffer = {}; |
||||
|
|
||||
|
var server = http.createServer(function (req, res) { |
||||
|
var commands = req.url.split("/"); |
||||
|
var command = commands[1]; |
||||
|
var body = ""; |
||||
|
var arg = commands[2]; |
||||
|
var n_chunks = parseInt(commands[3], 10); |
||||
|
var status = 200; |
||||
|
|
||||
|
if (command == "bytes") { |
||||
|
var n = parseInt(arg, 10) |
||||
|
if (n <= 0) |
||||
|
throw "bytes called with n <= 0" |
||||
|
if (stored[n] === undefined) { |
||||
|
stored[n] = ""; |
||||
|
for (var i = 0; i < n; i++) { |
||||
|
stored[n] += "C" |
||||
|
} |
||||
|
} |
||||
|
body = stored[n]; |
||||
|
|
||||
|
} else if (command == "buffer") { |
||||
|
var n = parseInt(arg, 10) |
||||
|
if (n <= 0) throw new Error("bytes called with n <= 0"); |
||||
|
if (storedBuffer[n] === undefined) { |
||||
|
storedBuffer[n] = new Buffer(n); |
||||
|
for (var i = 0; i < n; i++) { |
||||
|
storedBuffer[n][i] = "C".charCodeAt(0); |
||||
|
} |
||||
|
} |
||||
|
body = storedBuffer[n]; |
||||
|
|
||||
|
} else if (command == "quit") { |
||||
|
res.connection.server.close(); |
||||
|
body = "quitting"; |
||||
|
|
||||
|
} else if (command == "fixed") { |
||||
|
body = fixed; |
||||
|
|
||||
|
} else if (command == "echo") { |
||||
|
res.writeHead(200, { "Content-Type": "text/plain", |
||||
|
"Transfer-Encoding": "chunked" }); |
||||
|
req.pipe(res); |
||||
|
return; |
||||
|
|
||||
|
} else { |
||||
|
status = 404; |
||||
|
body = "not found\n"; |
||||
|
} |
||||
|
|
||||
|
// example: http://localhost:port/bytes/512/4
|
||||
|
// sends a 512 byte body in 4 chunks of 128 bytes
|
||||
|
if (n_chunks > 0) { |
||||
|
res.writeHead(status, { "Content-Type": "text/plain", |
||||
|
"Transfer-Encoding": "chunked" }); |
||||
|
// send body in chunks
|
||||
|
var len = body.length; |
||||
|
var step = ~~(len / n_chunks) || len; |
||||
|
|
||||
|
for (var i = 0; i < len; i += step) { |
||||
|
res.write(body.slice(i, i + step)); |
||||
|
} |
||||
|
|
||||
|
res.end(); |
||||
|
} else { |
||||
|
var content_length = body.length.toString(); |
||||
|
|
||||
|
res.writeHead(status, { "Content-Type": "text/plain", |
||||
|
"Content-Length": content_length }); |
||||
|
res.end(body); |
||||
|
} |
||||
|
|
||||
|
}); |
||||
|
|
||||
|
server.listen(port, function () { |
||||
|
var url = 'http://127.0.0.1:' + port + '/'; |
||||
|
|
||||
|
var n = process.argv.length - 1; |
||||
|
process.argv[n] = url + process.argv[n]; |
||||
|
|
||||
|
var cp = spawn('ab', process.argv.slice(2)); |
||||
|
cp.stdout.pipe(process.stdout); |
||||
|
cp.stderr.pipe(process.stderr); |
||||
|
cp.on('exit', function() { |
||||
|
server.close(); |
||||
|
}); |
||||
|
}); |
Loading…
Reference in new issue