sdk支持流式文本

2023-12-30 19:09:30 +08:00
parent bc409042b0
commit 8a13a32615
2 changed files with 85 additions and 15 deletions
@@ -0,0 +1,48 @@
+package cn.wildfirechat.messagecontentbuilder;
+
+import cn.wildfirechat.pojos.MessagePayload;
+import cn.wildfirechat.proto.ProtoConstants;
+import com.google.gson.Gson;
+
+import java.nio.charset.StandardCharsets;
+import java.util.Base64;
+import java.util.HashMap;
+import java.util.Map;
+
+public class StreamingTextMessageContentBuilder extends MessageContentBuilder{
+    private String text;
+    private String streamId;
+    private boolean generating;
+    public static StreamingTextMessageContentBuilder newBuilder(String streamId) {
+        StreamingTextMessageContentBuilder builder = new StreamingTextMessageContentBuilder();
+        builder.generating = false;
+        builder.streamId = streamId;
+        return builder;
+    }
+    public StreamingTextMessageContentBuilder text(String text) {
+        this.text = text;
+        return this;
+    }
+
+    public StreamingTextMessageContentBuilder generating(boolean generating) {
+        this.generating = generating;
+        return this;
+    }
+
+    @Override
+    public MessagePayload build() {
+        MessagePayload payload = encodeBase();
+        payload.setSearchableContent(text);
+        payload.setContent(streamId);
+        if(generating) {
+            //正在生成消息的类型是14，存储标记位为透传为4。
+            payload.setType(14);
+            payload.setPersistFlag(4);
+        } else {
+            //已经生成消息的类型是15，存储标记位为透传为3。
+            payload.setType(15);
+            payload.setPersistFlag(3);
+        }
+        return payload;
+    }
+}
@@ -2,6 +2,7 @@ package cn.wildfirechat.sdk;

 import cn.wildfirechat.common.ErrorCode;
 import cn.wildfirechat.messagecontentbuilder.RichNotificationContentBuilder;
+import cn.wildfirechat.messagecontentbuilder.StreamingTextMessageContentBuilder;
 import cn.wildfirechat.messagecontentbuilder.TextMessageContentBuilder;
 import cn.wildfirechat.pojos.*;
 import cn.wildfirechat.proto.ProtoConstants;
@@ -377,21 +378,6 @@ public class Main {
            System.exit(-1);
        }

-        result = RelationAdmin.sendFriendRequest("ff1", "ff2", "hello2", false);
-        if (result != null && result.getErrorCode() != ErrorCode.ERROR_CODE_SUCCESS) {
-            System.out.println("success");
-        } else {
-            System.out.println("failure");
-            System.exit(-1);
-        }
-
-        result = RelationAdmin.sendFriendRequest("ff1", "ff2", "hello3", true);
-        if (result != null && (result.getErrorCode() == ErrorCode.ERROR_CODE_SUCCESS || result.getErrorCode() == ErrorCode.ERROR_CODE_ALREADY_FRIENDS)) {
-            System.out.println("success");
-        } else {
-            System.out.println("send friend request success");
-        }
-
        IMResult<Void> updateFriendStatusResult = RelationAdmin.setUserFriend("ff1", "ff2", true, "{\"from\":1}");
        if (updateFriendStatusResult != null && updateFriendStatusResult.getErrorCode() == ErrorCode.ERROR_CODE_SUCCESS) {
            System.out.println("update friend status success");
@@ -847,6 +833,42 @@ public class Main {
        } else {
            System.out.println("failure");
        }
+
+        testStreamingText();
+    }
+
+    static void testStreamingText() throws Exception {
+        Conversation conversation = new Conversation();
+        conversation.setTarget("uygqmws2k");
+        conversation.setType(ProtoConstants.ConversationType.ConversationType_Private);
+
+        String fullText = "北京野火无限网络科技有限公司是成立于2019年底的一家科技创新企业，公司的主要目标是为广大企业和单位提供优质可控、私有部署的即时通讯和实时音视频能力，为社会信息化水平提高作出自己的贡献。\n" +
+            "\n" +
+            "野火IM是公司研发一套自主可控的即时通讯组件，具有全部私有化、功能齐全、协议稳定可靠、全平台支持、安全性高和支持国产化等技术特点。客户端分层设计，既可开箱即用，也可与现有系统深度融合。具有完善的服务端API和自定义消息功能，可以任意扩展功能。代码开源率高，方便二次开发和使用。支持多人实时音视频和会议功能，线上沟通更通畅。\n" +
+            "\n" +
+            "公司致力于开源项目，在Github上开源项目广受好评，其中Server项目有超过7.1K个Star，组织合计Star超过1万个。有大量的技术公司受益于我们的开源，为自己的产品添加了即时通讯能力，这也算是我们公司为社会信息化建设做出的一点点贡献吧。\n" +
+            "\n" +
+            "公司以即时通讯技术为核心，持续努力优化和完善即时通讯和实时音视频产品，努力为客户提供最优质的即时通讯和实时音视频能力。";
+        int i = 0;
+        String streamId = UUID.randomUUID().toString();
+        while (i < fullText.length()) {
+            i+= 15;
+
+            boolean finish = i >= fullText.length();
+            String partText = finish?fullText:fullText.substring(0, i);
+
+            MessagePayload payload = StreamingTextMessageContentBuilder.newBuilder(streamId).text(partText).generating(!finish).build();
+
+            IMResult<SendMessageResult> resultSendMessage = MessageAdmin.sendMessage("admin", conversation, payload, null);
+            if (resultSendMessage != null && resultSendMessage.getErrorCode() == ErrorCode.ERROR_CODE_SUCCESS) {
+                System.out.println("send message success");
+            } else {
+                System.out.println("send message failure");
+                System.exit(-1);
+            }
+
+            Thread.sleep(500);
+        }
    }

    //***********************************************