database: include legacy query() in prometheus metrics

aleves64 · Aug 15, 2017 · c6c3baf · c6c3baf
1 parent 82004aa
commit c6c3baf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/database.js b/src/database.js
@@ -120,14 +120,17 @@ module.exports.query = function (query, sub, callback) {
         console.log(query);
     }
 
+    const end = queryLatency.startTimer();
     db.knex.raw(query, sub)
         .then(res => {
             process.nextTick(callback, null, res[0]);
         }).catch(error => {
             LOGGER.error('Legacy DB query failed.  Query: %s, Substitutions: %j, Error: %s', query, sub, error);
             process.nextTick(callback, 'Database failure', null);
         }).finally(() => {
+            end();
             Metrics.stopTimer(timer);
+            queryCount.inc(1, new Date());
         });
 };