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,88 @@
+/*
+ * 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;
+
+import io.moquette.persistence.remote.*;
+import io.moquette.server.Server;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import win.liyufan.im.ErrorCode;
+import java.util.concurrent.*;
+import java.util.concurrent.atomic.AtomicInteger;
+
+public class RPCCenter {
+    private static final Logger LOG = LoggerFactory.getLogger(RPCCenter.class);
+    public static final String CHECK_USER_ONLINE_REQUEST = "check_user_online";
+    public static final String KICKOFF_USER_REQUEST = "kickoff_user";
+
+    private Server server;
+    public ConcurrentHashMap<Integer, RequestInfo> requestMap = new ConcurrentHashMap<>();
+    private AtomicInteger aiRequestId = new AtomicInteger(1);
+    public ScheduledExecutorService scheduledExecutorService = Executors.newScheduledThreadPool(2);
+
+    public void init(Server server) {
+        this.server = server;
+    }
+
+    public interface Callback {
+        void onSuccess(byte[] response);
+        void onError(ErrorCode errorCode);
+        void onTimeout();
+        Executor getResponseExecutor();
+    }
+
+    private static RPCCenter instance;
+
+    public static RPCCenter getInstance() {
+        if (instance == null) {
+            instance = new RPCCenter();
+        }
+        return instance;
+    }
+
+    protected RPCCenter() {}
+
+    public void sendRequest(String fromUser, String clientId, String request, byte[] message, String target, TargetEntry.Type type, Callback callback) {
+        int requestId = 0;
+
+        if (callback != null) {
+            requestId = aiRequestId.incrementAndGet();
+            if (requestId == Integer.MAX_VALUE) {
+                if(!aiRequestId.compareAndSet(Integer.MAX_VALUE, 1)) {
+                    requestId = aiRequestId.incrementAndGet();
+                }
+            }
+            requestMap.put(requestId, new RequestInfo(fromUser, clientId, callback, message, requestId, request));
+        }
+
+        server.internalRpcMsg(fromUser, clientId, message, requestId, "", request);
+    }
+
+    public void sendResponse(int errorCode, byte[] message, String toUuid, int requestId) {
+        LOG.debug("send async reponse to {} with requestId {}", toUuid, requestId);
+        if (requestId > 0) {
+            RequestInfo info = requestMap.remove(requestId);
+            LOG.debug("receive async reponse requestId {}, errorCode {}", requestId, errorCode);
+            if(info != null) {
+                info.future.cancel(true);
+                if (info.callback != null) {
+                    info.callback.getResponseExecutor().execute(() -> {
+                        if (errorCode == 0) {
+                            info.callback.onSuccess(message);
+                        } else {
+                            info.callback.onError(ErrorCode.fromCode(errorCode));
+                        }
+                    });
+                } else {
+
+                }
+            }
+        }
+    }
+}