60 lines
1.0 KiB
JavaScript
60 lines
1.0 KiB
JavaScript
|
|
var http = require('http');
|
|
var koa = require('..');
|
|
var app = koa();
|
|
|
|
// x-response-time
|
|
|
|
app.use(function(next){
|
|
return function *responseTime(){
|
|
var start = new Date;
|
|
yield next;
|
|
var ms = new Date - start;
|
|
this.set('X-Response-Time', ms + 'ms');
|
|
}
|
|
});
|
|
|
|
// logger
|
|
|
|
app.use(function(next){
|
|
return function *logger(){
|
|
var start = new Date;
|
|
yield next;
|
|
var ms = new Date - start;
|
|
console.log('%s %s - %s', this.method, this.url, ms);
|
|
}
|
|
});
|
|
|
|
// content-length
|
|
|
|
app.use(function(next){
|
|
return function *contentLength(){
|
|
yield next;
|
|
if (!this.body) return;
|
|
this.set('Content-Length', Buffer.byteLength(this.body));
|
|
}
|
|
});
|
|
|
|
// custom 404 handler
|
|
|
|
app.use(function(next){
|
|
return function *notfound(){
|
|
yield next;
|
|
if (this.body) return;
|
|
this.status = 404;
|
|
this.body = 'Sorry! No luck';
|
|
}
|
|
});
|
|
|
|
// response
|
|
|
|
app.use(function(next){
|
|
return function *response(){
|
|
yield next;
|
|
if ('/' != this.url) return;
|
|
this.body = 'Hello World';
|
|
}
|
|
});
|
|
|
|
app.listen(3000);
|