Revert "9ee129ae6a08935645734ad520bec72bf5f22217"

This reverts "9ee129ae6a08935645734ad520bec72bf5f22217" from #3525, as
the solution is not optimal and would cause a single Link to be added
in the Map if management server is behind a LB.

Signed-off-by: Rohit Yadav <rohit.yadav@shapeblue.com>

Author: yadvr

utils/src/main/java/com/cloud/utils/nio/NioServer.java

@@ -25,7 +25,6 @@
 import java.nio.channels.SelectionKey;
 import java.nio.channels.ServerSocketChannel;
 import java.nio.channels.spi.SelectorProvider;
-import java.util.Map;
 import java.util.WeakHashMap;
 
 import org.apache.cloudstack.framework.ca.CAService;
@@ -37,13 +36,13 @@ public class NioServer extends NioConnection {
     protected InetSocketAddress _localAddr;
     private ServerSocketChannel _serverSocket;
 
-    protected Map<String, Link> _links;
+    protected WeakHashMap<InetSocketAddress, Link> _links;
 
     public NioServer(final String name, final int port, final int workers, final HandlerFactory factory, final CAService caService) {
         super(name, port, workers, factory);
         setCAService(caService);
         _localAddr = null;
-        _links = new WeakHashMap<String, Link>(10240);
+        _links = new WeakHashMap<InetSocketAddress, Link>(1024);
     }
 
     public int getPort() {
@@ -76,12 +75,12 @@ public void cleanUp() throws IOException {
 
     @Override
     protected void registerLink(final InetSocketAddress addr, final Link link) {
-        _links.put(addr.getAddress().toString(), link);
+        _links.put(addr, link);
     }
 
     @Override
     protected void unregisterLink(final InetSocketAddress saddr) {
-        _links.remove(saddr.getAddress().toString());
+        _links.remove(saddr);
     }
 
     /**
@@ -94,7 +93,7 @@ protected void unregisterLink(final InetSocketAddress saddr) {
      * @return null if not sent.  attach object in link if sent.
      */
     public Object send(final InetSocketAddress saddr, final byte[] data) throws ClosedChannelException {
-        final Link link = _links.get(saddr.getAddress().toString());
+        final Link link = _links.get(saddr);
         if (link == null) {
             return null;
         }