feat: emit request and response event (#2020)

lib/application.js

@@ -5,6 +5,7 @@ const fs = require('fs');
 const graceful = require('graceful');
 const http = require('http');
 const cluster = require('cluster-client');
+const onFinished = require('on-finished');
 const { assign } = require('utility');
 const eggUtils = require('egg-core').utils;
 const EggApplication = require('./egg');
@@ -145,6 +146,12 @@ class Application extends EggApplication {
     return context;
   }
 
+  handleRequest(ctx, fnMiddleware) {
+    this.emit('request', ctx);
+    super.handleRequest(ctx, fnMiddleware);
+    onFinished(ctx.res, () => this.emit('response', ctx));
+  }
+
   /**
    * save routers to `run/router.json`
    * @private

package.json

@@ -45,6 +45,7 @@
     "koa-is-json": "^1.0.0",
     "koa-override": "^3.0.0",
     "mime-types": "^2.1.17",
+    "on-finished": "^2.3.0",
     "sendmessage": "^1.1.0",
     "urllib": "^2.25.1",
     "utility": "^1.13.1",

test/lib/application.test.js

@@ -5,6 +5,7 @@ const mm = require('egg-mock');
 const sleep = require('mz-modules/sleep');
 const fs = require('fs');
 const path = require('path');
+const pedding = require('pedding');
 const Application = require('../../lib/application');
 const utils = require('../utils');
 
@@ -216,5 +217,38 @@ describe('test/lib/application.test.js', () => {
         app.emit('cookieLimitExceed', { name: 'name', value: 'value', ctx });
       });
     });
+
+    describe('request and response event', () => {
+      it('should emit when request success', done => {
+        done = pedding(done, 3);
+        app.once('request', ctx => {
+          assert(ctx.path === '/class-controller');
+          done();
+        });
+        app.once('response', ctx => {
+          assert(ctx.status === 200);
+          done();
+        });
+        app.httpRequest()
+          .get('/class-controller')
+          .expect('this is bar!')
+          .expect(200, done);
+      });
+
+      it('should emit when request error', done => {
+        done = pedding(done, 3);
+        app.once('request', ctx => {
+          assert(ctx.path === '/obj-error');
+          done();
+        });
+        app.once('response', ctx => {
+          assert(ctx.status === 500);
+          done();
+        });
+        app.httpRequest()
+          .get('/obj-error')
+          .expect(500, done);
+      });
+    });
   });
 });