Revert "Refactor RateLimitUtil for improved concurrency (#68)"

This reverts commit d8f6e6d.

Author: tipsy

src/main/kotlin/app/util/RateLimitUtil.kt

@@ -1,9 +1,8 @@
 package app.util
 
 import io.javalin.Javalin
+import java.util.*
 import java.util.concurrent.ConcurrentHashMap
-import java.util.concurrent.Executors
-import java.util.concurrent.TimeUnit
 
 /**
  * A very naive IP-based rate-limiting mechanism
@@ -22,28 +21,30 @@ object RateLimitUtil {
     fun enableTerribleRateLimiting(app: Javalin) {
 
         app.before { ctx ->
-            ipReqCount.compute(ctx.ip(), { _, count ->
-                when (count) {
-                    null -> 1
-                    in 0..25 -> count + 1
-                    else -> throw TerribleRateLimitException()
-                }
-            })
+            if (ipReqCount[ctx.ip()] ?: 0 > 25) {
+                throw TerribleRateLimitException()
+            }
+            ipReqCount[ctx.ip()] = (ipReqCount[ctx.ip()] ?: 0) + 1
         }
 
-        app.exception(TerribleRateLimitException::class.java) { _, ctx ->
+        app.exception(TerribleRateLimitException::class.java) { e, ctx ->
             ctx.result("You can't spam this much. I'll give you a new request every five seconds.")
         }
 
-        Executors.newSingleThreadScheduledExecutor()
-                .scheduleAtFixedRate(decrementAllCounters, 0, 5, TimeUnit.SECONDS)
+        Timer().scheduleAtFixedRate(decrementAllCounters(), 0, 5000) // every 5s
 
     }
 
-    private val decrementAllCounters = Runnable {
-        ipReqCount.forEachKey(1, { ip ->
-            ipReqCount.computeIfPresent(ip, { _, count -> if (count > 1) count - 1 else null })
-        })
+    private fun decrementAllCounters() = object : TimerTask() {
+        override fun run() {
+            ipReqCount.forEach { ip, count ->
+                if (count > 0) {
+                    ipReqCount[ip] = ipReqCount[ip]!! - 1
+                } else {
+                    ipReqCount.remove(ip)
+                }
+            }
+        }
     }
 
 }