rollback study initialization if there is an error

MORE-Platform · Jun 21, 2023 · 41c8cd9 · 41c8cd9
1 parent 311e4bc
commit 41c8cd9
Showing 1 changed file with 17 additions and 5 deletions.
diff --git a/studymanager/src/main/java/io/redlink/more/studymanager/service/StudyService.java b/studymanager/src/main/java/io/redlink/more/studymanager/service/StudyService.java
@@ -80,16 +80,28 @@ public void setStatus(Long studyId, Study.Status status, User user) {
             throw BadRequestException.StateChange(study.getStudyState(), status);
         }
 
-        studyRepository.setStateById(studyId, status);
+        Study.Status oldState = study.getStudyState();
 
+        studyRepository.setStateById(studyId, status);
         studyRepository.getById(studyId).ifPresent(s -> {
-            interventionService.alignInterventionsWithStudyState(s);
-            participantService.alignParticipantsWithStudyState(s);
-            observationService.alignObservationsWithStudyState(s);
-            integrationService.alignIntegrationsWithStudyState(s);
+            try {
+                alignWithStudyState(s);
+            } catch (Exception e) {
+                //ROLLBACK
+                studyRepository.setStateById(studyId, oldState);
+                studyRepository.getById(studyId).ifPresent(this::alignWithStudyState);
+                throw new BadRequestException("Study cannot be initialized");
+            }
         });
     }
 
+    private void alignWithStudyState(Study s) {
+        interventionService.alignInterventionsWithStudyState(s);
+        participantService.alignParticipantsWithStudyState(s);
+        observationService.alignObservationsWithStudyState(s);
+        integrationService.alignIntegrationsWithStudyState(s);
+    }
+
     public Map<MoreUser, Set<StudyRole>> getACL(Long studyId) {
         return aclRepository.getACL(studyId);
     }