Guard avg monthly and yearly messages

f-ewald · May 28, 2022 · 43cfcbd · 43cfcbd
1 parent 7208064
commit 43cfcbd
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pkg/db.go b/pkg/db.go
@@ -96,10 +96,14 @@ func (db *Database) Statistics(ctx context.Context) (stats *Statistics, err erro
 	stats.AvgDailyMessages = float64(stats.TotalMessages) / totalDays
 
 	totalMonths := monthsSince(stats.FirstMessage, stats.LastMessage)
-	stats.AvgMonthlyMessages = float64(stats.TotalMessages) / float64(totalMonths)
+	if totalMonths > 0 {
+		stats.AvgMonthlyMessages = float64(stats.TotalMessages) / float64(totalMonths)
+	}
 
 	totalYears := yearsSince(stats.FirstMessage, stats.LastMessage)
-	stats.AvgYearlyMessages = float64(stats.TotalMessages) / float64(totalYears)
+	if totalYears > 0 {
+		stats.AvgYearlyMessages = float64(stats.TotalMessages) / float64(totalYears)
+	}
 
 	return stats, nil
 }