Kharec/goyco
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: reject negative/nonsensical flag values with a clear error instead of letting slice/channel allocations panic

Browse Source
This commit is contained in:
Sandro CURY CAZZANIGA
2025-11-21 14:55:51 +01:00
parent 0dcd5fec51
commit c7f30070c0
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
cmd/goyco/commands/seed.go
View File

@@ -69,6 +69,16 @@ func seedDatabase(userRepo repositories.UserRepository, postRepo repositories.Po
return err return err
} }
if *numUsers < 0 {
return fmt.Errorf("invalid value for --users: %d (must be >= 0)", *numUsers)
}
if *numPosts <= 0 {
return fmt.Errorf("invalid value for --posts: %d (must be > 0)", *numPosts)
}
if *votesPerPost < 0 {
return fmt.Errorf("invalid value for --votes-per-post: %d (must be >= 0)", *votesPerPost)
}
if !IsJSONOutput() { if !IsJSONOutput() {
fmt.Println("Starting database seeding...") fmt.Println("Starting database seeding...")
} }
@@ -93,7 +103,7 @@ func seedDatabase(userRepo repositories.UserRepository, postRepo repositories.Po
processor := NewParallelProcessor() processor := NewParallelProcessor()
var progress *ProgressIndicator var progress *ProgressIndicator
if !IsJSONOutput() { if !IsJSONOutput() && *numUsers > 0 {
progress = NewProgressIndicator(*numUsers, "Creating users (parallel)") progress = NewProgressIndicator(*numUsers, "Creating users (parallel)")
} }
users, err := processor.CreateUsersInParallel(userRepo, *numUsers, progress) users, err := processor.CreateUsersInParallel(userRepo, *numUsers, progress)
@@ -106,7 +116,7 @@ func seedDatabase(userRepo repositories.UserRepository, postRepo repositories.Po
allUsers := append([]database.User{*seedUser}, users...) allUsers := append([]database.User{*seedUser}, users...)
if !IsJSONOutput() { if !IsJSONOutput() && *numPosts > 0 {
progress = NewProgressIndicator(*numPosts, "Creating posts (parallel)") progress = NewProgressIndicator(*numPosts, "Creating posts (parallel)")
} }
posts, err := processor.CreatePostsInParallel(postRepo, seedUser.ID, *numPosts, progress) posts, err := processor.CreatePostsInParallel(postRepo, seedUser.ID, *numPosts, progress)
@@ -117,7 +127,7 @@ func seedDatabase(userRepo repositories.UserRepository, postRepo repositories.Po
progress.Complete() progress.Complete()
} }
if !IsJSONOutput() { if !IsJSONOutput() && len(posts) > 0 {
progress = NewProgressIndicator(len(posts), "Creating votes (parallel)") progress = NewProgressIndicator(len(posts), "Creating votes (parallel)")
} }
votes, err := processor.CreateVotesInParallel(voteRepo, allUsers, posts, *votesPerPost, progress) votes, err := processor.CreateVotesInParallel(voteRepo, allUsers, posts, *votesPerPost, progress)
@@ -128,7 +138,7 @@ func seedDatabase(userRepo repositories.UserRepository, postRepo repositories.Po
progress.Complete() progress.Complete()
} }
if !IsJSONOutput() { if !IsJSONOutput() && len(posts) > 0 {
progress = NewProgressIndicator(len(posts), "Updating scores (parallel)") progress = NewProgressIndicator(len(posts), "Updating scores (parallel)")
} }
err = processor.UpdatePostScoresInParallel(postRepo, voteRepo, posts, progress) err = processor.UpdatePostScoresInParallel(postRepo, voteRepo, posts, progress)