diff --git a/src/main/java/com/example/ReviewZIP/domain/post/PostsService.java b/src/main/java/com/example/ReviewZIP/domain/post/PostsService.java index cbd66770..1d7b80e7 100644 --- a/src/main/java/com/example/ReviewZIP/domain/post/PostsService.java +++ b/src/main/java/com/example/ReviewZIP/domain/post/PostsService.java @@ -1,9 +1,9 @@ package com.example.ReviewZIP.domain.post; import com.example.ReviewZIP.domain.follow.Follows; -import com.example.ReviewZIP.domain.follow.FollowsRepository; import com.example.ReviewZIP.domain.image.Images; import com.example.ReviewZIP.domain.image.ImagesRepository; +import com.example.ReviewZIP.domain.image.dto.response.ImageResponseDto; import com.example.ReviewZIP.domain.post.dto.request.PostRequestDto; import com.example.ReviewZIP.domain.post.dto.response.PostResponseDto; import com.example.ReviewZIP.domain.postHashtag.PostHashtags; @@ -19,7 +19,6 @@ import com.example.ReviewZIP.global.response.exception.handler.*; import lombok.RequiredArgsConstructor; import org.springframework.data.domain.Page; -import org.springframework.data.domain.PageImpl; import org.springframework.data.domain.PageRequest; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; @@ -65,12 +64,12 @@ public Posts createPost(PostRequestDto postRequestDto) { Posts savedPost = postsRepository.save(newPost); - for (Long imageId : postRequestDto.getImageIds()) { - Images image = imagesRepository.findById(imageId).orElseThrow(() -> new ImagesHandler(ErrorStatus.IMAGE_NOT_FOUND)); - image.setPost(savedPost); - image.setUser(user); - imagesRepository.save(image); - savedPost.getPostImageList().add(image); + for (ImageResponseDto.ImageDto image : postRequestDto.getImageList()) { + Images newImage = imagesRepository.findById(image.getImageId()).orElseThrow(() -> new ImagesHandler(ErrorStatus.IMAGE_NOT_FOUND)); + newImage.setPost(savedPost); + newImage.setUser(user); + imagesRepository.save(newImage); + savedPost.getPostImageList().add(newImage); } return savedPost; } diff --git a/src/main/java/com/example/ReviewZIP/domain/post/dto/request/PostRequestDto.java b/src/main/java/com/example/ReviewZIP/domain/post/dto/request/PostRequestDto.java index 50147719..d1f9b2aa 100644 --- a/src/main/java/com/example/ReviewZIP/domain/post/dto/request/PostRequestDto.java +++ b/src/main/java/com/example/ReviewZIP/domain/post/dto/request/PostRequestDto.java @@ -1,5 +1,6 @@ package com.example.ReviewZIP.domain.post.dto.request; +import com.example.ReviewZIP.domain.image.dto.response.ImageResponseDto; import lombok.Getter; import java.util.ArrayList; @@ -10,5 +11,5 @@ public class PostRequestDto { private Long userId; private String comment; private Double point; - private List imageIds = new ArrayList<>(); + private List imageList = new ArrayList<>(); } \ No newline at end of file