使用自定義攻擊者在Vegeta中的POST請求中發送有效載荷

Question

所以我看過幾篇文章，說您應該將目標放在一個臨時文件中，將主體放在一個.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)

}

Answer 1

您只需要做幾件事即可使攻擊者能夠以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"},
}