Makefile

@@ -2,7 +2,7 @@ cleandata:
 	rm users.csv sections.csv
 
 cleanbinary:
-	rm grading_data_generator
+	rm user_data_generator
 
 clean: cleandata cleanbinary
 

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) []User {
+func GenerateUserData(config Config) {
 	// Build a weighted list of statuses
 	weightedStatuses := make([]string, 0, 100)
 	for status, pct := range config.PercentStatus {
@@ -533,12 +533,6 @@ func GenerateUserData(config Config) []User {
 		}
 	}
 
-	return users
-
-}
-
-func SaveGeneratedUsers(users []User) {
-
 	f, err := os.Create("users.csv")
 	if err != nil {
 		log.Fatalf("failed to create users.csv: %v", err)
@@ -559,7 +553,7 @@ func SaveGeneratedUsers(users []User) {
 		log.Fatalf("csv flush error: %v", err)
 	}
 
-	fmt.Printf("Generated %d users to users.csv\n", len(users))
+	fmt.Printf("Generated %d users to users.csv\n", config.NumberOfUsers)
 }
 
 func GenerateSectionData(config Config) {
@@ -640,9 +634,7 @@ func main() {
 		log.Fatalf("percent_status values must total 100, got %d", total)
 	}
 
-	generated_users := GenerateUserData(config)
-	SaveGeneratedUsers(generated_users)
-
+	GenerateUserData(config)
 	GenerateSectionData(config)
 
 }