Using streams for getPlayers on SectionServer

Author: jamezrin

src/main/java/com/jaimemartz/playerbalancer/section/SectionServer.java

@@ -1,14 +1,19 @@
 package com.jaimemartz.playerbalancer.section;
 
+import com.google.common.collect.Iterables;
 import com.jaimemartz.playerbalancer.settings.props.features.BalancerProps;
 import net.md_5.bungee.BungeeServerInfo;
+import net.md_5.bungee.api.config.ServerInfo;
 import net.md_5.bungee.api.connection.ProxiedPlayer;
 
 import java.net.InetSocketAddress;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.List;
+import java.util.stream.Collector;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 public class SectionServer extends BungeeServerInfo {
     private final BalancerProps props;
@@ -29,13 +34,10 @@ public SectionServer(BalancerProps props, ServerSection section) {
     @Override
     public Collection<ProxiedPlayer> getPlayers() {
         if (props.isShowPlayers()) {
-            List<ProxiedPlayer> res = new ArrayList<>();
-            section.getServers().forEach(server -> {
-                res.addAll(server.getPlayers());
-            });
-            return res;
-        } else {
-            return Collections.emptyList();
-        }
+            return section.getServers().stream()
+                    .map(ServerInfo::getPlayers)
+                    .flatMap(Collection::stream)
+                    .collect(Collectors.toList());
+        } else return Collections.emptyList();
     }
 }