50 lines
980 B
Go
50 lines
980 B
Go
package main
|
|
|
|
import (
|
|
"math"
|
|
"math/rand"
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
|
|
vegeta "github.com/tsenart/vegeta/lib"
|
|
)
|
|
|
|
func main() {
|
|
go func() {
|
|
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
waiting := rand.Intn(450) + 50
|
|
time.Sleep(time.Duration(waiting) * time.Millisecond)
|
|
w.Write([]byte("Hello World"))
|
|
})
|
|
|
|
err := http.ListenAndServe(":8080", nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}()
|
|
|
|
targeter := vegeta.NewStaticTargeter(vegeta.Target{
|
|
Method: "GET",
|
|
URL: "http://localhost:8080",
|
|
})
|
|
|
|
rate := vegeta.Rate{Freq: 500, Per: time.Second}
|
|
|
|
duration := 5 * time.Second
|
|
|
|
attacker := vegeta.NewAttacker()
|
|
|
|
vegeta.Proxy(func(request *http.Request) (*url.URL, error) {
|
|
return url.Parse("http://test-api.imfree.site:20001")
|
|
})(attacker)
|
|
|
|
result := attacker.Attack(targeter, rate, duration, "test")
|
|
|
|
maxNum := 0
|
|
for res := range result {
|
|
math.Max(float64(res.Latency.Milliseconds()), float64(maxNum))
|
|
}
|
|
println(maxNum)
|
|
}
|