✨ feat(im): 优化【消息引用】的功能，来自第一波 code review

YunaiV
1 parent ae36d1ad
Showing 2 changed files with 7 additions and 2 deletions
yudao-module-im/src/main/java/cn/iocoder/yudao/module/im/service/websocket/dto/message/QuoteMessage.java
yudao-module-im/src/main/java/cn/iocoder/yudao/module/im/util/ImMessageUtils.java
@@ -22,6 +22,11 @@ public class QuoteMessage {
     public static final String FIELD_NAME = "quote";
     /**
+     * QuoteMessage 里 messageId 字段名
+     */
+    public static final String FIELD_MESSAGE_ID = "messageId";
+
+    /**
      * 被引用消息编号
      */
     private Long messageId;
@@ -28,7 +28,7 @@ public class ImMessageUtils {
         if (!(quoteRaw instanceof Map)) {
             return null;
         }
-        return Convert.toLong(((Map<?, ?>) quoteRaw).get("messageId"));
+        return Convert.toLong(((Map<?, ?>) quoteRaw).get(QuoteMessage.FIELD_MESSAGE_ID));
     }
     /**
@@ -58,7 +58,7 @@ public class ImMessageUtils {
             return content;
         }
         Map<String, Object> map = JsonUtils.parseMap(content);
-        if (map == null || !map.containsKey(QuoteMessage.FIELD_NAME)) {
+        if (map == null) {
             return content;
         }
         map.remove(QuoteMessage.FIELD_NAME);