feat: 대시보드 초기화 및 재계산 api 추가(#120)

dnd-side-project · Mar 15, 2024 · 6dbc09c · 6dbc09c
1 parent 7aba98b
commit 6dbc09c
Show file tree

Hide file tree

Showing 5 changed files with 44 additions and 2 deletions.
diff --git a/src/main/java/com/dnd/namuiwiki/config/EventConfiguration.java b/src/main/java/com/dnd/namuiwiki/config/EventConfiguration.java
@@ -1,8 +1,10 @@
 package com.dnd.namuiwiki.config;
 
+import com.dnd.namuiwiki.domain.dashboard.DashboardRepository;
 import com.dnd.namuiwiki.domain.dashboard.DashboardService;
 import com.dnd.namuiwiki.domain.statistic.StatisticsService;
 import com.dnd.namuiwiki.domain.survey.SurveyEventHandler;
+import com.dnd.namuiwiki.domain.survey.SurveyRepository;
 import lombok.RequiredArgsConstructor;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -13,9 +15,12 @@ public class EventConfiguration {
     private final StatisticsService statisticsService;
     private final DashboardService dashboardService;
 
+    private final SurveyRepository surveyRepository;
+    private final DashboardRepository dashboardRepository;
+
     @Bean
     public SurveyEventHandler surveyEventHandler() {
-        return new SurveyEventHandler(statisticsService, dashboardService);
+        return new SurveyEventHandler(statisticsService, dashboardService, surveyRepository, dashboardRepository);
     }
 
 }
diff --git a/src/main/java/com/dnd/namuiwiki/domain/survey/SurveyController.java b/src/main/java/com/dnd/namuiwiki/domain/survey/SurveyController.java
@@ -57,4 +57,11 @@ public ResponseEntity<?> getSurvey(@PathVariable("surveyId") String surveyId) {
         return ResponseDto.ok(response);
     }
 
+    @Operation(hidden = true)
+    @PostMapping("/reset/dashboard")
+    public ResponseEntity<?> resetDashboard(@RequestParam String pwd) {
+        surveyService.resetDashboard(pwd);
+        return ResponseDto.noContent();
+    }
+
 }
diff --git a/src/main/java/com/dnd/namuiwiki/domain/survey/SurveyEventHandler.java b/src/main/java/com/dnd/namuiwiki/domain/survey/SurveyEventHandler.java
@@ -1,8 +1,10 @@
 package com.dnd.namuiwiki.domain.survey;
 
+import com.dnd.namuiwiki.domain.dashboard.DashboardRepository;
 import com.dnd.namuiwiki.domain.dashboard.DashboardService;
 import com.dnd.namuiwiki.domain.statistic.StatisticsService;
 import com.dnd.namuiwiki.domain.survey.model.dto.CreateSurveySuccessEvent;
+import com.dnd.namuiwiki.domain.survey.model.dto.ResetDashboardEvent;
 import com.dnd.namuiwiki.domain.survey.model.entity.Survey;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -12,9 +14,13 @@
 @Slf4j
 @RequiredArgsConstructor
 public class SurveyEventHandler {
+
     private final StatisticsService statisticsService;
     private final DashboardService dashboardService;
 
+    private final SurveyRepository surveyRepository;
+    private final DashboardRepository dashboardRepository;
+
     @Async
     @EventListener
     public void handleSurveySuccessEvent(CreateSurveySuccessEvent event) {
@@ -25,4 +31,15 @@ public void handleSurveySuccessEvent(CreateSurveySuccessEvent event) {
         statisticsService.updateStatistics(survey);
     }
 
+    @Async
+    @EventListener
+    public void handleResetDashboardEvent(ResetDashboardEvent event) {
+        log.info("SurveyHandler.handleResetDashboardEvent");
+
+        dashboardRepository.deleteAll();
+        surveyRepository.findAll().forEach(dashboardService::updateStatistics);
+
+        log.info("SurveyHandler.handleResetDashboardEvent done");
+    }
+
 }
diff --git a/src/main/java/com/dnd/namuiwiki/domain/survey/SurveyService.java b/src/main/java/com/dnd/namuiwiki/domain/survey/SurveyService.java
@@ -11,12 +11,13 @@
 import com.dnd.namuiwiki.domain.survey.model.dto.AnswerDto;
 import com.dnd.namuiwiki.domain.survey.model.dto.CreateSurveyRequest;
 import com.dnd.namuiwiki.domain.survey.model.dto.CreateSurveyResponse;
+import com.dnd.namuiwiki.domain.survey.model.dto.CreateSurveySuccessEvent;
 import com.dnd.namuiwiki.domain.survey.model.dto.GetAnswersByQuestionResponse;
 import com.dnd.namuiwiki.domain.survey.model.dto.GetSurveyResponse;
 import com.dnd.namuiwiki.domain.survey.model.dto.ReceivedSurveyDto;
+import com.dnd.namuiwiki.domain.survey.model.dto.ResetDashboardEvent;
 import com.dnd.namuiwiki.domain.survey.model.dto.SentSurveyDto;
 import com.dnd.namuiwiki.domain.survey.model.dto.SingleAnswerWithSurveyDetailDto;
-import com.dnd.namuiwiki.domain.survey.model.dto.CreateSurveySuccessEvent;
 import com.dnd.namuiwiki.domain.survey.model.entity.Answer;
 import com.dnd.namuiwiki.domain.survey.model.entity.Survey;
 import com.dnd.namuiwiki.domain.survey.type.AnswerType;
@@ -221,4 +222,12 @@ public void setQuestionIdForSurveyAnswers(String pwd) {
         });
     }
 
+    public void resetDashboard(String pwd) {
+        if (!SETTING_PASSWORD.equals(pwd)) {
+            throw new ApplicationErrorException(ApplicationErrorType.NO_PERMISSION);
+        }
+
+        applicationEventPublisher.publishEvent(new ResetDashboardEvent());
+    }
+
 }
diff --git a/src/main/java/com/dnd/namuiwiki/domain/survey/model/dto/ResetDashboardEvent.java b/src/main/java/com/dnd/namuiwiki/domain/survey/model/dto/ResetDashboardEvent.java
@@ -0,0 +1,4 @@
+package com.dnd.namuiwiki.domain.survey.model.dto;
+
+public class ResetDashboardEvent {
+}