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

时间:2018-11-15 04:11:05

标签: json go post load-testing vegeta

所以我看过几篇文章说应该将目标放在临时文件中,将正文放在.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)

}

1 个答案:

答案 0 :(得分:0)

您需要做一些事情,以使攻击者能够以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"},
}