Add PlayerService.count()

anvilpowered · Jun 10, 2024 · 4df899f · 4df899f
1 parent b9181ba
commit 4df899f
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/core/src/main/kotlin/org/anvilpowered/anvil/core/user/PlayerService.kt b/core/src/main/kotlin/org/anvilpowered/anvil/core/user/PlayerService.kt
@@ -27,4 +27,6 @@ interface PlayerService {
     operator fun get(id: UUID): Player?
 
     fun getAll(startsWith: String = ""): Sequence<Player>
+
+    fun count(): Int
 }
diff --git a/paper/src/main/kotlin/org/anvilpowered/anvil/paper/user/PaperPlayerService.kt b/paper/src/main/kotlin/org/anvilpowered/anvil/paper/user/PaperPlayerService.kt
@@ -34,4 +34,6 @@ object PaperPlayerService : PlayerService {
         "" -> Bukkit.getOnlinePlayers().asSequence().map { it.toAnvilPlayer() }
         else -> Bukkit.matchPlayer(startsWith).asSequence().map { it.toAnvilPlayer() }
     }
+
+    override fun count(): Int = Bukkit.getOnlinePlayers().size
 }
diff --git a/sponge/src/main/kotlin/org/anvilpowered/anvil/sponge/user/SpongePlayerService.kt b/sponge/src/main/kotlin/org/anvilpowered/anvil/sponge/user/SpongePlayerService.kt
@@ -34,4 +34,6 @@ object SpongePlayerService : PlayerService {
         "" -> Sponge.server().onlinePlayers().asSequence().map { it.toAnvilPlayer() }
         else -> Sponge.server().onlinePlayers().asSequence().filter { it.name().startsWith(startsWith) }.map { it.toAnvilPlayer() }
     }
+
+    override fun count(): Int = Sponge.server().onlinePlayers().size
 }
diff --git a/velocity/src/main/kotlin/org/anvilpowered/anvil/velocity/user/VelocityPlayerService.kt b/velocity/src/main/kotlin/org/anvilpowered/anvil/velocity/user/VelocityPlayerService.kt
@@ -34,4 +34,6 @@ class VelocityPlayerService(private val proxyServer: ProxyServer) : PlayerServic
         "" -> proxyServer.allPlayers.asSequence().map { it.toAnvilPlayer() }
         else -> proxyServer.matchPlayer(startsWith).asSequence().map { it.toAnvilPlayer() }
     }
+
+    override fun count(): Int = proxyServer.playerCount
 }