Go语言中性能测试的度量标准

import (
    "context"
    "fmt"
    "net/http"
    "sync/atomic"
    "testing"
    "time"
)

func TestPerformance(t *testing.T) {
    // 计数器
    var totalRequests, totalTPS, totalRT int64
    var maxConcurrency int32

    // 创建HTTP服务器
    server := http.Server{
        Addr:    ":8080",
        Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            // 处理请求
            time.Sleep(time.Millisecond * 100)
            w.Write([]byte("Hello, world!"))
        }),
    }

    // 启动HTTP服务器
    go server.ListenAndServe()

    // 启动性能测试
    for i := 0; i < 10000; i++ {
        go func() {
            // 发起HTTP请求
            resp, err := http.Get("http://localhost:8080")
            if err != nil {
                return
            }
            resp.Body.Close()

            // 更新计数器
            atomic.AddInt64(&totalRequests, 1)
            atomic.AddInt64(&totalRT, time.Since(time.Now()).Nanoseconds())

            if currentConcurrency := atomic.AddInt32(&maxConcurrency, 1); currentConcurrency > maxConcurrency {
                maxConcurrency = currentConcurrency
            }
            atomic.AddInt32(&maxConcurrency, -1)
        }()
    }

    // 停止性能测试
    time.Sleep(time.Second * 10)
    server.Shutdown(context.Background())

    // 计算度量标准
    averageRT := float64(totalRT) / float64(totalRequests) / 1000000.0
    averageTPS := float64(totalRequests) / float64(time.Second * 10)

    // 打印结果
    fmt.Printf("Total requests: %d\n", totalRequests)
    fmt.Printf("Average response time: %.2f ms\n", averageRT)
    fmt.Printf("Average TPS: %.2f\n", averageTPS)
    fmt.Printf("Maximum concurrency: %d\n", maxConcurrency)
}