From 090c7008023091c17c0c6494d35d3edc001535fd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9B=B9=E9=B9=8F=E9=A3=9E?= <rakor@qq.com>
Date: Mon, 30 Mar 2026 17:10:08 +0800
Subject: [PATCH] =?UTF-8?q?refactor(chat):=20=E4=BC=98=E5=8C=96=E8=81=8A?=
 =?UTF-8?q?=E5=A4=A9=E6=B6=88=E6=81=AFDTO=E7=9A=84=E9=99=84=E4=BB=B6?=
 =?UTF-8?q?=E5=92=8C=E5=9B=BE=E7=89=87=E5=88=97=E8=A1=A8=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 添加CollectionUtil工具类用于集合判空检查
- 添加Objects和Collectors工具类用于流式处理
- 重写getAttachments方法，过滤null值并返回处理后的列表
- 重写getImages方法，过滤null值并返回处理后的列表
- 提升数据访问的安全性和代码可读性
---
 .../common/pojo/dto/ChatMessageDTO.java           | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/dto/ChatMessageDTO.java b/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/dto/ChatMessageDTO.java
index 0ad044a8..469f7a3c 100644
--- a/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/dto/ChatMessageDTO.java
+++ b/nflg-mobilebroken-common/src/main/java/com/nflg/mobilebroken/common/pojo/dto/ChatMessageDTO.java
@@ -1,5 +1,6 @@
 package com.nflg.mobilebroken.common.pojo.dto;
 
+import cn.hutool.core.collection.CollectionUtil;
 import com.fasterxml.jackson.annotation.JsonIgnore;
 import com.nflg.mobilebroken.common.pojo.request.RemindUserRequest;
 import com.nflg.mobilebroken.common.pojo.vo.FileUploadVO;
@@ -8,6 +9,8 @@ import lombok.experimental.Accessors;
 
 import java.time.Instant;
 import java.util.List;
+import java.util.Objects;
+import java.util.stream.Collectors;
 
 @Data
 @Accessors(chain = true)
@@ -36,9 +39,21 @@ public class ChatMessageDTO {
     // 附件列表
     private List<FileUploadVO> attachments;
 
+    public List<FileUploadVO> getAttachments() {
+        return CollectionUtil.isEmpty(attachments)
+                ? null
+                : attachments.stream().filter(Objects::nonNull).collect(Collectors.toList());
+    }
+
     //图片列表
     private List<FileUploadVO> images;
 
+    public List<FileUploadVO> getImages() {
+        return CollectionUtil.isEmpty(images)
+                ? null
+                : images.stream().filter(Objects::nonNull).collect(Collectors.toList());
+    }
+
     // 创建时间
     private Instant createTime;