diff --git a/src/main/java/com/lesso/neverland/album/dto/AlbumByTimeDto.java b/src/main/java/com/lesso/neverland/album/dto/AlbumByTimeDto.java index f08502a..c111ea2 100644 --- a/src/main/java/com/lesso/neverland/album/dto/AlbumByTimeDto.java +++ b/src/main/java/com/lesso/neverland/album/dto/AlbumByTimeDto.java @@ -2,6 +2,7 @@ import com.lesso.neverland.album.domain.Album; +import java.util.ArrayList; import java.util.List; public record AlbumByTimeDto(Long albumIdx, @@ -11,7 +12,18 @@ public record AlbumByTimeDto(Long albumIdx, String puzzleDate, Integer puzzlerCount, List puzzlerImageList) { + + public static AlbumByTimeDto from(Album album) { + List puzzlerImageList = new ArrayList<>(); + puzzlerImageList.add(album.getPuzzle().getUser().getProfile().getProfileImage()); + + puzzlerImageList.addAll( + album.getPuzzle().getPuzzleMembers().stream() + .map(puzzleMember -> puzzleMember.getUser().getProfile().getProfileImage()) + .limit(2) + .toList()); + return new AlbumByTimeDto( album.getAlbumIdx(), album.getPuzzle().getTitle(), @@ -19,8 +31,7 @@ public static AlbumByTimeDto from(Album album) { album.getAlbumImage(), album.getPuzzle().getPuzzleDate().toString(), album.getPuzzle().getPuzzleMembers().size(), - album.getPuzzle().getPuzzleMembers().stream() - .map(puzzleMember -> puzzleMember.getUser().getProfile().getProfileImage()).toList() + puzzlerImageList ); } }