From d0ebdd812df9f5354ab6ae3648ef4d022007dbcd Mon Sep 17 00:00:00 2001 From: Joshua Herring Date: Sun, 5 Apr 2026 20:34:21 -0400 Subject: [PATCH] Separate User generation from csv generation --- main.go | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/main.go b/main.go index 2e8b5f6..29d8b3c 100644 --- a/main.go +++ b/main.go @@ -509,7 +509,7 @@ type Config struct { var statuses = []string{"JD", "LLM", "MCL", "PHD", "SJD", "MLS", "SPEC", "MNR", "CERT"} -func GenerateUserData(config Config) { +func GenerateUserData(config Config) []User { // Build a weighted list of statuses weightedStatuses := make([]string, 0, 100) for status, pct := range config.PercentStatus { @@ -533,6 +533,12 @@ func GenerateUserData(config Config) { } } + return users + +} + +func SaveGeneratedUsers(users []User) { + f, err := os.Create("users.csv") if err != nil { log.Fatalf("failed to create users.csv: %v", err) @@ -553,7 +559,7 @@ func GenerateUserData(config Config) { log.Fatalf("csv flush error: %v", err) } - fmt.Printf("Generated %d users to users.csv\n", config.NumberOfUsers) + fmt.Printf("Generated %d users to users.csv\n", len(users)) } func GenerateSectionData(config Config) { @@ -634,7 +640,9 @@ func main() { log.Fatalf("percent_status values must total 100, got %d", total) } - GenerateUserData(config) + generated_users := GenerateUserData(config) + SaveGeneratedUsers(generated_users) + GenerateSectionData(config) }