Fix stats 500 error on empty db (#128)

ducompsoc · Sep 5, 2024 · 85875e6 · 85875e6
1 parent 971014e
commit 85875e6
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 6 deletions.
diff --git a/client/src/components/admin/add-projects/AddProjectsStatsPanel.tsx b/client/src/components/admin/add-projects/AddProjectsStatsPanel.tsx
@@ -31,7 +31,6 @@ const AddProjectsStatsPanel = () => {
         <div className="flex flex-col justify-evenly w-full mt-8">
             <div className="flex justify-evenly basis-2/5">
                 <StatBlock name="Active Projects" value={stats.num} />
-                <StatBlock name="Average Votes" value={stats.avg_votes} />
                 <StatBlock name="Average Seen" value={stats.avg_seen} />
             </div>
         </div>

diff --git a/server/database/admin.go b/server/database/admin.go
@@ -43,7 +43,12 @@ func AggregateStats(db *mongo.Database) (*models.Stats, error) {
 	projCursor.Next(context.Background())
 	err = projCursor.Decode(&projAvgSeen)
 	if err != nil {
-		return nil, err
+		// This means no documents were found
+		if err.Error() == "EOF" {
+			projAvgSeen = AvgSeenAgg{AvgSeen: 0}
+		} else {
+			return nil, err
+		}
 	}
 
 	// Get the average judge seen using an aggregation pipeline
@@ -64,9 +69,12 @@ func AggregateStats(db *mongo.Database) (*models.Stats, error) {
 	var judgeAvgSeen AvgSeenAgg
 	judgeCursor.Next(context.Background())
 	err = judgeCursor.Decode(&judgeAvgSeen)
-	println(judgeAvgSeen.AvgSeen)
 	if err != nil {
-		return nil, err
+		if err.Error() == "EOF" {
+			judgeAvgSeen = AvgSeenAgg{AvgSeen: 0}
+		} else {
+			return nil, err
+		}
 	}
 
 	// Create the stats object

diff --git a/server/database/judge.go b/server/database/judge.go
@@ -112,7 +112,11 @@ func AggregateJudgeStats(db *mongo.Database) (*models.JudgeStats, error) {
 	cursor.Next(context.Background())
 	err = cursor.Decode(&stats)
 	if err != nil {
-		return nil, err
+		if err.Error() == "EOF" {
+			stats = models.JudgeStats{Num: 0, AvgSeen: 0, NumActive: 0}
+		} else {
+			return nil, err
+		}
 	}
 
 	// Set the total number of judges

diff --git a/server/database/project.go b/server/database/project.go
@@ -85,7 +85,11 @@ func AggregateProjectStats(db *mongo.Database) (*models.ProjectStats, error) {
 	cursor.Next(context.Background())
 	err = cursor.Decode(&stats)
 	if err != nil {
-		return nil, err
+		if err.Error() == "EOF" {
+			stats = models.ProjectStats{Num: 0, AvgSeen: 0, NumActive: 0}
+		} else {
+			return nil, err
+		}
 	}
 
 	// Set the total number of projects