diff --git a/proxy/src/main/java/net/md_5/bungee/ConnectionThrottle.java b/proxy/src/main/java/net/md_5/bungee/ConnectionThrottle.java index e3c45fcd..dbf30da6 100644 --- a/proxy/src/main/java/net/md_5/bungee/ConnectionThrottle.java +++ b/proxy/src/main/java/net/md_5/bungee/ConnectionThrottle.java @@ -2,13 +2,15 @@ package net.md_5.bungee; import gnu.trove.map.hash.TObjectLongHashMap; import java.net.InetAddress; +import java.util.HashMap; +import java.util.Map; import lombok.RequiredArgsConstructor; @RequiredArgsConstructor public class ConnectionThrottle { - private final TObjectLongHashMap throttle = new TObjectLongHashMap<>(); + private final Map throttle = new HashMap<>(); private final int throttleTime; public void unthrottle(InetAddress address) @@ -18,10 +20,10 @@ public class ConnectionThrottle public boolean throttle(InetAddress address) { - long value = throttle.get( address ); + Long value = throttle.get( address ); long currentTime = System.currentTimeMillis(); throttle.put( address, currentTime ); - return value != 0 && currentTime - value < throttleTime; + return value != null && currentTime - value < throttleTime; } }