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,42 @@
+/*
+ * 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.imhandler;
+
+import cn.wildfirechat.proto.WFCMessage;
+import io.moquette.spi.impl.Qos1PublishHandler;
+import io.netty.buffer.ByteBuf;
+import win.liyufan.im.ErrorCode;
+import win.liyufan.im.IMTopic;
+
+import java.util.LinkedHashSet;
+import java.util.Set;
+
+@Handler(value = IMTopic.RecallMessageTopic)
+public class RecallMessageHandler extends IMHandler<WFCMessage.INT64Buf> {
+    @Override
+    public ErrorCode action(ByteBuf ackPayload, String clientID, String fromUser, WFCMessage.INT64Buf int64Buf, Qos1PublishHandler.IMCallback callback) {
+        ErrorCode errorCode = m_messagesStore.recallMessage(int64Buf.getId(), fromUser);
+
+        if(errorCode != ErrorCode.ERROR_CODE_SUCCESS) {
+            return errorCode;
+        }
+
+        Set<String> notifyReceivers = new LinkedHashSet<>();
+        WFCMessage.Message message = m_messagesStore.getMessage(int64Buf.getId());
+        if (message == null) {
+            return ErrorCode.ERROR_CODE_NOT_EXIST;
+        }
+
+        m_messagesStore.getNotifyReceivers(fromUser, message, notifyReceivers);
+        this.publisher.publishRecall2Receivers(int64Buf.getId(), fromUser, notifyReceivers, clientID);
+
+        return errorCode;
+    }
+
+}