Merge pull request #81 from kjiyun/alarm

kjiyun · web-flow · commit b25db5d8a2bb · 2025-02-11T16:30:10.000+09:00
#80 Feat: 실시간 알림 isOn 기능 추가
diff --git a/src/main/java/com/memesphere/domain/chartdata/repository/ChartDataRepository.java b/src/main/java/com/memesphere/domain/chartdata/repository/ChartDataRepository.java
@@ -2,12 +2,12 @@
 
 import com.memesphere.domain.chartdata.entity.ChartData;
 import com.memesphere.domain.memecoin.entity.MemeCoin;
+import org.springframework.data.domain.Pageable;
 import org.springframework.data.jpa.repository.EntityGraph;
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.data.jpa.repository.Query;
 import org.springframework.data.repository.query.Param;
 
-import java.awt.print.Pageable;
 import java.math.BigDecimal;
 import java.time.LocalDateTime;
 import java.util.List;
@@ -28,4 +28,6 @@ public interface ChartDataRepository extends JpaRepository<ChartData, Long> {
     LocalDateTime findRecordedTimeByCoinId1();
 
     List<ChartData> findByMemeCoinOrderByRecordedTimeDesc(MemeCoin memeCoin);
+    //TODO: 위아래 코드 합치는 방법 찾기
+    List<ChartData> findByMemeCoinOrderByRecordedTimeDesc(MemeCoin memeCoin, Pageable pageable);
 }
diff --git a/src/main/java/com/memesphere/domain/chat/service/ChatService.java b/src/main/java/com/memesphere/domain/chat/service/ChatService.java
@@ -82,7 +82,6 @@ public String postChatLike(Long chat_id, Long user_id) {
 
         // 사용자가 좋아요 눌렀는지 확인
         Optional<ChatLike> existingLike = chatLikeRepository.findByChatAndUser(chat, user);
-        System.out.println("좋아요" + existingLike.isPresent());
 
         if (existingLike.isPresent()) {
             chatLikeRepository.delete(existingLike.get());
diff --git a/src/main/java/com/memesphere/domain/notification/controller/CoinNotificationController.java b/src/main/java/com/memesphere/domain/notification/controller/CoinNotificationController.java
@@ -6,9 +6,11 @@
 import com.memesphere.domain.notification.dto.response.NotificationListResponse;
 import com.memesphere.domain.notification.dto.response.NotificationResponse;
 import com.memesphere.domain.notification.service.CoinNotificationService;
+import com.memesphere.global.jwt.CustomUserDetails;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import lombok.RequiredArgsConstructor;
+import org.springframework.security.core.annotation.AuthenticationPrincipal;
 import org.springframework.web.bind.annotation.*;
 
 @Tag(name="알림", description = "알림 관련 API")
@@ -65,8 +67,9 @@ public ApiResponse<NotificationListResponse> getNotificationList() {
                     - "is_rising": 상승(true) 또는 하락(false)
                     - "is_on": 알림 on(true) 또는 off(false)
                     ```""")
-    public ApiResponse<NotificationResponse> postNotification(@RequestBody NotificationRequest request) {
-        return ApiResponse.onSuccess(coinNotificationService.addNotification(request));
+    public ApiResponse<NotificationResponse> postNotification(@RequestBody NotificationRequest request,
+                                                              @AuthenticationPrincipal CustomUserDetails customUserDetails) {
+        return ApiResponse.onSuccess(coinNotificationService.addNotification(request, customUserDetails.getUser()));
     }
 
     @PatchMapping("/{notification-id}")
@@ -83,7 +86,7 @@ public ApiResponse<NotificationResponse> postNotification(@RequestBody Notificat
                     ```
                     알림 등록 API 응답 형식과 동일
                     ```""")
-    public ApiResponse<NotificationResponse> updateNotificationStatus(@PathVariable("notification-id") Long id) {
+    public ApiResponse<String> updateNotificationStatus(@PathVariable("notification-id") Long id) {
         return ApiResponse.onSuccess(coinNotificationService.modifyNotification(id));
     }
 
diff --git a/src/main/java/com/memesphere/domain/notification/converter/NotificationConverter.java b/src/main/java/com/memesphere/domain/notification/converter/NotificationConverter.java
@@ -5,17 +5,19 @@
 import com.memesphere.domain.notification.dto.response.NotificationListResponse;
 import com.memesphere.domain.notification.dto.response.NotificationResponse;
 import com.memesphere.domain.notification.entity.Notification;
+import com.memesphere.domain.user.entity.User;
 
 import java.util.List;
 
 public class NotificationConverter {
 
-    public static Notification toNotification(NotificationRequest notificationRequest, MemeCoin memeCoin) {
+    public static Notification toNotification(NotificationRequest notificationRequest, MemeCoin memeCoin, User user) {
         return Notification.builder()
                 .memeCoin(memeCoin)
                 .volatility(notificationRequest.getVolatility())
                 .stTime(notificationRequest.getStTime())
                 .isRising(notificationRequest.getIsRising())
+                .user(user)
                 .build();
     }
 
diff --git a/src/main/java/com/memesphere/domain/notification/dto/request/NotificationRequest.java b/src/main/java/com/memesphere/domain/notification/dto/request/NotificationRequest.java
@@ -22,4 +22,7 @@ public class NotificationRequest {
 
     @Schema(description = "상승 또는 하락", example = "True")
     private Boolean isRising;
+
+    @Schema(description = "알림 켜기/끄기", example = "True")
+    private Boolean isOn;
 }
diff --git a/src/main/java/com/memesphere/domain/notification/entity/Notification.java b/src/main/java/com/memesphere/domain/notification/entity/Notification.java
@@ -41,4 +41,8 @@ public class Notification extends BaseEntity {
     @ManyToOne(fetch=FetchType.LAZY)
     @JoinColumn(name="coin_id")
     private MemeCoin memeCoin;
+
+    public void updateIsOn(Boolean isOn) {
+        this.isOn = isOn;
+    }
 }
diff --git a/src/main/java/com/memesphere/domain/notification/repository/EmitterRepository.java b/src/main/java/com/memesphere/domain/notification/repository/EmitterRepository.java
@@ -1,5 +1,6 @@
 package com.memesphere.domain.notification.repository;
 
+import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
 import org.yaml.snakeyaml.emitter.Emitter;
 
@@ -11,4 +12,5 @@ public interface EmitterRepository {
     void deleteById(String emitterId);
     Map<String, SseEmitter> findAllEmitterStartWithByUserId(String UserId);
     Map<String, Object> findAllEventCacheStartWithByUserId(String UserId);
+    void saveEventCache(String emitterId, Object event);
 }
diff --git a/src/main/java/com/memesphere/domain/notification/repository/EmitterRepositoryImpl.java b/src/main/java/com/memesphere/domain/notification/repository/EmitterRepositoryImpl.java
@@ -39,4 +39,9 @@ public Map<String, Object> findAllEventCacheStartWithByUserId(String UserId) {
                 .filter(entry -> entry.getKey().startsWith(UserId))
                 .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
     }
+
+    @Override
+    public void saveEventCache(String emitterId, Object event) {
+        eventCache.put(emitterId, event);
+    }
 }
diff --git a/src/main/java/com/memesphere/domain/notification/service/CoinNotificationService.java b/src/main/java/com/memesphere/domain/notification/service/CoinNotificationService.java
@@ -3,10 +3,11 @@
 import com.memesphere.domain.notification.dto.request.NotificationRequest;
 import com.memesphere.domain.notification.dto.response.NotificationListResponse;
 import com.memesphere.domain.notification.dto.response.NotificationResponse;
+import com.memesphere.domain.user.entity.User;
 
 public interface CoinNotificationService {
     NotificationListResponse findNotificationList();
-    NotificationResponse addNotification(NotificationRequest notificationRequest);
-    NotificationResponse modifyNotification(Long notificationId);
+    NotificationResponse addNotification(NotificationRequest notificationRequest, User user);
+    String modifyNotification(Long notificationId);
     NotificationListResponse removeNotification(Long notificationId);
 }
diff --git a/src/main/java/com/memesphere/domain/notification/service/CoinNotificationServiceImpl.java b/src/main/java/com/memesphere/domain/notification/service/CoinNotificationServiceImpl.java
@@ -3,6 +3,7 @@
 import com.memesphere.domain.memecoin.entity.MemeCoin;
 import com.memesphere.domain.memecoin.repository.MemeCoinRepository;
 import com.memesphere.domain.notification.entity.Notification;
+import com.memesphere.domain.user.entity.User;
 import com.memesphere.global.apipayload.code.status.ErrorStatus;
 import com.memesphere.global.apipayload.exception.GeneralException;
 import com.memesphere.domain.notification.converter.NotificationConverter;
@@ -44,11 +45,11 @@ public NotificationListResponse findNotificationList() {
 
     // 알림 등록 API
     @Override
-    public NotificationResponse addNotification(NotificationRequest notificationRequest) {
+    public NotificationResponse addNotification(NotificationRequest notificationRequest, User user) {
         MemeCoin memeCoin = memeCoinRepository.findByName(notificationRequest.getName())
                 .orElseThrow(() -> new GeneralException(ErrorStatus.MEMECOIN_NOT_FOUND));
 
-        Notification notification = NotificationConverter.toNotification(notificationRequest, memeCoin);
+        Notification notification = NotificationConverter.toNotification(notificationRequest, memeCoin, user);
         notificationRepository.save(notification);
 
         NotificationResponse notificationResponse = NotificationConverter.toNotificationCreateResponse(notification, memeCoin);
@@ -57,8 +58,20 @@ public NotificationResponse addNotification(NotificationRequest notificationRequ
     }
 
     @Override
-    public NotificationResponse modifyNotification(Long notificationId) {
-        return null;
+    public String modifyNotification(Long notificationId) {
+
+        Notification existingNotification = notificationRepository.findById(notificationId)
+                .orElseThrow(() -> new GeneralException(ErrorStatus.NOTIFICATION_NOT_FOUND));
+
+        if (existingNotification.getIsOn()) { // 알림이 켜져있는 경우
+            existingNotification.updateIsOn(false);
+            notificationRepository.save(existingNotification);
+            return "알림을 껐습니다.";
+        } else { // 알림이 꺼져있는 경우
+            existingNotification.updateIsOn(true);
+            notificationRepository.save(existingNotification);
+            return "알림을 켰습니다.";
+        }
     }
 
     @Override
diff --git a/src/main/java/com/memesphere/domain/notification/service/PushNotificationService.java b/src/main/java/com/memesphere/domain/notification/service/PushNotificationService.java
@@ -1,8 +1,9 @@
 package com.memesphere.domain.notification.service;
 
+import com.memesphere.domain.notification.entity.Notification;
 import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
 
 public interface PushNotificationService {
     SseEmitter subscribe(Long userId, String lastEventId);
-    void send(Long userId);
+    void send(Notification notification, Long userId);
 }
diff --git a/src/main/java/com/memesphere/domain/notification/service/PushNotificationServiceImpl.java b/src/main/java/com/memesphere/domain/notification/service/PushNotificationServiceImpl.java
@@ -14,18 +14,16 @@
 import lombok.RequiredArgsConstructor;
 import lombok.extern.log4j.Log4j2;
 import org.springframework.data.domain.PageRequest;
+import org.springframework.data.domain.Pageable;
 import org.springframework.data.domain.Sort;
 import org.springframework.stereotype.Service;
 import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
 
-import java.awt.print.Pageable;
 import java.io.IOException;
 import java.math.BigDecimal;
+import java.math.RoundingMode;
 import java.time.LocalDateTime;
-import java.util.Comparator;
-import java.util.List;
-import java.util.Map;
-import java.util.Optional;
+import java.util.*;
 import java.util.stream.Collectors;
 
 @Log4j2
@@ -55,48 +53,32 @@ public SseEmitter subscribe(Long userId, String lastEventId) {
         // 최초 연결 더미데이터가 없으면 503 에러가 나므로 더미 데이터 생성
         sendToClient(emitter, emitterId, "EventStream Created. [userId=" + userId + "]");
 
+        // 클라이언트가 미수신한 Event 목록이 존재할 경우 전송하여 Event 유실을 예방
         if (!lastEventId.isEmpty()) {
             Map<String, Object> events = emitterRepository.findAllEventCacheStartWithByUserId(String.valueOf(userId));
             events.entrySet().stream()
                     .filter(entry -> lastEventId.compareTo(entry.getKey()) < 0)
                     .forEach(entry -> sendToClient(emitter, entry.getKey(), entry.getValue()));
         }
-        return emitter;
-    }
-
-    @Override
-    public void send(Long userId) {
 
         List<Notification> notifications = notificationRepository.findAllByUserId(userId); // 사용자가 등록한 알림 전부 가져오기
 
-        if (notifications.isEmpty()) {
-            return; // 저장된 알림이 없는 경우 아무것도 반환하지 않음
-        }
-
         // 변동성을 초과하는 알림 필터링
         List<Notification> filteredNotifications = notifications.stream()
                 .filter(notification -> isVolatilityExceeded(notification))
                 .collect(Collectors.toList());
-        System.out.println("알림:"+filteredNotifications.size());
-
-        if (filteredNotifications.isEmpty()) {
-            return; // 기준을 충족하는 변동성이 없으면 전송하지 않음
-        }
-
-        // 실시간 알림 전송
-        Map<String, SseEmitter> sseEmitters = emitterRepository.findAllEmitterStartWithByUserId(String.valueOf(userId));
 
-        sseEmitters.forEach((key, emitter) -> {
-            filteredNotifications.forEach(notification -> {
-                sendToClient(emitter, key, "변동성 초과 알림");
+        if (!filteredNotifications.isEmpty()) {
+            notifications.forEach(notification -> {
+                send(notification, userId);
             });
-        });
+        }
+        return emitter;
     }
 
     private void sendToClient(SseEmitter emitter, String emitterId, Object data) {
         try {
             if (emitter != null) {
-                System.out.println("-------");
                 emitter.send(SseEmitter.event()
                         .id(emitterId)
                         .data(data));
@@ -108,6 +90,18 @@ private void sendToClient(SseEmitter emitter, String emitterId, Object data) {
         }
     }
 
+    @Override
+    public void send(Notification notification, Long userId) {
+
+        // 실시간 알림 전송 - 로그인 한 유저의 SseEmitter 모두 가져오기
+        Map<String, SseEmitter> sseEmitters = emitterRepository.findAllEmitterStartWithByUserId(String.valueOf(userId));
+
+        sseEmitters.forEach((key, emitter) -> {
+                emitterRepository.saveEventCache(key, notification);
+                sendToClient(emitter, key, NotificationConverter.toNotificationCreateResponse(notification, notification.getMemeCoin()));
+        });
+    }
+
     private boolean isVolatilityExceeded(Notification notification) {
         MemeCoin memeCoin = notification.getMemeCoin();
         if (memeCoin == null) {
@@ -119,7 +113,25 @@ private boolean isVolatilityExceeded(Notification notification) {
             throw new GeneralException(ErrorStatus.CANNOT_LOAD_CHARTDATA);
         }
 
-        return true;
+        Integer count = notification.getStTime() / 10; //몇 번 가져올 것인지 결정
+        Pageable pageable = (Pageable) PageRequest.of(0, count, Sort.by(Sort.Direction.DESC, "createdAt"));
+        List<ChartData> lastNData = chartDataRepository.findByMemeCoinOrderByRecordedTimeDesc(memeCoin, pageable);
+
+        if (lastNData.size() < count) {
+            return false; // 비교할 데이터가 부족하면 알림을 보내지 않음
+        }
+
+        BigDecimal sum = lastNData.stream()
+                .map(ChartData::getPrice)
+                .reduce(BigDecimal.ZERO, BigDecimal::add);
+        BigDecimal average = sum.divide(BigDecimal.valueOf(lastNData.size()), 4, RoundingMode.HALF_UP);
+        BigDecimal definedVolatility = new BigDecimal(notification.getVolatility());
+
+        if (notification.getIsRising()) { // 상승인 경우
+            return average.compareTo(definedVolatility) > 0;
+        } else {
+            return average.compareTo(definedVolatility) < 0;
+        }
     }
 
 }
diff --git a/src/main/java/com/memesphere/global/apipayload/code/status/ErrorStatus.java b/src/main/java/com/memesphere/global/apipayload/code/status/ErrorStatus.java
@@ -33,6 +33,7 @@ public enum ErrorStatus implements BaseCode {
     // notification 에러
     CANNOT_CHECK_VOLATILITY(HttpStatus.NOT_FOUND, "CANNOT CHECK VOLATILITY", "변동성을 확인할 수 없습니다."),
     CANNOT_PUSH_NOTIFICATION(HttpStatus.BAD_REQUEST, "CANNOT PUSH NOTIFICATION", "알림 전송을 실패했습니다."),
+    NOTIFICATION_NOT_FOUND(HttpStatus.NOT_FOUND, "NOTIFICATION_NOT_FOUND", "알림을 찾을 수 없습니다."),
 
     // 유저 로그인 에러
     USER_NOT_FOUND(HttpStatus.NOT_FOUND, "USER NOT FOUND", "유저를 찾을 수 없습니다."),

Original file line number	Diff line number	Diff line change
`@@ -22,4 +22,7 @@ public class NotificationRequest {`
`22`	`22`
`23`	`23`	`@Schema(description = "상승 또는 하락", example = "True")`
`24`	`24`	`private Boolean isRising;`
	`25`	`+`
	`26`	`+ @Schema(description = "알림 켜기/끄기", example = "True")`
	`27`	`+ private Boolean isOn;`
`25`	`28`	`}`
Original file line number	Diff line number	Diff line change
`@@ -41,4 +41,8 @@ public class Notification extends BaseEntity {`
`41`	`41`	`@ManyToOne(fetch=FetchType.LAZY)`
`42`	`42`	`@JoinColumn(name="coin_id")`
`43`	`43`	`private MemeCoin memeCoin;`
	`44`	`+`
	`45`	`+ public void updateIsOn(Boolean isOn) {`
	`46`	`+ this.isOn = isOn;`
	`47`	`+ }`
`44`	`48`	`}`
Original file line number	Diff line number	Diff line change
`@@ -39,4 +39,9 @@ public Map<String, Object> findAllEventCacheStartWithByUserId(String UserId) {`
`39`	`39`	`.filter(entry -> entry.getKey().startsWith(UserId))`
`40`	`40`	`.collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));`
`41`	`41`	`}`
	`42`	`+`
	`43`	`+ @Override`
	`44`	`+ public void saveEventCache(String emitterId, Object event) {`
	`45`	`+ eventCache.put(emitterId, event);`
	`46`	`+ }`
`42`	`47`	`}`