在Golang项目里，RPC接口是服务间通信的常用方式，其性能表现直接关系系统的整体吞吐量，因此做好RPC接口性能测试是开发过程中不可或缺的环节。不同的业务场景对RPC接口的性能要求不同，需要选择合适的测试方法才能得到准确的测试结果。
使用Golang内置testing包做基准测试
Golang自带的testing包提供了基准测试功能，适合对单个RPC接口做小规模的性能验证，能够快速得到接口的耗时、吞吐量等基础指标。首先需要准备RPC客户端连接，然后在基准测试函数中调用目标RPC接口。
下面是一个简单的示例，假设我们有一个用户查询的RPC接口，服务地址为127.0.0.1:8080：
package rpctest
import (
"context"
"testing"
"time"
// 假设这里是你的RPC客户端生成包
"your_project/rpc/client"
)
// 初始化RPC客户端，实际项目中可以根据你的RPC框架调整
func initUserClient() *client.UserClient {
// 这里替换为你的RPC客户端初始化逻辑
return client.NewUserClient("127.0.0.1:8080")
}
func BenchmarkGetUserInfo(b *testing.B) {
cli := initUserClient()
ctx := context.Background()
// 重置计时器，排除客户端初始化的耗时
b.ResetTimer()
for i := 0; i < b.N; i++ {
// 调用RPC接口，这里替换为你的实际RPC方法
_, err := cli.GetUserInfo(ctx, &client.GetUserInfoReq{UserId: 10001})
if err != nil {
b.Errorf("RPC调用失败: %v", err)
}
}
}
运行基准测试的命令为go test -bench=BenchmarkGetUserInfo -benchtime=5s，其中-benchtime可以指定测试时长，测试结果会输出每轮操作的耗时、每秒可以执行的次数等指标。
结合第三方压测工具进行多并发测试
如果需要模拟多用户并发调用RPC接口的场景，内置的基准测试可能不够灵活，这时候可以结合第三方压测工具，比如vegeta、hey等，这些工具支持自定义并发数、请求总量、请求速率等参数，更贴近真实生产场景。
以vegeta为例，首先需要编写一个调用RPC接口的小程序，将RPC调用逻辑封装成HTTP接口或者直接作为vegeta的目标，这里以封装成简单调用逻辑为例：
package main
import (
"context"
"fmt"
"os"
"time"
"your_project/rpc/client"
)
func main() {
// 初始化RPC客户端
cli := client.NewUserClient("127.0.0.1:8080")
ctx := context.Background()
// 模拟单次RPC调用
start := time.Now()
_, err := cli.GetUserInfo(ctx, &client.GetUserInfoReq{UserId: 10001})
if err != nil {
fmt.Printf("调用失败: %vn", err)
os.Exit(1)
}
fmt.Printf("调用耗时: %vn", time.Since(start))
}
然后使用vegeta进行压测，指定并发数和请求总数，就可以得到不同并发下的RPC接口性能表现，包括延迟分布、成功率、吞吐量等详细数据。
自定义压测逻辑适配复杂场景
如果业务场景有特殊要求，比如需要模拟不同参数的请求、需要统计不同错误类型的占比，也可以自己编写压测逻辑，通过goroutine模拟并发，自定义统计逻辑。
下面是一个自定义压测的示例，模拟100个并发用户，每个用户调用100次RPC接口，统计总耗时、平均耗时、错误率：
package main
import (
"context"
"fmt"
"sync"
"time"
"your_project/rpc/client"
)
func main() {
concurrency := 100 // 并发数
reqPerUser := 100 // 每个用户请求次数
cli := client.NewUserClient("127.0.0.1:8080")
ctx := context.Background()
var wg sync.WaitGroup
// 统计变量
var totalReq int
var totalErr int
var totalCost time.Duration
// 互斥锁保护统计变量
var mu sync.Mutex
startTime := time.Now()
for i := 0; i < concurrency; i++ {
wg.Add(1)
go func(userID int) {
defer wg.Done()
for j := 0; j < reqPerUser; j++ {
reqStart := time.Now()
_, err := cli.GetUserInfo(ctx, &client.GetUserInfoReq{UserId: int64(userID*1000 + j)})
reqCost := time.Since(reqStart)
mu.Lock()
totalReq++
totalCost += reqCost
if err != nil {
totalErr++
}
mu.Unlock()
}
}(i)
}
wg.Wait()
totalTime := time.Since(startTime)
fmt.Printf("总请求数: %dn", totalReq)
fmt.Printf("总错误数: %d, 错误率: %.2f%%n", totalErr, float64(totalErr)/float64(totalReq)*100)
fmt.Printf("总耗时: %vn", totalTime)
fmt.Printf("平均每个请求耗时: %vn", totalCost/time.Duration(totalReq))
fmt.Printf("每秒处理请求数(QPS): %.2fn", float64(totalReq)/totalTime.Seconds())
}
测试注意事项
- 测试前要保证RPC服务端处于稳定状态，避免服务端本身的波动影响测试结果。
- 测试时要逐步增加并发数，不要一开始就使用过高的并发，避免压垮服务端导致结果失真。
- 多次测试取平均值，避免单次测试的偶然性影响结果准确性。
- 关注RPC接口的延迟分布，而不仅仅是平均延迟，长尾延迟往往对系统影响更大。
不同测试方法对比
下面是几种测试方法的适用场景对比：
| 测试方法 | 适用场景 | 优点 | 缺点 |
|---|---|---|---|
| 内置testing基准测试 | 单个接口快速验证、开发阶段调试 | 无需额外依赖、使用简单 | 并发模拟能力弱、场景不够灵活 |
| 第三方压测工具 | 模拟真实并发场景、生成详细压测报告 | 功能丰富、支持复杂压测场景 | 需要额外学习工具使用、需要适配接口 |
| 自定义压测逻辑 | 特殊业务场景、自定义统计需求 | 灵活度高、完全适配业务需求 | 需要自己编写代码、统计逻辑容易出错 |