Revert "refactor a bit"

This reverts commit 20b1dd5.

Author: tipsy

src/main/kotlin/app/GhService.kt

@@ -18,6 +18,9 @@ object GhService {
 
     private val log = LoggerFactory.getLogger(GhService.javaClass)
 
+    // Allows for parallel iteration and O(1) put/remove
+    private val clientSessions = ConcurrentHashMap<WsSession, Boolean>()
+
     private val tokens = Config.getApiTokens()?.split(",") ?: listOf("") // empty token is limited to 60 requests
     private val clients = tokens.map { GitHubClient().apply { setOAuth2Token(it) } }
     private val repoServices = clients.map { RepositoryService(it) }
@@ -32,9 +35,8 @@ object GhService {
 
     val remainingRequests: Int get() = clients.sumBy { it.remainingRequests }
 
-    // Allows for parallel iteration and O(1) put/remove
-    private val clientSessions = ConcurrentHashMap<WsSession, Boolean>()
     fun registerClient(ws: WsSession) = clientSessions.put(ws, true) == true
+
     fun unregisterClient(ws: WsSession) = clientSessions.remove(ws) == true
 
     init {
@@ -54,11 +56,13 @@ object GhService {
 
             // update all connected clients with remainingRequests twice per second
             scheduleAtFixedRate({
+                val payload = remainingRequests.toString()
                 clientSessions.forEachKey(1) {
                     try {
-                        it.send(remainingRequests.toString())
-                    } catch (e: Exception) {
-                        log.error(e.toString())
+                        if (it.isOpen)
+                            it.send(payload)
+                    } catch (e: IOException) {
+                        log?.error(e.toString())
                     }
                 }
             }, 0, 500, TimeUnit.MILLISECONDS)

src/main/kotlin/app/Main.kt

@@ -61,7 +61,7 @@ fun main(args: Array<String>) {
         }
 
         ws("/rate-limit-status") { ws ->
-            ws.onConnect { session -> GhService.registerClient(session) }
+            ws.onConnect { GhService.registerClient(it) }
             ws.onClose { session, _, _ -> GhService.unregisterClient(session) }
             ws.onError { session, _ -> GhService.unregisterClient(session) }
         }