Update index.js

index.js

@@ -108,6 +108,42 @@ async function sendRequestAndSetGlobals() {
 
 setInterval(sendRequestAndSetGlobals, 3600000);
 
+let requestsCount = 0;
+let totalTime = 0;
+let domainsQueried = {};
+
+app.use((req, res, next) => {
+  const startTime = Date.now();
+  requestsCount++;
+  next();
+  const endTime = Date.now();
+  totalTime += endTime - startTime;
+  const averageResponseTime = totalTime / requestsCount;
+
+  // track domains queried
+  if (req.query.domain) {
+    const domain = req.query.domain;
+    if (!domainsQueried[domain]) {
+      domainsQueried[domain] = 1;
+    } else {
+      domainsQueried[domain]++;
+    }
+  }
+});
+
+app.get('/stats', (req, res) => {
+  const stats = {
+    requestsCount,
+    averageResponseTime,
+    uptime: process.uptime(),
+    domainsQueried: Object.keys(domainsQueried).map((domain) => ({
+      domain,
+      count: domainsQueried[domain],
+    })),
+  };
+
+  res.json(stats);
+});
 
 app.get('/whois', async (req, res) => {
   const { domain } = req.query;