Optimize the number of audio send factories

Author: freyacodes

LavalinkServer/src/main/java/lavalink/server/io/SocketContext.java

@@ -55,6 +55,8 @@ public class SocketContext {
     private final Map<String, Player> players = new ConcurrentHashMap<>();
     private ScheduledExecutorService statsExecutor;
     public final ScheduledExecutorService playerUpdateService;
+    private static final int audioSendFactoryCount = Runtime.getRuntime().availableProcessors() * 2;
+    private final ConcurrentHashMap<Integer, IAudioSendFactory> sendFactories = new ConcurrentHashMap<>();
 
     SocketContext(WebSocket socket, String userId, int shardCount) {
         this.socket = socket;
@@ -76,7 +78,7 @@ Core getCore(int shardId) {
         return cores.computeIfAbsent(shardId,
                 __ -> {
                     if (nasSupported)
-                        return new Core(userId, new CoreClientImpl(socket, shardId), createAudioSendFactory());
+                        return new Core(userId, new CoreClientImpl(socket, shardId), getAudioSendFactory(shardId));
                     else
                         return new Core(userId, new CoreClientImpl(socket, shardId));
                 }
@@ -126,16 +128,18 @@ void shutdown() {
         players.values().forEach(Player::stop);
     }
 
-    private IAudioSendFactory createAudioSendFactory() {
-        Integer customBuffer = Launcher.config.getBufferDurationMs();
-        NativeAudioSendFactory nativeAudioSendFactory;
-        if (customBuffer != null) {
-            nativeAudioSendFactory = new NativeAudioSendFactory(customBuffer);
-        } else {
-            nativeAudioSendFactory = new NativeAudioSendFactory();
-        }
+    private IAudioSendFactory getAudioSendFactory(int shardId) {
+        return sendFactories.computeIfAbsent(shardId % audioSendFactoryCount, integer -> {
+            Integer customBuffer = Launcher.config.getBufferDurationMs();
+            NativeAudioSendFactory nativeAudioSendFactory;
+            if (customBuffer != null) {
+                nativeAudioSendFactory = new NativeAudioSendFactory(customBuffer);
+            } else {
+                nativeAudioSendFactory = new NativeAudioSendFactory();
+            }
 
-        return AsyncPacketProviderFactory.adapt(nativeAudioSendFactory);
+            return AsyncPacketProviderFactory.adapt(nativeAudioSendFactory);
+        });
     }
 
 }