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,55 @@
+/*
+ * 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.ProtoConstants;
+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 win.liyufan.im.MessageShardingUtil;
+
+import static win.liyufan.im.ErrorCode.ERROR_CODE_SUCCESS;
+
+@Handler(IMTopic.HandleFriendRequestTopic)
+public class HandleFriendRequestHandler extends IMHandler<WFCMessage.HandleFriendRequest> {
+    @Override
+    public ErrorCode action(ByteBuf ackPayload, String clientID, String fromUser, WFCMessage.HandleFriendRequest request, Qos1PublishHandler.IMCallback callback) {
+            WFCMessage.Message.Builder builder = WFCMessage.Message.newBuilder();
+            builder.setFromUser(request.getTargetUid());
+            long[] heads = new long[2];
+            ErrorCode errorCode = m_messagesStore.handleFriendRequest(fromUser, request, builder, heads);
+
+            if (errorCode == ERROR_CODE_SUCCESS) {
+                long messageId = MessageShardingUtil.generateId();
+                builder.setMessageId(messageId);
+                saveAndPublish(request.getTargetUid(), null, builder.build(), System.currentTimeMillis());
+
+                WFCMessage.MessageContent.Builder contentBuilder = WFCMessage.MessageContent.newBuilder().setType(90).setContent("以上是打招呼信息");
+                builder = WFCMessage.Message.newBuilder();
+                builder.setFromUser(request.getTargetUid());
+                builder.setConversation(WFCMessage.Conversation.newBuilder().setTarget(fromUser).setLine(0).setType(ProtoConstants.ConversationType.ConversationType_Private).build());
+                builder.setContent(contentBuilder);
+                messageId = MessageShardingUtil.generateId();
+                builder.setMessageId(messageId);
+                saveAndPublish(request.getTargetUid(), null, builder.build(), System.currentTimeMillis());
+
+                contentBuilder.setContent("你们已经成为好友了，现在可以开始聊天了");
+                builder.setContent(contentBuilder);
+                messageId = MessageShardingUtil.generateId();
+                builder.setMessageId(messageId);
+                saveAndPublish(request.getTargetUid(), null, builder.build(), System.currentTimeMillis());
+
+                publisher.publishNotification(IMTopic.NotifyFriendTopic, request.getTargetUid(), heads[0]);
+                publisher.publishNotification(IMTopic.NotifyFriendTopic, fromUser, heads[1]);
+            }
+            return errorCode;
+    }
+}