From 74461609c16696c2da109d341c9a48e99d8c6bab Mon Sep 17 00:00:00 2001
From: Myles <mylescomputer@gmail.com>
Date: Mon, 21 Mar 2016 19:42:52 +0000
Subject: [PATCH] Change to use maps, much better for optimisation.

---
 .../ViaVersion/api/data/UserConnection.java    | 18 ++++++------------
 1 file changed, 6 insertions(+), 12 deletions(-)

diff --git a/src/main/java/us/myles/ViaVersion/api/data/UserConnection.java b/src/main/java/us/myles/ViaVersion/api/data/UserConnection.java
index 9a702dd6e..8f87f33e1 100644
--- a/src/main/java/us/myles/ViaVersion/api/data/UserConnection.java
+++ b/src/main/java/us/myles/ViaVersion/api/data/UserConnection.java
@@ -8,11 +8,13 @@ import lombok.Setter;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 
 public class UserConnection {
     @Getter
     private final SocketChannel channel;
-    List<StoredObject> storedObjects = new ArrayList<>();
+    Map<Class, StoredObject> storedObjects = new ConcurrentHashMap<>();
     @Getter
     @Setter
     private boolean active = true;
@@ -26,23 +28,15 @@ public class UserConnection {
     }
 
     public <T extends StoredObject> T get(Class<T> objectClass) {
-        for (StoredObject o : storedObjects) {
-            if (o.getClass().equals(objectClass))
-                return (T) o;
-        }
-        return null;
+        return (T) storedObjects.get(objectClass);
     }
 
     public <T extends StoredObject> boolean has(Class<T> objectClass) {
-        for (StoredObject o : storedObjects) {
-            if (o.getClass().equals(objectClass))
-                return true;
-        }
-        return false;
+        return storedObjects.containsKey(objectClass);
     }
 
     public void put(StoredObject object) {
-        storedObjects.add(object);
+        storedObjects.put(object.getClass(), object);
     }
 
     public void sendRawPacket(final ByteBuf packet, boolean currentThread) {