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>
This commit is contained in:
Rohit Yadav 2019-10-22 14:58:11 +05:30
parent 9ee129ae6a
commit 9258c7c3c9

View File

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