refactor(chat): 优化聊天消息DTO的附件和图片列表处理

- 添加CollectionUtil工具类用于集合判空检查
- 添加Objects和Collectors工具类用于流式处理
- 重写getAttachments方法,过滤null值并返回处理后的列表
- 重写getImages方法,过滤null值并返回处理后的列表
- 提升数据访问的安全性和代码可读性
This commit is contained in:
曹鹏飞 2026-03-30 17:10:08 +08:00
parent 25e15d7639
commit 090c700802
1 changed files with 15 additions and 0 deletions

View File

@ -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;