diff --git a/src/main/java/io/github/xxyopen/novel/dto/resp/BookCommentRespDto.java b/src/main/java/io/github/xxyopen/novel/dto/resp/BookCommentRespDto.java index a101936..a632fdc 100644 --- a/src/main/java/io/github/xxyopen/novel/dto/resp/BookCommentRespDto.java +++ b/src/main/java/io/github/xxyopen/novel/dto/resp/BookCommentRespDto.java @@ -35,6 +35,8 @@ public class BookCommentRespDto { private Long commentUserId; + private String commentUserPhoto; + @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") private LocalDateTime commentTime; diff --git a/src/main/java/io/github/xxyopen/novel/service/impl/BookServiceImpl.java b/src/main/java/io/github/xxyopen/novel/service/impl/BookServiceImpl.java index 486835e..c67dd78 100644 --- a/src/main/java/io/github/xxyopen/novel/service/impl/BookServiceImpl.java +++ b/src/main/java/io/github/xxyopen/novel/service/impl/BookServiceImpl.java @@ -26,6 +26,7 @@ import java.security.NoSuchAlgorithmException; import java.security.SecureRandom; import java.time.LocalDateTime; import java.util.*; +import java.util.function.Function; import java.util.stream.Collectors; /** @@ -244,12 +245,13 @@ public class BookServiceImpl implements BookService { // 查询评论用户信息,并设置需要返回的评论用户名 List userIds = bookComments.stream().map(BookComment::getUserId).toList(); List userInfos = userDaoManager.listUsers(userIds); - Map userInfoMap = userInfos.stream().collect(Collectors.toMap(UserInfo::getId, UserInfo::getUsername)); + Map userInfoMap = userInfos.stream().collect(Collectors.toMap(UserInfo::getId, Function.identity())); List commentInfos = bookComments.stream() .map(v -> BookCommentRespDto.CommentInfo.builder() .id(v.getId()) .commentUserId(v.getUserId()) - .commentUser(userInfoMap.get(v.getUserId())) + .commentUser(userInfoMap.get(v.getUserId()).getUsername()) + .commentUserPhoto(userInfoMap.get(v.getUserId()).getUserPhoto()) .commentContent(v.getCommentContent()) .commentTime(v.getCreateTime()).build()).toList(); bookCommentRespDto.setComments(commentInfos);