1010import org .springframework .data .domain .Page ;
1111import org .springframework .data .domain .PageRequest ;
1212import org .springframework .data .domain .Pageable ;
13+ import org .springframework .data .domain .Sort ;
1314import org .springframework .stereotype .Service ;
1415
1516import java .util .List ;
@@ -34,22 +35,26 @@ public String getDefaultImage(String image){
3435 }
3536
3637 // 메인페이지 보기
37- public MainPageResponseDto mainPageView (int sort ){
38+ public MainPageResponseDto mainPageView (int sort , int size ){
3839 // 사용자 정보 불러오기
3940 Member member = memberService .getCurrentMemberInfo ();
4041 String imageUrl = member .getProfileImageUrl ();
4142 String profileImageUrl = getDefaultImage (imageUrl );
4243
4344 // 페이지 (글 12개) 불러오기
44- List <Post > posts ;
45+ Page <Post > posts ;
46+ Sort sortOption ;
4547
4648 if (sort == 0 ) { // 최신순 정렬
47- posts = postRepository . findTop12ByOrderByCreatedAtDesc ( );
49+ sortOption = Sort . by ( Sort . Direction . DESC , "createdAt" );
4850 } else if (sort == 1 ) { // 인기순 정렬
49- posts = postRepository . findTop12ByOrderByPostLikesDesc ( );
51+ sortOption = Sort . by ( Sort . Direction . DESC , "postLikes" );
5052 } else
5153 throw new CustomException (ErrorCode .SORT_NOT_EXIST );
5254
55+ Pageable pageable = PageRequest .of (0 , size , sortOption );
56+ posts = postRepository .findAll (pageable );
57+
5358 List <MainPagePostResponseDto > postList = posts .stream ()
5459 .map (post -> new MainPagePostResponseDto (
5560 post .getPostTitle (),
@@ -68,14 +73,14 @@ public MainPageResponseDto mainPageView(int sort){
6873 }
6974
7075 // 메인페이지 전체보기
71- public MainPageResponseDto mainPageFullView (int sort , int page ){
76+ public MainPageResponseDto mainPageFullView (int sort , int page , int size ){
7277 // 사용자 정보 불러오기
7378 Member member = memberService .getCurrentMemberInfo ();
7479 String imageUrl = member .getProfileImageUrl ();
7580 String profileImageUrl = getDefaultImage (imageUrl );
7681
7782 // 페이지 설정 (한 페이지당 12개)
78- Pageable pageable = PageRequest .of (page - 1 , 12 );
83+ Pageable pageable = PageRequest .of (page - 1 , size );
7984 Page <Post > postPage ;
8085
8186 if (sort == 0 ) { // 최신순 정렬
0 commit comments