使用自定义攻击者在Vegeta中发送POST请求中的有效负载



所以我看到几篇帖子说,你应该把目标放在一个临时文件中,把主体放在.json文件中,但我需要向我的网站发送大量随机数据,理想情况下,我不想不断地向这些文件写入新的随机数据,所以我想在一个文件中完成这一切。如果这是不可能的,并且我使用多个文件,请告诉我。

我现在所要做的就是向一个网页发送POST请求,这个网页只是一个包含4个输入的表单:标题、数字、音量和年份。我有以下代码,但现在它没有发送值。它正在发送一个有效负载,但它没有任何值。这意味着一个键[]和值"不断存储在我的后台映射中。有人知道它发送空白的原因吗?有人能告诉我该怎么修吗?

package main
import (
"encoding/json"
"fmt"
"time"
vegeta "github.com/tsenart/vegeta/lib"
)
func NewCustomTargeter() vegeta.Targeter {
return func(tgt *vegeta.Target) error {
if tgt == nil {
return vegeta.ErrNilTarget
}
tgt.Method = "POST"
tgt.URL = "http://localhost:8080/create.html"
payload := map[string]string{
"title":  "junk",
"number": "junk2",
"volume": "junk3",
"year":   "junk4",
}
body, _ := json.Marshal(payload)
tgt.Body = []byte(body)
return nil
}
}
func main() {
rate := vegeta.Rate{Freq: 100, Per: 2 * time.Second}
duration := 10 * time.Second
targeter := NewCustomTargeter()
attacker := vegeta.NewAttacker()
var metrics vegeta.Metrics
for res := range attacker.Attack(targeter, rate, duration, "Load Test") {
metrics.Add(res)
}
metrics.Close()
fmt.Printf("%+v  n", metrics)
}

您需要做一些事情,使攻击者能够以FormData格式发送数据。

首先,将Content-type标头值设置为application/x-www-form-urlencoded。您可能需要导入net/http包。

header := http.Header{}
header.Set("Content-type", "application/x-www-form-urlencoded")
tgt.Header = header

然后,将数据设置为url.Values格式。将其编码值传递到tgt.Body中。您还需要导入net/url包。

form := url.Values{}
form.Set("title", "junk")
form.Set("number", "junk2")
form.Set("volume", "junk3")
form.Set("year", "junk4")
tgt.Body = []byte(form.Encode())

其他注意事项

在数据准备上,我们还可以使用地图文字样式
但由于url.Valuesmap[string][]string(而非map[string]string(的别名,
可能需要进行一些调整。

form := url.Values{
"title":  []string{"junk"},
"number": []string{"junk2"},
"volume": []string{"junk3"},
"year":   []string{"junk4"},
}

最新更新