// attack validates the attack arguments, sets up the // required resources, launches the attack and writes the results func rapidAttack(rate uint64, duration time.Duration, targets *vegeta.Targets, ordering, output string, header http.Header, previousResults vegeta.Results) (vegeta.Results, error) { if rate == 0 { return nil, fmt.Errorf(errRatePrefix + "can't be zero") } if duration == 0 { return nil, fmt.Errorf(errDurationPrefix + "can't be zero") } targets.SetHeader(header) switch ordering { case "random": targets.Shuffle(time.Now().UnixNano()) case "sequential": break default: return nil, fmt.Errorf(errOrderingPrefix+"`%s` is invalid", ordering) } log.Printf("Vegeta is attacking %d targets in %s order for %s with %d requests/sec...\n", len(*targets), ordering, duration, rate) results := vegeta.Attack(*targets, rate, duration) log.Println("Done!") return append(previousResults, results...), nil }
// attack validates the attack arguments, sets up the // required resources, launches the attack and writes the results func attack(rate uint64, duration time.Duration, targetsf, ordering, output string, header http.Header) error { if rate == 0 { return fmt.Errorf(errRatePrefix + "can't be zero") } if duration == 0 { return fmt.Errorf(errDurationPrefix + "can't be zero") } in, err := file(targetsf, false) if err != nil { return fmt.Errorf(errTargetsFilePrefix+"(%s): %s", targetsf, err) } defer in.Close() targets, err := vegeta.NewTargetsFrom(in) if err != nil { return fmt.Errorf(errTargetsFilePrefix+"(%s): %s", targetsf, err) } targets.SetHeader(header) switch ordering { case "random": targets.Shuffle(time.Now().UnixNano()) case "sequential": break default: return fmt.Errorf(errOrderingPrefix+"`%s` is invalid", ordering) } out, err := file(output, true) if err != nil { return fmt.Errorf(errOutputFilePrefix+"(%s): %s", output, err) } defer out.Close() log.Printf("Vegeta is attacking %d targets in %s order for %s with %d requests/sec...\n", len(targets), ordering, duration, rate) results := vegeta.Attack(targets, rate, duration) log.Println("Done!") log.Printf("Writing results to '%s'...", output) if err := results.Encode(out); err != nil { return err } return nil }