Refactor seeding

beatlabs · Oct 13, 2024 · d2dfa1b · d2dfa1b
1 parent 2726352
commit d2dfa1b
Showing 1 changed file with 17 additions and 13 deletions.
diff --git a/seed/seed.go b/seed/seed.go
@@ -97,19 +97,7 @@ func (s *Seeder) Seed(cfg *config.Config) error {
 		return err
 	}
 
-	sb := strings.Builder{}
-	for f, seeded := range seedMap {
-		if !seeded {
-			_, err := sb.WriteString(fmt.Sprintf("field %s not seeded", f.Name()))
-			if err != nil {
-				return err
-			}
-		}
-	}
-	if sb.Len() > 0 {
-		return errors.New(sb.String())
-	}
-	return nil
+	return evaluateSeedMap(seedMap)
 }
 
 func processSeedField(f *config.Field, seedMap map[*config.Field]bool) error {
@@ -278,3 +266,19 @@ func processFlags(infos []*flagInfo, flagSet *flag.FlagSet, seedMap map[*config.
 	}
 	return nil
 }
+
+func evaluateSeedMap(seedMap map[*config.Field]bool) error {
+	sb := strings.Builder{}
+	for f, seeded := range seedMap {
+		if !seeded {
+			_, err := sb.WriteString(fmt.Sprintf("field %s not seeded", f.Name()))
+			if err != nil {
+				return err
+			}
+		}
+	}
+	if sb.Len() > 0 {
+		return errors.New(sb.String())
+	}
+	return nil
+}