From dc24fa6b48d7de014cd0c8a2a0977d0e46612db6 Mon Sep 17 00:00:00 2001
From: Christine Dodrill <xena@yolo-swag.com>
Date: Thu, 27 Aug 2015 20:32:07 -0700
Subject: [PATCH] Collect and report statistics about usage of the api server

---
 ponyapi.nim | 24 ++++++++++++++++++++++++
 stats.nim   | 25 +++++++++++++++++++++++++
 2 files changed, 49 insertions(+)
 create mode 100644 stats.nim

diff --git a/ponyapi.nim b/ponyapi.nim
index d78ff05..0224181 100644
--- a/ponyapi.nim
+++ b/ponyapi.nim
@@ -5,6 +5,7 @@ import jester
 import json
 import os
 import random
+import stats
 import strutils
 import times
 
@@ -66,6 +67,7 @@ routes:
     "http://github.com/Xe/PonyAPI".uri.redirect
 
   get "/all":
+    stats.all.success.inc
     resp Http200, myHeaders, pretty(%%episodes, 4)
 
   get "/newest":
@@ -80,9 +82,11 @@ routes:
         ep = episode
         break
 
+    stats.newest.success.inc
     resp Http200, myHeaders, pretty(%%ep, 4)
 
   get "/random":
+    stats.newest.success.inc
     resp Http200, myHeaders, pretty(%%episodes.randomChoice(), 4)
 
   get "/last_aired":
@@ -97,6 +101,7 @@ routes:
         ep = episodes[epid-1]
         break
 
+    stats.lastAired.success.inc
     resp Http200, myHeaders, pretty(%%ep, 4)
 
   get "/season/@snumber":
@@ -105,8 +110,10 @@ routes:
       eps: seq[Episode] = lc[x | (x <- episodes, x.season == season), Episode]
 
     if eps.len == 0:
+      stats.seasonLookup.fails.inc
       resp Http404, myHeaders, pretty(%!"No episodes found", 4)
     else:
+      stats.seasonLookup.success.inc
       resp Http200, myHeaders, pretty(%%eps, 4)
 
   get "/season/@snumber/episode/@epnumber":
@@ -121,8 +128,10 @@ routes:
           ep = episode
 
     if ep.air_date == 0:
+      stats.episodeLookup.fails.inc
       resp Http404, myHeaders, pretty(%!"Not found", 4)
     else:
+      stats.episodeLookup.success.inc
       resp Http200, myHeaders, pretty(%%ep, 4)
 
   get "/search":
@@ -130,6 +139,7 @@ routes:
       query = @"q".toLower
 
     if query == "":
+      stats.search.fails.inc
       halt Http406, myHeaders, pretty(%!"Need to specify a query", 4)
 
     var
@@ -137,8 +147,22 @@ routes:
         lc[x | (x <- episodes, x.name.toLower.contains query), Episode]
 
     if eps.len == 0:
+      stats.search.fails.inc
       resp Http404, myHeaders, pretty(%!"No episodes found", 4)
     else:
+      stats.search.success.inc
       resp Http200, myHeaders, pretty(%%eps, 4)
 
+  get "/_stats":
+    resp Http200, myHeaders, pretty(%*
+      [
+        stats.all,
+        stats.newest,
+        stats.random,
+        stats.lastAired,
+        stats.seasonLookup,
+        stats.episodeLookup,
+        stats.search,
+      ], 4)
+
 runForever()
diff --git a/stats.nim b/stats.nim
new file mode 100644
index 0000000..78d0eb2
--- /dev/null
+++ b/stats.nim
@@ -0,0 +1,25 @@
+import json
+
+type
+  Bucket* = object of RootObj
+    ## A collection of stats
+    name*: string
+    fails*: int
+    success*: int
+
+proc `%`*(b: Bucket): JsonNode =
+  %*
+    {
+      "name": b.name,
+      "fails": b.fails,
+      "success": b.success,
+    }
+
+var
+  all* = Bucket(name: "all")
+  newest* = Bucket(name: "newest")
+  random* = Bucket(name: "random")
+  lastAired* = Bucket(name: "lastAired")
+  seasonLookup* = Bucket(name: "seasonLookup")
+  episodeLookup* = Bucket(name: "episodeLookup")
+  search* = Bucket(name: "search")