Fix runtime stats sending

mthenw · Nov 28, 2018 · c35287c · c35287c
1 parent 28de6d1
commit c35287c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/index.js b/index.js
@@ -157,8 +157,9 @@ if (program.daemonize) {
    * Handle signals
    */
   const cleanExit = () => {
-    stats.timeEnd('runtime', 'runtime');
-    process.exit();
+    stats.timeEnd('runtime', 'runtime', () => {
+      process.exit();
+    });
   };
   process.on('SIGINT', cleanExit);
   process.on('SIGTERM', cleanExit);

diff --git a/lib/stats.js b/lib/stats.js
@@ -54,12 +54,12 @@ Stats.prototype.time = function time(category, action) {
   this.timer[category][action] = Date.now();
 };
 
-Stats.prototype.timeEnd = function timeEnd(category, action) {
+Stats.prototype.timeEnd = function timeEnd(category, action, cb) {
   if (!this.tracker) {
     return;
   }
 
-  this.tracker.timing(category, action, Date.now() - this.timer[category][action]).send();
+  this.tracker.timing(category, action, Date.now() - this.timer[category][action]).send(cb);
 };
 
 module.exports = enabled => new Stats(enabled);