Upstream/VIAaaS
1
0
Fork 0
mirror of https://github.com/ViaVersion/VIAaaS.git synced 2025-02-11 00:52:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix duplicated message

Browse Source
This commit is contained in:
creeper123123321 2021-04-01 08:20:53 -03:00
parent aea3632671
commit 986f627d22
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/main/kotlin/com/github/creeper123123321/viaaas/web/WebClient.kt
View File

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