chore: rowserrcheck

flipt-io · Jun 27, 2023 · 3bf7d97 · 3bf7d97
1 parent 076d480
commit 3bf7d97
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/internal/storage/sql/common/rollout.go b/internal/storage/sql/common/rollout.go
@@ -226,7 +226,12 @@ func (s *Store) ListRollouts(ctx context.Context, namespaceKey, flagKey string,
 			rollout := rolloutsById[rolloutId]
 			rollout.Rule = &flipt.Rollout_Segment{Segment: rule}
 		}
+
+		if err := rows.Err(); err != nil {
+			return results, err
+		}
 	}
+
 	// get all rules from rollout_percentage_rules table
 	if len(rolloutsByType[flipt.RolloutType_PERCENTAGE_ROLLOUT_TYPE]) > 0 {
 		allRuleIds := make([]string, 0, len(rolloutsByType[flipt.RolloutType_PERCENTAGE_ROLLOUT_TYPE]))
@@ -262,6 +267,10 @@ func (s *Store) ListRollouts(ctx context.Context, namespaceKey, flagKey string,
 			rollout := rolloutsById[rolloutId]
 			rollout.Rule = &flipt.Rollout_Percentage{Percentage: rule}
 		}
+
+		if err := rows.Err(); err != nil {
+			return results, err
+		}
 	}
 
 	var next *flipt.Rollout