Fix for loop

netboxlabs · Sep 16, 2024 · d025fc8 · d025fc8
1 parent bde2e0e
commit d025fc8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/diode-server/reconciler/logs_retriever.go b/diode-server/reconciler/logs_retriever.go
@@ -88,8 +88,8 @@ func retrieveIngestionStatsSummary(ctx context.Context, client RedisClient) (*re
 
 	results := make([]*redis.Cmd, 0)
 	results = append(results, pipe.Do(ctx, "FT.SEARCH", "ingest-entity", "*", "LIMIT", 0, 0))
-	for i := 1; i < 5; i++ {
-		results = append(results, pipe.Do(ctx, "FT.SEARCH", "ingest-entity", fmt.Sprintf("@state:[%d %d]", i, i), "LIMIT", 0, 0))
+	for s := reconcilerpb.State_NEW; s <= reconcilerpb.State_NO_CHANGES; s++ {
+		results = append(results, pipe.Do(ctx, "FT.SEARCH", "ingest-entity", fmt.Sprintf("@state:[%d %d]", s, s), "LIMIT", 0, 0))
 	}
 
 	_, err := pipe.Exec(ctx)