From c6c3bafca268c7b79791a175a1f72f47eac735f6 Mon Sep 17 00:00:00 2001
From: Calvin Montgomery <calzoneman@gmail.com>
Date: Mon, 14 Aug 2017 18:24:53 -0700
Subject: [PATCH] database: include legacy query() in prometheus metrics

---
 src/database.js | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/database.js b/src/database.js
index 9ac47194..b8512cdb 100644
--- a/src/database.js
+++ b/src/database.js
@@ -120,6 +120,7 @@ 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]);
@@ -127,7 +128,9 @@ module.exports.query = function (query, sub, callback) {
             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());
         });
 };