wildfire chat server source initial submit

2019-03-16 17:04:23 +08:00
parent e59217bd10
commit cab3ca0916
1466 changed files with 528862 additions and 15919 deletions
@@ -0,0 +1,162 @@
+/*
+ * Copyright (c) 2012-2017 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 io.moquette.server;
+
+import java.util.NoSuchElementException;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicReference;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import io.moquette.server.netty.AutoFlushHandler;
+import io.moquette.server.netty.NettyUtils;
+import io.moquette.server.netty.metrics.BytesMetrics;
+import io.moquette.server.netty.metrics.BytesMetricsHandler;
+import io.moquette.server.netty.metrics.MessageMetrics;
+import io.moquette.server.netty.metrics.MessageMetricsHandler;
+import io.netty.channel.Channel;
+
+/**
+ * Value object to maintain the information of single connection, like ClientID, Channel, and clean
+ * session flag.
+ */
+public class ConnectionDescriptor {
+
+    private static final Logger LOG = LoggerFactory.getLogger(ConnectionDescriptor.class);
+
+    public enum ConnectionState {
+        // Connection states
+        DISCONNECTED,
+        SENDACK,
+        SESSION_CREATED,
+        MESSAGES_REPUBLISHED,
+        ESTABLISHED,
+        // Disconnection states
+        MESSAGES_DROPPED,
+        INTERCEPTORS_NOTIFIED;
+    }
+
+    public final String clientID;
+    private final Channel channel;
+    private final AtomicReference<ConnectionState> channelState = new AtomicReference<>(ConnectionState.DISCONNECTED);
+
+    public ConnectionDescriptor(String clientID, Channel session) {
+        this.clientID = clientID;
+        this.channel = session;
+    }
+
+    public void writeAndFlush(Object payload) {
+        this.channel.writeAndFlush(payload);
+    }
+
+    public void setupAutoFlusher(int flushIntervalMs) {
+        try {
+            this.channel.pipeline().addAfter(
+                    "idleEventHandler",
+                    "autoFlusher",
+                    new AutoFlushHandler(flushIntervalMs, TimeUnit.MILLISECONDS));
+        } catch (NoSuchElementException nseex) {
+            // the idleEventHandler is not present on the pipeline
+            this.channel.pipeline()
+                    .addFirst("autoFlusher", new AutoFlushHandler(flushIntervalMs, TimeUnit.MILLISECONDS));
+        }
+    }
+
+    public boolean doesNotUseChannel(Channel channel) {
+        return !(this.channel.equals(channel));
+    }
+
+    public boolean close() {
+        LOG.info("Closing connection descriptor. MqttClientId = {}.", clientID);
+        final boolean success = assignState(ConnectionState.INTERCEPTORS_NOTIFIED, ConnectionState.DISCONNECTED);
+        if (!success) {
+            return false;
+        }
+        this.channel.close();
+        return true;
+    }
+
+    public String getUsername() {
+        return NettyUtils.userName(this.channel);
+    }
+
+    public void abort() {
+        LOG.info("Closing connection descriptor. MqttClientId = {}.", clientID);
+        // try {
+        // this.channel.disconnect().sync();
+        this.channel.close(); // .sync();
+        // } catch (InterruptedException e) {
+        // e.printStackTrace();
+        // }
+    }
+
+    public boolean assignState(ConnectionState expected, ConnectionState newState) {
+        LOG.debug(
+                "Updating state of connection descriptor. MqttClientId = {}, expectedState = {}, newState = {}.",
+                clientID,
+                expected,
+                newState);
+        boolean retval = channelState.compareAndSet(expected, newState);
+        if (!retval) {
+            LOG.error(
+                    "Unable to update state of connection descriptor."
+                    + " MqttclientId = {}, expectedState = {}, newState = {}.",
+                    clientID,
+                    expected,
+                    newState);
+        }
+        return retval;
+    }
+
+    @Override
+    public String toString() {
+        return "ConnectionDescriptor{" + "clientID=" + clientID + ", state="
+                + channelState.get() + '}';
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o)
+            return true;
+        if (o == null || getClass() != o.getClass())
+            return false;
+
+        ConnectionDescriptor that = (ConnectionDescriptor) o;
+
+        if (clientID != null ? !clientID.equals(that.clientID) : that.clientID != null)
+            return false;
+        return !(channel != null ? !channel.equals(that.channel) : that.channel != null);
+    }
+
+    public BytesMetrics getBytesMetrics() {
+        return BytesMetricsHandler.getBytesMetrics(channel);
+    }
+
+    public MessageMetrics getMessageMetrics() {
+        return MessageMetricsHandler.getMessageMetrics(channel);
+    }
+
+    @Override
+    public int hashCode() {
+        int result = clientID != null ? clientID.hashCode() : 0;
+        result = 31 * result + (channel != null ? channel.hashCode() : 0);
+        return result;
+    }
+
+    public Channel getChannel() {
+        return channel;
+    }
+}