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

有志者，事竟成！如果你在学习Golang，那么本文《使用自定义攻击者在 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.values 是 map[string][]string 的别名（不是 map[string]string），
需要一些调整。

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

理论要掌握，实操不能落！以上关于《使用自定义攻击者在 Vegeta 中的 POST 请求中发送有效负载》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！