Fixed mem leak due to caching of object without hasCode (NettyChannelContext

broker/src/main/java/org/eclipse/moquette/server/netty/NettyMQTTHandler.java

@@ -37,7 +37,7 @@ public class NettyMQTTHandler extends ChannelInboundHandlerAdapter {
     
     private static final Logger LOG = LoggerFactory.getLogger(NettyMQTTHandler.class);
     private IMessaging m_messaging;
-    private final Map<ChannelHandlerContext, NettyChannel> m_channelMapper = new HashMap<ChannelHandlerContext, NettyChannel>();
+//    private final Map<ChannelHandlerContext, NettyChannel> m_channelMapper = new HashMap<>();
     
     @Override
     public void channelRead(ChannelHandlerContext ctx, Object message) {
@@ -54,15 +54,15 @@ public class NettyMQTTHandler extends ChannelInboundHandlerAdapter {
                 case PUBREL:
                 case DISCONNECT:
                 case PUBACK:    
-                    NettyChannel channel;
-                    synchronized(m_channelMapper) {
-                        if (!m_channelMapper.containsKey(ctx)) {
-                            m_channelMapper.put(ctx, new NettyChannel(ctx));
-                        }
-                        channel = m_channelMapper.get(ctx);
-                    }
+//                    NettyChannel channel;
+//                    synchronized(m_channelMapper) {
+//                        if (!m_channelMapper.containsKey(ctx)) {
+//                            m_channelMapper.put(ctx, new NettyChannel(ctx));
+//                        }
+//                        channel = m_channelMapper.get(ctx);
+//                    }
                     
-                    m_messaging.handleProtocolMessage(channel, msg);
+                    m_messaging.handleProtocolMessage(new NettyChannel(ctx), msg);
                     break;
                 case PINGREQ:
                     PingRespMessage pingResp = new PingRespMessage();
@@ -76,13 +76,14 @@ public class NettyMQTTHandler extends ChannelInboundHandlerAdapter {
     
     @Override
     public void channelInactive(ChannelHandlerContext ctx) throws Exception {
-        NettyChannel channel = m_channelMapper.get(ctx);
-        String clientID = (String) channel.getAttribute(NettyChannel.ATTR_KEY_CLIENTID);
-        m_messaging.lostConnection(channel, clientID);
+//        NettyChannel channel = m_channelMapper.get(ctx);
+//        String clientID = (String) channel.getAttribute(NettyChannel.ATTR_KEY_CLIENTID);
+        String clientID = (String) NettyUtils.getAttribute(ctx, NettyChannel.ATTR_KEY_CLIENTID);
+        m_messaging.lostConnection(clientID);
         ctx.close(/*false*/);
-        synchronized(m_channelMapper) {
-            m_channelMapper.remove(ctx);
-        }
+//        synchronized(m_channelMapper) {
+//            m_channelMapper.remove(ctx);
+//        }
     }
     
     public void setMessaging(IMessaging messaging) {

broker/src/main/java/org/eclipse/moquette/server/netty/NettyUtils.java

@@ -0,0 +1,50 @@
+/*
+ * Copyright (c) 2012-2014 The original author or authors
+ * ------------------------------------------------------
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * and Apache License v2.0 which accompanies this distribution.
+ *
+ * The Eclipse Public License is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * The Apache License v2.0 is available at
+ * http://www.opensource.org/licenses/apache2.0.php
+ *
+ * You may elect to redistribute this code under either of these licenses.
+ */
+package org.eclipse.moquette.server.netty;
+
+import io.netty.channel.ChannelHandlerContext;
+import io.netty.handler.timeout.IdleStateHandler;
+import io.netty.util.Attribute;
+import io.netty.util.AttributeKey;
+
+/**
+ * Some Netty's channels utilities.
+ *
+ * @author andrea
+ */
+public class NettyUtils {
+
+    public static Object getAttribute(ChannelHandlerContext channel, AttributeKey<Object> key) {
+        Attribute<Object> attr = channel.attr(key);
+        return attr.get();
+    }
+
+    public static void setAttribute(ChannelHandlerContext channel, AttributeKey<Object> key, Object value) {
+        Attribute<Object> attr = channel.attr(key);
+        attr.set(value);
+    }
+
+    public static void setIdleTime(ChannelHandlerContext channel, int idleTime) {
+        if (channel.pipeline().names().contains("idleStateHandler")) {
+            channel.pipeline().remove("idleStateHandler");
+        }
+        if (channel.pipeline().names().contains("idleEventHandler")) {
+            channel.pipeline().remove("idleEventHandler");
+        }
+        channel.pipeline().addFirst("idleStateHandler", new IdleStateHandler(0, 0, idleTime));
+        channel.pipeline().addAfter("idleStateHandler", "idleEventHandler", new MoquetteIdleTimoutHandler());
+    }
+}