Vegeta HTTP 负载测试工具

Go接口压测的第三方包，一个很好用的负载测试工具。vegeta测试工具组件（可执行文件）支持linux以及mac系统，这里指的是通过终端命令行进行进行测试，不需要从代码层面使用这个工具的时候支持linux以及mac系统，在windows上可能会出现问题。但是通过vegeta包代码执行的时候是可以在多平台执行的，但是在数据的展示上会有局限性，当然也可能是博主本人不会用没有找到对应的使用方式。
比如测试工具组件执行完命令后可以在终端控制台很直观的看到Histogram图，但是在代码层面测试后就很难分析出Histogram的情况
下面贴一下代码层面的吧，至于组件的方式大家百度一下有很多详细的教程

http的形式

传参，用户验证参数

1. package main
3. import (
4.         "encoding/base64"
5.         "encoding/json"
6.         "fmt"
7.         vegeta "github.com/tsenart/vegeta/lib"
8.         "net/http"
9.         "time"
10. )
12. func main() {
13.         // 测试目标 URL
14.         target := "https://your.domain.name/"
15.         // 测试参数
16.         rate := 20                  // 每秒请求数
17.         duration := 1 * time.Second // 测试时间
19.         // 请求体数据
20.         data := map[string]string{
21.                 "key1": "value1",
22.                 "key2": "value2",
23.         }
24.         body, _ := json.Marshal(&data)
26.         // Basic验证
27.         authString := "user:password"
28.         encodedAuthString := base64.StdEncoding.EncodeToString([]byte(authString))
29.         headers := http.Header{
30.                 "Content-Type":  []string{"application/json"},
31.                 "Authorization": []string{"Basic " + encodedAuthString},
32.         }
34.         // 使用自定义transport创建attacker
35.         attacker := vegeta.NewAttacker()
37.         // 创建 vegeta 目标
38.         targets := []vegeta.Target{{Method: "GET", URL: target, Header: headers, Body: body}}
40.         // 创建和执行 vegeta 测试
41.         results := attacker.Attack(vegeta.NewStaticTargeter(targets...), vegeta.Rate{Freq: rate, Per: time.Second}, duration, "Load Test")
42.         // 分析结果
43.         metrics := &vegeta.Metrics{}
44.         for res := range results {
45.                 metrics.Add(res)
46.         }
48.         metrics.Close()
49.         // 打印结果
50.         fmt.Printf("Requests: %d\n", metrics.Requests)
51.         fmt.Printf("成功率: %.2f%%\n", metrics.Success*100)
52.         fmt.Printf("最大延迟: %s\n", metrics.Latencies.Max)
53.         fmt.Printf("平均延迟: %s\n", metrics.Latencies.Mean)
54.         fmt.Printf("P95 延迟: %s\n", metrics.Latencies.P95)
55.         fmt.Printf("P99 延迟: %s\n", metrics.Latencies.P99)
56.         fmt.Printf("QPS: %.2f\n", metrics.Rate)
57. }

复制代码

Https的形式

传参，用户验证参数，跨过证书验证

1. package main
3. import (
4.         "crypto/tls"
5.         "encoding/base64"
6.         "encoding/json"
7.         "fmt"
8.         vegeta "github.com/tsenart/vegeta/lib"
9.         "net/http"
10.         "time"
11. )
13. func main() {
14.         // 测试目标 URL
15.         target := "https://your.domain.name/"
16.         // 测试参数
17.         rate := 20                  // 每秒请求数
18.         duration := 1 * time.Second // 测试时间
20.         // 请求体数据
21.         data := map[string]string{
22.                 "key1": "value1",
23.                 "key2": "value2",
24.         }
25.         body, _ := json.Marshal(&data)
27.         // Basic验证
28.         authString := "user:password"
29.         encodedAuthString := base64.StdEncoding.EncodeToString([]byte(authString))
30.         headers := http.Header{
31.                 "Content-Type":  []string{"application/json"},
32.                 "Authorization": []string{"Basic " + encodedAuthString},
33.         }
35.         // 创建http.Transport以跳过证书验证
36.         transport := &http.Transport{
37.                 TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
38.         }
40.         // 使用自定义transport创建attacker
41.         attacker := vegeta.NewAttacker(vegeta.Client(&http.Client{Transport: transport}))
43.         // 创建 vegeta 目标
44.         targets := []vegeta.Target{{Method: "GET", URL: target, Header: headers, Body: body}}
46.         // 创建和执行 vegeta 测试
47.         results := attacker.Attack(vegeta.NewStaticTargeter(targets...), vegeta.Rate{Freq: rate, Per: time.Second}, duration, "Load Test")
49.         // 分析结果
50.         metrics := &vegeta.Metrics{}
51.         for res := range results {
52.                 metrics.Add(res)
53.         }
55.         metrics.Close()
56.         // 打印结果
57.         fmt.Printf("Requests: %d\n", metrics.Requests)
58.         fmt.Printf("成功率: %.2f%%\n", metrics.Success*100)
59.         fmt.Printf("最大延迟: %s\n", metrics.Latencies.Max)
60.         fmt.Printf("平均延迟: %s\n", metrics.Latencies.Mean)
61.         fmt.Printf("P95 延迟: %s\n", metrics.Latencies.P95)
62.         fmt.Printf("P99 延迟: %s\n", metrics.Latencies.P99)
63.         fmt.Printf("QPS: %.2f\n", metrics.Rate)
64. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！