From 986f627d22a1aaeb55ce9f8fe5ff06646f6e7f89 Mon Sep 17 00:00:00 2001
From: creeper123123321 <7974274+creeper123123321@users.noreply.github.com>
Date: Thu, 1 Apr 2021 08:20:53 -0300
Subject: [PATCH] fix duplicated message

---
 .../com/github/creeper123123321/viaaas/web/WebClient.kt     | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/main/kotlin/com/github/creeper123123321/viaaas/web/WebClient.kt b/src/main/kotlin/com/github/creeper123123321/viaaas/web/WebClient.kt
index 5190f35..e3441b6 100644
--- a/src/main/kotlin/com/github/creeper123123321/viaaas/web/WebClient.kt
+++ b/src/main/kotlin/com/github/creeper123123321/viaaas/web/WebClient.kt
@@ -1,13 +1,11 @@
 package com.github.creeper123123321.viaaas.web
 
 import com.github.creeper123123321.viaaas.config.VIAaaSConfig
-import com.github.creeper123123321.viaaas.viaWebServer
 import com.google.common.collect.Sets
 import com.google.common.util.concurrent.RateLimiter
 import io.ktor.features.*
 import io.ktor.websocket.*
 import java.util.*
-import java.util.concurrent.ConcurrentHashMap
 import java.util.concurrent.atomic.AtomicInteger
 
 data class WebClient(
@@ -19,15 +17,15 @@ data class WebClient(
         val atInt = AtomicInteger()
         fun next() = atInt.getAndAdd(1)
     }
+
     val id = "${ws.call.request.local.remoteHost}(${ws.call.request.origin.remoteHost})-${IdGen.next()}"
     val listenedIds: MutableSet<UUID> = Sets.newConcurrentHashSet()
     val rateLimiter = RateLimiter.create(VIAaaSConfig.rateLimitWs)
 
     fun listenId(uuid: UUID): Boolean {
         if (listenedIds.size >= VIAaaSConfig.listeningWsLimit) return false // This is getting insane
-        server.listeners.put(uuid, this)
         listenedIds.add(uuid)
-        return true
+        return server.listeners.put(uuid, this)
     }
 
     fun unlistenId(uuid: UUID) {