senaduka/old-vegeta
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fail fast on output opening errors

Browse Source
This commit is contained in:
Tomás Senart
2013-08-14 18:04:20 +02:00
parent fe12ab2e9b
commit c8261d7cdd
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
main.go
View File

@@ -60,10 +60,6 @@ func main() {
rep = NewTextReporter()
}
log.Printf("Vegeta is attacking %d targets in %s order for %s...\n", len(targets), *ordering, *duration)
attack(targets, *ordering, *rate, *duration, rep)
log.Println("Done!")
var out io.WriteCloser
switch *output {
case "stdout":
@@ -71,10 +67,15 @@ func main() {
default:
out, err = os.Create(*output)
if err != nil {
log.Printf("Couldn't open `%s` for writing report: %s", *output, err)
log.Fatalf("Couldn't open `%s` for writing report: %s", *output, err)
}
defer out.Close()
}
log.Printf("Vegeta is attacking %d targets in %s order for %s...\n", len(targets), *ordering, *duration)
attack(targets, *ordering, *rate, *duration, rep)
log.Println("Done!")
// Report results!
log.Printf("Writing report to '%s'...", *output)
if rep.Report(out) != nil {