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 attachments; + public List getAttachments() { + return CollectionUtil.isEmpty(attachments) + ? null + : attachments.stream().filter(Objects::nonNull).collect(Collectors.toList()); + } + //图片列表 private List images; + public List getImages() { + return CollectionUtil.isEmpty(images) + ? null + : images.stream().filter(Objects::nonNull).collect(Collectors.toList()); + } + // 创建时间 private Instant createTime;