Feat: 사용자 좋아요 기능 추가

skyeong42 · skyeong42 · commit 55c772537492 · 2025-02-28T10:34:18.000+09:00
diff --git a/src/main/java/cmf/commitField/domain/chat/chatRoom/controller/ChatRoomController.java b/src/main/java/cmf/commitField/domain/chat/chatRoom/controller/ChatRoomController.java
@@ -115,6 +115,7 @@ public GlobalResponse<Object> getByUserRoomPartList(Pageable pageable) {
         }
     }
 
+    //채팅방 제목 수정
     @PutMapping("/room/update/{roomId}")
     @LoginCheck
     public GlobalResponse<Object> updateRoom(
@@ -183,6 +184,19 @@ public GlobalResponse<Object> getRoomUsers(
         }
     }
 
+    // 전체 리스트에서 좋아요 순으로 정렬
+    @GetMapping("/room/heart")
+    @LoginCheck
+    public GlobalResponse<Object> roomHeartSort(Pageable pageable) {
+        List<ChatRoomDto> roomList = chatRoomService.getRoomHeartSortList(pageable);
+        if (roomList.size() == 0) {
+            return GlobalResponse.error(ErrorCode.NO_ROOM);
+        } else {
+            return GlobalResponse.success(roomList);
+        }
+    }
+
+
 
 
 }
diff --git a/src/main/java/cmf/commitField/domain/chat/chatRoom/dto/ChatRoomDto.java b/src/main/java/cmf/commitField/domain/chat/chatRoom/dto/ChatRoomDto.java
@@ -1,5 +1,6 @@
 package cmf.commitField.domain.chat.chatRoom.dto;
 
+import cmf.commitField.domain.chat.chatRoom.entity.ChatRoom;
 import lombok.*;
 
 @Getter
@@ -15,4 +16,16 @@ public class ChatRoomDto {
     private Long currentUserCount;
 
     private Integer userCountMax;
+
+    private Integer heartCount;
+
+    public static ChatRoomDto of(ChatRoom chatRoom) {
+        return ChatRoomDto.builder()
+                .id(chatRoom.getId())
+                .title(chatRoom.getTitle())
+                .currentUserCount((long) chatRoom.getUserChatRooms().size())
+                .userCountMax(chatRoom.getUserCountMax())
+                .heartCount(chatRoom.getHearts().size())
+                .build();
+    }
 }
diff --git a/src/main/java/cmf/commitField/domain/chat/chatRoom/entity/ChatRoom.java b/src/main/java/cmf/commitField/domain/chat/chatRoom/entity/ChatRoom.java
@@ -2,6 +2,7 @@
 
 import cmf.commitField.domain.chat.chatMessage.entity.ChatMsg;
 import cmf.commitField.domain.chat.userChatRoom.entity.UserChatRoom;
+import cmf.commitField.domain.heart.entity.Heart;
 import cmf.commitField.domain.user.entity.User;
 import cmf.commitField.global.jpa.BaseEntity;
 import jakarta.persistence.*;
@@ -37,6 +38,9 @@ public class ChatRoom extends BaseEntity {
     @OneToMany(mappedBy = "chatRoom", fetch = FetchType.LAZY)
     private List<ChatMsg> chatMsgs;
 
+    @OneToMany(mappedBy = "chatRoom", fetch = FetchType.LAZY)
+    private List<Heart> hearts;
+
     @Override
     public String toString() {
         return "ChatRoom{" +
diff --git a/src/main/java/cmf/commitField/domain/chat/chatRoom/repository/ChatRoomRepository.java b/src/main/java/cmf/commitField/domain/chat/chatRoom/repository/ChatRoomRepository.java
@@ -27,4 +27,9 @@ public interface ChatRoomRepository extends JpaRepository<ChatRoom, Long> {
     @Query(value = "SELECT ROOM_CREATOR FROM chat_room WHERE ID = ?", nativeQuery = true)
     Optional<Long> findChatRoomIdByRoomId(@Param("roomId") Long roomId);
 
+    @Query("SELECT c FROM ChatRoom c LEFT JOIN Heart h ON c.id = h.chatRoom.id " +
+            "GROUP BY c.id " +
+            "ORDER BY COUNT(h.user.id) DESC")
+    Page<ChatRoom> findAllByOrderByHearts(Pageable pageable);
+
 }
diff --git a/src/main/java/cmf/commitField/domain/chat/chatRoom/service/ChatRoomService.java b/src/main/java/cmf/commitField/domain/chat/chatRoom/service/ChatRoomService.java
@@ -29,4 +29,8 @@ public interface ChatRoomService {
     void updateRoom(Long roomId, ChatRoomUpdateRequest chatRoomUpdateRequest, Long userId);
 
     List<ChatRoomUserDto> getRoomUsers(Long roomId, Long userId);
+
+    List<ChatRoomDto> getRoomHeartSortList(Pageable pageable);
+
+
 }
diff --git a/src/main/java/cmf/commitField/domain/chat/chatRoom/service/ChatRoomServiceImpl.java b/src/main/java/cmf/commitField/domain/chat/chatRoom/service/ChatRoomServiceImpl.java
@@ -26,6 +26,7 @@
 import java.util.List;
 import java.util.Objects;
 import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
 
 @Service
 @RequiredArgsConstructor
@@ -73,19 +74,9 @@ public void createRoom(ChatRoomRequest chatRoomRequest, Long userId) {
 
     // 방 조회 DTO 변환 메서드 추출
     private static List<ChatRoomDto> getChatRoomDtos(Page<ChatRoom> all) {
-        List<ChatRoomDto> chatRoomList = new ArrayList<>();
-
-        for (ChatRoom list : all) {
-            ChatRoomDto dto = ChatRoomDto.builder()
-                    .id(list.getId())
-                    .title(list.getTitle())
-                    .currentUserCount((long) list.getUserChatRooms().size())
-                    .userCountMax(list.getUserCountMax())
-                    .build();
-
-            chatRoomList.add(dto);
-        }
-        return chatRoomList;
+        return all.stream()
+                .map(ChatRoomDto::of)
+                .collect(Collectors.toList());
     }
 
 
@@ -250,4 +241,24 @@ public List<ChatRoomUserDto> getRoomUsers(Long roomId, Long userId) {
         return chatRoomUserDtos;
     }
 
+    // 좋아요 순으로 정렬 후 방 전체 조회
+    @Override
+    @Transactional(readOnly = true)
+    public List<ChatRoomDto> getRoomHeartSortList(Pageable pageable) {
+        Page<ChatRoom> all = chatRoomRepository.findAllByOrderByHearts(pageable);
+        List<ChatRoom> chatRooms = all.toList();
+        List<ChatRoomDto> chatRoomDtos = new ArrayList<>();
+        for (ChatRoom chatRoom : chatRooms) {
+            ChatRoomDto build = ChatRoomDto.builder()
+                    .id(chatRoom.getId())
+                    .title(chatRoom.getTitle())
+                    .userCountMax(chatRoom.getUserCountMax())
+                    .currentUserCount((long) chatRoom.getUserChatRooms().size())
+                    .heartCount(chatRoom.getHearts().size())
+                    .build();
+            chatRoomDtos.add(build);
+        }
+        return chatRoomDtos;
+    }
+
 }
diff --git a/src/main/java/cmf/commitField/domain/heart/controller/HeartController.java b/src/main/java/cmf/commitField/domain/heart/controller/HeartController.java
@@ -0,0 +1,49 @@
+package cmf.commitField.domain.heart.controller;
+
+import cmf.commitField.domain.heart.service.HeartService;
+import cmf.commitField.domain.user.entity.CustomOAuth2User;
+import cmf.commitField.global.globalDto.GlobalResponse;
+import cmf.commitField.global.security.LoginCheck;
+import lombok.RequiredArgsConstructor;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+import org.springframework.security.oauth2.client.authentication.OAuth2AuthenticationToken;
+import org.springframework.web.bind.annotation.*;
+
+@RestController
+@RequiredArgsConstructor
+@RequestMapping("/heart")
+public class HeartController {
+
+    private final HeartService heartService;
+
+    @PostMapping("/{roomId}")
+    @LoginCheck
+    public GlobalResponse<Object> heart(@PathVariable Long roomId) {
+        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+
+        if (authentication instanceof OAuth2AuthenticationToken) {
+            CustomOAuth2User principal = (CustomOAuth2User) authentication.getPrincipal();
+            Long userId = principal.getId(); // OAuth2 사용자 ID 추출
+            heartService.heart(userId, roomId);
+            return GlobalResponse.success("좋아요를 눌렀습니다.");
+        } else {
+            throw new IllegalArgumentException("로그인 후에 이용해 주세요.");
+        }
+    }
+
+    @DeleteMapping("/{roomId}")
+    @LoginCheck
+    public GlobalResponse<Object> heartDelete(@PathVariable Long roomId) {
+        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+
+        if (authentication instanceof OAuth2AuthenticationToken) {
+            CustomOAuth2User principal = (CustomOAuth2User) authentication.getPrincipal();
+            Long userId = principal.getId(); // OAuth2 사용자 ID 추출
+            heartService.heartDelete(userId, roomId);
+            return GlobalResponse.success("좋아요를 취소했습니다.");
+        } else {
+            throw new IllegalArgumentException("로그인 후에 이용해 주세요.");
+        }
+    }
+}
diff --git a/src/main/java/cmf/commitField/domain/heart/entity/Heart.java b/src/main/java/cmf/commitField/domain/heart/entity/Heart.java
@@ -0,0 +1,28 @@
+package cmf.commitField.domain.heart.entity;
+
+import cmf.commitField.domain.chat.chatRoom.entity.ChatRoom;
+import cmf.commitField.domain.user.entity.User;
+import cmf.commitField.global.jpa.BaseEntity;
+import jakarta.persistence.Entity;
+import jakarta.persistence.FetchType;
+import jakarta.persistence.JoinColumn;
+import jakarta.persistence.ManyToOne;
+import lombok.*;
+import lombok.experimental.SuperBuilder;
+
+@Entity
+@SuperBuilder
+@Getter
+@Setter
+@NoArgsConstructor(access = AccessLevel.PROTECTED)
+@AllArgsConstructor(access = AccessLevel.PROTECTED)
+public class Heart extends BaseEntity {
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "user_id")
+    private User user;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "chat_room_id")
+    private ChatRoom chatRoom;
+}
diff --git a/src/main/java/cmf/commitField/domain/heart/repository/HeartRepository.java b/src/main/java/cmf/commitField/domain/heart/repository/HeartRepository.java
@@ -0,0 +1,12 @@
+package cmf.commitField.domain.heart.repository;
+
+import cmf.commitField.domain.heart.entity.Heart;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+import java.util.Optional;
+
+@Repository
+public interface HeartRepository extends JpaRepository<Heart, Long> {
+    Optional<Heart> findByUserIdAndChatRoomId(Long userId, Long chatRoomId);
+}
diff --git a/src/main/java/cmf/commitField/domain/heart/service/HeartService.java b/src/main/java/cmf/commitField/domain/heart/service/HeartService.java
@@ -0,0 +1,8 @@
+package cmf.commitField.domain.heart.service;
+
+public interface HeartService {
+
+    void heart(Long userId, Long roomId);
+
+    void heartDelete(Long userId, Long roomId);
+}
diff --git a/src/main/java/cmf/commitField/domain/heart/service/HeartServiceImpl.java b/src/main/java/cmf/commitField/domain/heart/service/HeartServiceImpl.java
@@ -0,0 +1,69 @@
+package cmf.commitField.domain.heart.service;
+
+import cmf.commitField.domain.chat.chatRoom.entity.ChatRoom;
+import cmf.commitField.domain.chat.chatRoom.repository.ChatRoomRepository;
+import cmf.commitField.domain.heart.entity.Heart;
+import cmf.commitField.domain.heart.repository.HeartRepository;
+import cmf.commitField.domain.user.entity.User;
+import cmf.commitField.domain.user.repository.UserRepository;
+import cmf.commitField.global.error.ErrorCode;
+import cmf.commitField.global.exception.CustomException;
+import lombok.RequiredArgsConstructor;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.Optional;
+
+@Service
+@RequiredArgsConstructor
+public class HeartServiceImpl implements HeartService {
+    private final UserRepository userRepository;
+    private final ChatRoomRepository chatRoomRepository;
+    private final HeartRepository heartRepository;
+
+    @Override
+    public void heart(Long userId, Long roomId) {
+        Optional<User> user = userRepository
+                .findById(userId);
+        Optional<ChatRoom> room = chatRoomRepository
+                .findById(roomId);
+        if (user.isEmpty()) {
+            throw new CustomException(ErrorCode.MEMBER_NOT_FOUND);
+        }
+        if (room.isEmpty()) {
+            throw new CustomException(ErrorCode.NOT_FOUND_ROOM);
+        }
+        Optional<Heart> userToRoomHeart = heartRepository
+                .findByUserIdAndChatRoomId(userId, roomId);
+        if (userToRoomHeart.isPresent()) {
+            throw new CustomException(ErrorCode.ALREADY_HEART_TO_ROOM);
+        }
+        Heart build = Heart.builder()
+                .user(user.get())
+                .chatRoom(room.get())
+                .build();
+        heartRepository.save(build);
+    }
+
+    @Override
+    @Transactional
+    public void heartDelete(Long userId, Long roomId) {
+        Optional<User> user = userRepository
+                .findById(userId);
+        Optional<ChatRoom> room = chatRoomRepository
+                .findById(roomId);
+        if (user.isEmpty()) {
+            throw new CustomException(ErrorCode.MEMBER_NOT_FOUND);
+        }
+        if (room.isEmpty()) {
+            throw new CustomException(ErrorCode.NOT_FOUND_ROOM);
+        }
+        Optional<Heart> userToRoomHeart = heartRepository
+                .findByUserIdAndChatRoomId(userId, roomId);
+        if (userToRoomHeart.isEmpty()) {
+            throw new CustomException(ErrorCode.NOT_EXIST_ROOM_HEART);
+        }
+        Heart heart = userToRoomHeart.get();
+        heartRepository.delete(heart);
+    }
+}
diff --git a/src/main/java/cmf/commitField/domain/user/entity/User.java b/src/main/java/cmf/commitField/domain/user/entity/User.java
@@ -3,6 +3,7 @@
 import cmf.commitField.domain.chat.chatMessage.entity.ChatMsg;
 import cmf.commitField.domain.chat.chatRoom.entity.ChatRoom;
 import cmf.commitField.domain.chat.userChatRoom.entity.UserChatRoom;
+import cmf.commitField.domain.heart.entity.Heart;
 import cmf.commitField.domain.pet.entity.Pet;
 import cmf.commitField.global.jpa.BaseEntity;
 import jakarta.persistence.*;
@@ -49,6 +50,9 @@ public enum Role {
     @OneToMany(mappedBy = "user", fetch = FetchType.LAZY, cascade = CascadeType.PERSIST)
     private List<Pet> pets = new ArrayList<>();
 
+    @OneToMany(mappedBy = "user", fetch = FetchType.LAZY)
+    private List<Heart> hearts;
+
     public User(String username, String email, String nickname, String avatarUrl, Boolean status, List<ChatRoom> cr, List<UserChatRoom> ucr, List<ChatMsg> cmsg){
         this.username=username;
         this.email=email;
diff --git a/src/main/java/cmf/commitField/global/error/ErrorCode.java b/src/main/java/cmf/commitField/global/error/ErrorCode.java
@@ -62,7 +62,12 @@ public enum ErrorCode {
     CHAT_NOT_FOUND(HttpStatus.NOT_FOUND, "해당 채팅방의 메시지들을 찾지 못했습니다."),
 
     // Lock
-    FAILED_GET_LOCK(HttpStatus.LOCKED, "락을 획득하지 못했습니다.");
+    FAILED_GET_LOCK(HttpStatus.LOCKED, "락을 획득하지 못했습니다."),
+
+    //Heart
+    NOT_EXIST_ROOM_HEART(HttpStatus.BAD_REQUEST, "해당 채팅방에 좋아요가 눌러져 있지 않습니다."),
+    ALREADY_HEART_TO_ROOM(HttpStatus.BAD_REQUEST, "이미 해당 채팅방에 좋아요를 누르셨습니다.");
+
 
     private final HttpStatus httpStatus;
     private final String message;

Original file line number	Diff line number	Diff line change
`@@ -115,6 +115,7 @@ public GlobalResponse<Object> getByUserRoomPartList(Pageable pageable) {`
`115`	`115`	`}`
`116`	`116`	`}`
`117`	`117`
	`118`	`+ //채팅방 제목 수정`
`118`	`119`	`@PutMapping("/room/update/{roomId}")`
`119`	`120`	`@LoginCheck`
`120`	`121`	`public GlobalResponse<Object> updateRoom(`
`@@ -183,6 +184,19 @@ public GlobalResponse<Object> getRoomUsers(`
`183`	`184`	`}`
`184`	`185`	`}`
`185`	`186`
	`187`	`+ // 전체 리스트에서 좋아요 순으로 정렬`
	`188`	`+ @GetMapping("/room/heart")`
	`189`	`+ @LoginCheck`
	`190`	`+ public GlobalResponse<Object> roomHeartSort(Pageable pageable) {`
	`191`	`+ List<ChatRoomDto> roomList = chatRoomService.getRoomHeartSortList(pageable);`
	`192`	`+ if (roomList.size() == 0) {`
	`193`	`+ return GlobalResponse.error(ErrorCode.NO_ROOM);`
	`194`	`+ } else {`
	`195`	`+ return GlobalResponse.success(roomList);`
	`196`	`+ }`
	`197`	`+ }`
	`198`	`+`
	`199`	`+`
`186`	`200`
`187`	`201`
`188`	`202`	`}`
Original file line number	Diff line number	Diff line change
`@@ -27,4 +27,9 @@ public interface ChatRoomRepository extends JpaRepository<ChatRoom, Long> {`
`27`	`27`	`@Query(value = "SELECT ROOM_CREATOR FROM chat_room WHERE ID = ?", nativeQuery = true)`
`28`	`28`	`Optional<Long> findChatRoomIdByRoomId(@Param("roomId") Long roomId);`
`29`	`29`
	`30`	`+ @Query("SELECT c FROM ChatRoom c LEFT JOIN Heart h ON c.id = h.chatRoom.id " +`
	`31`	`+ "GROUP BY c.id " +`
	`32`	`+ "ORDER BY COUNT(h.user.id) DESC")`
	`33`	`+ Page<ChatRoom> findAllByOrderByHearts(Pageable pageable);`
	`34`	`+`
`30`	`35`	`}`
Original file line number	Diff line number	Diff line change
`@@ -29,4 +29,8 @@ public interface ChatRoomService {`
`29`	`29`	`void updateRoom(Long roomId, ChatRoomUpdateRequest chatRoomUpdateRequest, Long userId);`
`30`	`30`
`31`	`31`	`List<ChatRoomUserDto> getRoomUsers(Long roomId, Long userId);`
	`32`	`+`
	`33`	`+ List<ChatRoomDto> getRoomHeartSortList(Pageable pageable);`
	`34`	`+`
	`35`	`+`
`32`	`36`	`}`