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,45 @@
+/*
+ * This file is part of the Wildfire Chat package.
+ * (c) Heavyrain2012 <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+package io.moquette.persistence.remote;
+
+import io.moquette.persistence.RPCCenter;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.concurrent.ScheduledFuture;
+import java.util.concurrent.TimeUnit;
+
+public class RequestInfo {
+    private static final Logger LOG = LoggerFactory.getLogger(RequestInfo.class);
+    public final RPCCenter.Callback callback;
+    final byte[] message;
+    final int requestId;
+    final String fromUser;
+    final String request;
+    final String clientId;
+    public final ScheduledFuture future;
+
+    public RequestInfo(String fromUser, String clientId, RPCCenter.Callback callback, byte[] message, int requestId, String request) {
+        this.callback = callback;
+        this.message = message;
+        this.requestId = requestId;
+        this.fromUser = fromUser;
+        this.clientId = clientId;
+        this.request = request;
+        this.future = RPCCenter.getInstance().scheduledExecutorService.schedule(() -> {
+            RequestInfo info = RPCCenter.getInstance().requestMap.remove(requestId);
+            if (info != null) {
+                LOG.error("Request timeout. fromUser {}, cliendId {}, requestId {}, request {}", fromUser, clientId, requestId, request);
+                info.callback.getResponseExecutor().execute(() -> {
+                    info.callback.onTimeout();
+                });
+            }
+        }, 5, TimeUnit.SECONDS);
+    }
+}