Save statistics control status

thewca · Sep 23, 2024 · 921ce2f · 921ce2f
1 parent 43d9870
commit 921ce2f
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 0 deletions.
diff --git a/server/src/main/java/org/worldcubeassociation/statistics/enums/StatisticsControlStatus.java b/server/src/main/java/org/worldcubeassociation/statistics/enums/StatisticsControlStatus.java
@@ -0,0 +1,7 @@
+package org.worldcubeassociation.statistics.enums;
+
+public enum StatisticsControlStatus {
+    STARTED,
+    COMPLETED,
+    FAILED
+}
diff --git a/server/src/main/java/org/worldcubeassociation/statistics/model/StatisticsControl.java b/server/src/main/java/org/worldcubeassociation/statistics/model/StatisticsControl.java
@@ -27,6 +27,8 @@ public class StatisticsControl extends BaseEntity {
 
     private String message;
 
+    private String status;
+
     @Column(name = "created_at")
     private LocalDateTime createdAt;
 

diff --git a/...src/main/java/org/worldcubeassociation/statistics/service/impl/StatisticsServiceImpl.java b/...src/main/java/org/worldcubeassociation/statistics/service/impl/StatisticsServiceImpl.java
@@ -33,6 +33,7 @@
 import org.worldcubeassociation.statistics.dto.StatisticsRequestDTO;
 import org.worldcubeassociation.statistics.dto.StatisticsResponseDTO;
 import org.worldcubeassociation.statistics.enums.DisplayModeEnum;
+import org.worldcubeassociation.statistics.enums.StatisticsControlStatus;
 import org.worldcubeassociation.statistics.exception.NotFoundException;
 import org.worldcubeassociation.statistics.model.Statistics;
 import org.worldcubeassociation.statistics.model.StatisticsControl;
@@ -180,6 +181,7 @@ private void resourcesToStatistics(List<Resource> resources) throws IOException
             var statisticsControl = new StatisticsControl();
             statisticsControl.setPath(resource.getFilename());
             statisticsControl.setCreatedAt(LocalDateTime.now());
+            statisticsControl.setStatus(StatisticsControlStatus.STARTED.name());
             statisticsControlRepository.save(statisticsControl);
 
             try {
@@ -190,10 +192,12 @@ private void resourcesToStatistics(List<Resource> resources) throws IOException
                 sqlToStatistics(request);
 
                 statisticsControl.setCompletedAt(LocalDateTime.now());
+                statisticsControl.setStatus(StatisticsControlStatus.COMPLETED.name());
                 statisticsControlRepository.save(statisticsControl);
             } catch (Exception e) {
                 log.error("Error while processing {}", resource.getFilename(), e);
                 statisticsControl.setMessage(StringUtils.abbreviate(e.getMessage(), 200));
+                statisticsControl.setStatus(StatisticsControlStatus.FAILED.name());
                 statisticsControlRepository.save(statisticsControl);
             }
         }

diff --git a/server/src/main/resources/schema.sql b/server/src/main/resources/schema.sql
@@ -14,6 +14,7 @@ create table if not exists statistics (
 
 create table if not exists statistics_control (
     path varchar(100),
+    status varchar(20) not null,
     message varchar(200),
     created_at datetime not null,
     completed_at datetime,