feat: add create alarm method

Orange-Co · Sep 11, 2024 · 258d72e · 258d72e
1 parent 3b29bb5
commit 258d72e
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/main/java/co/orange/ddanzi/service/AlarmService.java b/src/main/java/co/orange/ddanzi/service/AlarmService.java
@@ -7,6 +7,7 @@
 import co.orange.ddanzi.domain.user.Alarm;
 import co.orange.ddanzi.domain.user.User;
 import co.orange.ddanzi.domain.user.enums.FcmCase;
+import co.orange.ddanzi.dto.alarm.AlarmResponseDto;
 import co.orange.ddanzi.dto.alarm.MyAlarm;
 import co.orange.ddanzi.global.jwt.AuthUtils;
 import co.orange.ddanzi.repository.AlarmRepository;
@@ -49,8 +50,9 @@ public ApiResponse<?> getAlarms(){
             }
             else
                 myAlarm.orderId(order.getId());
+            myAlarmList.add(myAlarm.build());
         }
-        return ApiResponse.onSuccess(Success.GET_ALARM_LIST_SUCCESS,myAlarmList);
+        return ApiResponse.onSuccess(Success.GET_ALARM_LIST_SUCCESS, AlarmResponseDto.builder().alarmList(myAlarmList).build());
     }
 
     @Transactional
@@ -76,4 +78,14 @@ private String createAlarmTime(LocalDateTime createdAt){
         }
     }
 
+    public void createAlarm(User user, FcmCase alarmCase, Order order){
+        log.info("Create Alarm Record");
+        Alarm newAlarm = Alarm.builder()
+                .alarmCase(alarmCase)
+                .user(user)
+                .order(order)
+                .build();
+        alarmRepository.save(newAlarm);
+    }
+
 }