diff --git a/src/main/kotlin/dev/arbjerg/lavalink/client/LavalinkClient.kt b/src/main/kotlin/dev/arbjerg/lavalink/client/LavalinkClient.kt index e91ed5d..7bfe4ee 100644 --- a/src/main/kotlin/dev/arbjerg/lavalink/client/LavalinkClient.kt +++ b/src/main/kotlin/dev/arbjerg/lavalink/client/LavalinkClient.kt @@ -104,12 +104,9 @@ class LavalinkClient(val userId: Long) : Closeable, Disposable { */ @JvmOverloads fun getOrCreateLink(guildId: Long, region: VoiceRegion? = null): Link { - if (!linkMap.containsKey(guildId)) { - val bestNode = loadBalancer.selectNode(region, guildId) - linkMap[guildId] = Link(guildId, bestNode) + return linkMap.getOrPut(guildId) { + Link(guildId, loadBalancer.selectNode(region, guildId)) } - - return linkMap[guildId]!! } /**