Reapply "feat: implement TempObjectSet for enhanced channel management and statistics tracking [please revert after testing]"

This reverts commit 5683e9c.

Author: twisti-dev

surf-api-bukkit/surf-api-bukkit-server/src/main/kotlin/dev/slne/surf/surfapi/bukkit/server/packet/listener/PlayerChannelInjector.kt

@@ -2,6 +2,7 @@ package dev.slne.surf.surfapi.bukkit.server.packet.listener
 
 import com.github.benmanes.caffeine.cache.Caffeine
 import com.sksamuel.aedile.core.expireAfterAccess
+import dev.slne.surf.surfapi.bukkit.api.extensions.server
 import dev.slne.surf.surfapi.bukkit.api.nms.NmsUseWithCaution
 import dev.slne.surf.surfapi.bukkit.api.nms.nmsBridge
 import dev.slne.surf.surfapi.bukkit.server.impl.nms.SurfBukkitNmsBridgeImpl
@@ -15,7 +16,6 @@ import dev.slne.surf.surfapi.core.api.reflection.SurfProxy
 import dev.slne.surf.surfapi.core.api.reflection.createProxy
 import dev.slne.surf.surfapi.core.api.reflection.surfReflection
 import dev.slne.surf.surfapi.core.api.util.logger
-import dev.slne.surf.surfapi.core.api.util.mutableObjectSetOf
 import dev.slne.surf.surfapi.core.api.util.synchronize
 import io.netty.channel.Channel
 import io.netty.channel.ChannelDuplexHandler
@@ -25,6 +25,8 @@ import io.papermc.paper.connection.PaperPlayerLoginConnection
 import io.papermc.paper.connection.ReadablePlayerCookieConnectionImpl
 import io.papermc.paper.event.connection.PlayerConnectionValidateLoginEvent
 import io.papermc.paper.network.ChannelInitializeListenerHolder
+import it.unimi.dsi.fastutil.objects.ObjectOpenHashSet
+import it.unimi.dsi.fastutil.objects.ObjectSet
 import net.kyori.adventure.key.Key
 import net.minecraft.network.Connection
 import net.minecraft.network.HandlerNames
@@ -35,6 +37,7 @@ import org.bukkit.event.EventPriority
 import org.bukkit.event.Listener
 import org.bukkit.event.player.PlayerJoinEvent
 import java.util.*
+import java.util.function.Predicate
 import kotlin.time.Duration.Companion.minutes
 import dev.slne.surf.surfapi.bukkit.api.event.register as registerListener
 import dev.slne.surf.surfapi.bukkit.api.event.unregister as unregisterListener
@@ -50,7 +53,62 @@ object PlayerChannelInjector : Listener {
         .expireAfterAccess(1.minutes)
         .build<UUID, Connection>()
 
-    private val injectedChannels = mutableObjectSetOf<Channel>().synchronize()
+    private val injectedChannels = TempObjectSet<Channel>().synchronize()
+
+    private class TempObjectSet<T>(set: ObjectSet<T>? = null) : ObjectOpenHashSet<T>(set) {
+        private var added: Long = 0
+        private var removed: Long = 0
+
+        private enum class Operation {
+            ADD, REMOVE, REMOVE_ALL, REMOVE_IF, ADD_ALL
+        }
+
+        fun printStats(operation: Operation) {
+            log.atInfo()
+                .log("O:$operation,A:$added,R:$removed,D:${added - removed},S:$size,P:${server.onlinePlayers.size}")
+        }
+
+        override fun addAll(c: Collection<T?>): Boolean {
+            added += c.size
+            printStats(Operation.ADD_ALL)
+
+            return super.addAll(c)
+        }
+
+        override fun removeAll(c: Collection<T?>): Boolean {
+            removed += c.size
+            printStats(Operation.REMOVE_ALL)
+
+            return super.removeAll(c)
+        }
+
+        override fun add(element: T?): Boolean {
+            added += 1
+            printStats(Operation.ADD)
+
+            return super.add(element)
+        }
+
+        override fun remove(element: T?): Boolean {
+            removed += 1
+            printStats(Operation.REMOVE)
+
+            return super.remove(element)
+        }
+
+        override fun removeIf(filter: Predicate<in T>): Boolean {
+            val removedElements = this.filter(filter::test)
+
+            removed += removedElements.size
+            printStats(Operation.REMOVE_IF)
+
+            return super.removeIf(filter)
+        }
+
+        companion object {
+            private const val serialVersionUID: Long = -2618969176232686100L
+        }
+    }
 
     fun register() {
         ChannelInitializeListenerHolder.addListener(CHANNEL_KEY) { this.injectChannel(it) }