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

Avoid using io.WriteCloser

Browse Source
This commit is contained in:
Tomás Senart
2013-08-16 19:13:49 +02:00
parent 6dc5d189f9
commit e296881ac2
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
main.go
View File

@@ -62,16 +62,17 @@ func main() {
rep = NewTextReporter() rep = NewTextReporter()
} }
var out io.WriteCloser var out io.Writer
switch *output { switch *output {
case "stdout": case "stdout":
out = os.Stdout out = os.Stdout
default: default:
out, err = os.Create(*output) file, err := os.Create(*output)
if err != nil { if err != nil {
log.Fatalf("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() defer file.Close()
out = file
} }
log.Printf("Vegeta is attacking %d targets in %s order for %s...\n", len(targets), *ordering, *duration) log.Printf("Vegeta is attacking %d targets in %s order for %s...\n", len(targets), *ordering, *duration)