fix event name cancel => close

nodules · May 17, 2019 · 810e118 · 810e118
1 parent d3dca78
commit 810e118
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/index.js b/index.js
@@ -17,22 +17,28 @@ module.exports = function requestTimeMiddleware(label, cb) {
 
         res
             .on('finish', onFinish)
-            .on('cancel', onCancel);
+            .on('close', onClose);
 
         function onFinish() {
+            unbind();
+
             var timeObj = timerStop();
             cb(timeObj.time, req);
         }
 
-        function onCancel() {
-            res
-                .removeListener('cancel', onCancel)
-                .removeListener('finish', onFinish);
+        function onClose() {
+            unbind();
 
             // FIXME: what should we do here?
             cb(0, req);
         }
 
+        function unbind() {
+            res
+                .removeListener('close', onClose)
+                .removeListener('finish', onFinish);
+        }
+
         next();
     };
 };