From c8261d7cdd53508ec63adf04d4bfb55bca5f8ba5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tom=C3=A1s=20Senart?= Date: Wed, 14 Aug 2013 18:04:20 +0200 Subject: [PATCH] Fail fast on output opening errors --- main.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/main.go b/main.go index f18afa3..a165da4 100644 --- a/main.go +++ b/main.go @@ -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 {