sdk添加注册消息的接口,可以允许sdk注册自定义消息

This commit is contained in:
heavyrian2012
2025-02-19 19:27:42 +08:00
parent ff75379779
commit d14dc74566
2 changed files with 15 additions and 5 deletions
File diff suppressed because one or more lines are too long
@@ -12,7 +12,7 @@ import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
public class MessageContentFactory {
private static final Map<Integer, Class<MessageContent>> contentClassMap = new ConcurrentHashMap<>();
private static final Map<Integer, Class<? extends MessageContent>> contentClassMap = new ConcurrentHashMap<>();
static {
registerAllMessageContent();
@@ -36,7 +36,7 @@ public class MessageContentFactory {
}
public static MessageContent decodeMessageContent(MessagePayload messagePayload) {
Class<MessageContent> cls = contentClassMap.get(messagePayload.getType());
Class<? extends MessageContent> cls = contentClassMap.get(messagePayload.getType());
MessageContent messageContent;
if(cls != null) {
try {
@@ -53,6 +53,11 @@ public class MessageContentFactory {
return messageContent;
}
public static void registerCustomMessageContent(Class<? extends MessageContent> cls) throws Exception {
MessageContent content = cls.newInstance();
contentClassMap.put(content.getContentType(), cls);
}
private static void registerAllMessageContent() {
try {
for (Class cls : ClassUtil.getAllAssignedClass(MessageContent.class)) {