From 8e77cb35ffc1d1e5becedfbfb40e250f86e0d456 Mon Sep 17 00:00:00 2001
From: md_5 <md_5@live.com.au>
Date: Sat, 28 Sep 2013 18:32:39 +1000
Subject: [PATCH] Use JDK map for throttle

---
 .../src/main/java/net/md_5/bungee/ConnectionThrottle.java | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

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<InetAddress> throttle = new TObjectLongHashMap<>();
+    private final Map<InetAddress, Long> 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;
     }
 }