Fix: Ensure we capture all chunks if multiple when using node-http-proxy

Author: dgilling

lib/index.js

@@ -91,34 +91,32 @@ module.exports = function (options) {
     };
 
     // Manage to get information from the response too, just like Connect.logger does:
-    var write = res.write;
+    res._mo_write = res.write;
+    var resBodyBuf;
 
     res.write = function (chunk, encoding, callback) {
-      // console.log('res write is invoked');
-
-      res.time = new Date();
-      res.responseTime = (new Date) - req._startTime;
-
-      res.write = write;
-      res.write(chunk, encoding, callback);
-
-
-      logEvent(chunk, req, res, options, moesifController);
-
+      if (chunk) {
+        resBodyBuf = resBodyBuf ? Buffer.concat([resBodyBuf, Buffer.from(chunk)]) : Buffer.from(chunk);
+      }
+      res._mo_write(chunk, encoding, callback);
     };
 
     // Manage to get information from the response too, just like Connect.logger does:
-    var end = res.end;
+    res._mo_end = res.end;
 
     res.end = function (chunk, encoding, callback) {
-      // console.log('res end is invoked');
+
       res.time = new Date();
       res.responseTime = (new Date) - req._startTime;
 
-      res.end = end;
+      if (chunk) {
+        resBodyBuf = resBodyBuf ? Buffer.concat([resBodyBuf, Buffer.from(chunk)]) : Buffer.from(chunk);
+      }
+
+      res.end = res._mo_end;
       res.end(chunk, encoding, callback);
 
-      logEvent(chunk, req, res, options, moesifController);
+      logEvent(resBodyBuf, req, res, options, moesifController);
     };
 
     next();

package.json

@@ -1,6 +1,6 @@
 {
   "name": "moesif-express",
-  "version": "2.3.4",
+  "version": "2.3.5",
   "description": "Collection/Data Ingestion Middleware for Express",
   "main": "./lib/index.js",
   "keywords": [