49 lines
1.5 KiB
Java
49 lines
1.5 KiB
Java
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;
|
|
}
|
|
}
|