Fix potential stats panic (#463)

castai · Jan 29, 2025 · c468e37 · c468e37
1 parent 1fd2e9f
commit c468e37
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/cmd/agent/daemon/state/stats_pipeline.go b/cmd/agent/daemon/state/stats_pipeline.go
@@ -221,12 +221,16 @@ func getPSIStatsDiff(prev, curr *castaipb.PSIStats) *castaipb.PSIStats {
 	if prev == nil || curr == nil {
 		return nil
 	}
-	return &castaipb.PSIStats{
-		Some: &castaipb.PSIData{
+	res := &castaipb.PSIStats{}
+	if curr.Some != nil && prev.Some != nil {
+		res.Some = &castaipb.PSIData{
 			Total: curr.Some.Total - prev.Some.Total,
-		},
-		Full: &castaipb.PSIData{
+		}
+	}
+	if curr.Full != nil && prev.Full != nil {
+		res.Full = &castaipb.PSIData{
 			Total: curr.Full.Total - prev.Full.Total,
-		},
+		}
 	}
+	return res
 }