diff --git a/src/main/java/com/jaimemartz/playerbalancer/section/SectionServer.java b/src/main/java/com/jaimemartz/playerbalancer/section/SectionServer.java index 729bc21..53e873f 100644 --- a/src/main/java/com/jaimemartz/playerbalancer/section/SectionServer.java +++ b/src/main/java/com/jaimemartz/playerbalancer/section/SectionServer.java @@ -1,7 +1,9 @@ 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; @@ -9,6 +11,9 @@ 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 class SectionServer extends BungeeServerInfo { @Override public Collection getPlayers() { if (props.isShowPlayers()) { - List 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(); } }