98 lines
4.7 KiB
Go
98 lines
4.7 KiB
Go
// Package golink 连接
|
|
package golink
|
|
|
|
import (
|
|
"math/rand"
|
|
"time"
|
|
|
|
"go_dreamfactory/stress/model"
|
|
)
|
|
|
|
// ReqListWeigh 接口加权压测
|
|
type ReqListWeigh struct {
|
|
list []Req
|
|
weighCount uint32 // 总权重
|
|
}
|
|
|
|
// Req req
|
|
type Req struct {
|
|
req *model.Request // 请求信息
|
|
weights uint32 // 权重，数字越大访问频率越高
|
|
}
|
|
|
|
// setWeighCount 设置权重
|
|
func (r *ReqListWeigh) setWeighCount() {
|
|
r.weighCount = 0
|
|
for _, value := range r.list {
|
|
r.weighCount = r.weighCount + value.weights
|
|
}
|
|
}
|
|
|
|
var (
|
|
clientWeigh *ReqListWeigh
|
|
r *rand.Rand
|
|
)
|
|
|
|
// 多接口压测示例
|
|
func init() {
|
|
// TODO::压测多个接口示例
|
|
// 需要压测的接口参数
|
|
clients := make([]Req, 0)
|
|
clients = append(clients, Req{req: &model.Request{
|
|
URL: "https://page.aliyun.com/delivery/plan/list", // 请求url
|
|
Form: "http", // 请求方式 示例参数:http/webSocket/tcp
|
|
Method: "POST", // 请求方法 示例参数:GET/POST/PUT
|
|
Headers: map[string]string{
|
|
"referer": "https://cn.aliyun.com/",
|
|
"cookie": "aliyun_choice=CN; JSESSIONID=J8866281-CKCFJ4BUZ7GDO9V89YBW1-KJ3J5V9K-GYUW7; maliyun_temporary_console0=1AbLByOMHeZe3G41KYd5WWZvrM%2BGErkaLcWfBbgveKA9ifboArprPASvFUUfhwHtt44qsDwVqMk8Wkdr1F5LccYk2mPCZJiXb0q%2Bllj5u3SQGQurtyPqnG489y%2FkoA%2FEvOwsXJTvXTFQPK%2BGJD4FJg%3D%3D; cna=L3Q5F8cHDGgCAXL3r8fEZtdU; isg=BFNThsmSCcgX-sUcc5Jo2s2T4tF9COfKYi8g9wVwr3KphHMmjdh3GrHFvPTqJD_C; l=eBaceXLnQGBjstRJBOfwPurza77OSIRAguPzaNbMiT5POw1B5WAlWZbqyNY6C3GVh6lwR37EODnaBeYBc3K-nxvOu9eFfGMmn",
|
|
}, // headers 头信息
|
|
Body: "adPlanQueryParam=%7B%22adZone%22%3A%7B%22positionList%22%3A%5B%7B%22positionId%22%3A83%7D%5D%7D%2C%22requestId%22%3A%2217958651-f205-44c7-ad5d-f8af92a6217a%22%7D", // 消息体
|
|
Verify: "statusCode", // 验证的方法 示例参数:statusCode、json
|
|
Timeout: 30 * time.Second, // 是否开启Debug模式
|
|
Debug: false, // 是否开启Debug模式
|
|
}, weights: 2})
|
|
|
|
clients = append(clients, Req{req: &model.Request{
|
|
URL: "https://page.aliyun.com/delivery/plan/list", // 请求url
|
|
Form: "http", // 请求方式 示例参数:http/webSocket/tcp
|
|
Method: "POST", // 请求方法 示例参数:GET/POST/PUT
|
|
Headers: map[string]string{
|
|
"referer": "https://cn.aliyun.com/",
|
|
"cookie": "aliyun_choice=CN; JSESSIONID=J8866281-CKCFJ4BUZ7GDO9V89YBW1-KJ3J5V9K-GYUW7; maliyun_temporary_console0=1AbLByOMHeZe3G41KYd5WWZvrM%2BGErkaLcWfBbgveKA9ifboArprPASvFUUfhwHtt44qsDwVqMk8Wkdr1F5LccYk2mPCZJiXb0q%2Bllj5u3SQGQurtyPqnG489y%2FkoA%2FEvOwsXJTvXTFQPK%2BGJD4FJg%3D%3D; cna=L3Q5F8cHDGgCAXL3r8fEZtdU; isg=BFNThsmSCcgX-sUcc5Jo2s2T4tF9COfKYi8g9wVwr3KphHMmjdh3GrHFvPTqJD_C; l=eBaceXLnQGBjstRJBOfwPurza77OSIRAguPzaNbMiT5POw1B5WAlWZbqyNY6C3GVh6lwR37EODnaBeYBc3K-nxvOu9eFfGMmn",
|
|
}, // headers 头信息
|
|
Body: "adPlanQueryParam=%7B%22adZone%22%3A%7B%22positionList%22%3A%5B%7B%22positionId%22%3A83%7D%5D%7D%2C%22requestId%22%3A%2217958651-f205-44c7-ad5d-f8af92a6217a%22%7D", // 消息体
|
|
Verify: "statusCode", // 验证的方法 示例参数:statusCode、json
|
|
Timeout: 30 * time.Second, // 是否开启Debug模式
|
|
Debug: false, // 是否开启Debug模式
|
|
}, weights: 1})
|
|
|
|
r = rand.New(rand.NewSource(time.Now().Unix()))
|
|
clientWeigh = &ReqListWeigh{
|
|
list: clients,
|
|
}
|
|
|
|
// TODO::注释下面一行代码
|
|
clientWeigh.list = nil
|
|
|
|
clientWeigh.setWeighCount()
|
|
}
|
|
|
|
// getRequest 获取请求
|
|
func getRequest(request *model.Request) *model.Request {
|
|
if clientWeigh == nil || clientWeigh.weighCount <= 0 {
|
|
return request
|
|
}
|
|
n := uint32(r.Int31n(int32(clientWeigh.weighCount)))
|
|
var (
|
|
count uint32
|
|
)
|
|
for _, value := range clientWeigh.list {
|
|
if count >= n {
|
|
// value.req.Print()
|
|
return value.req
|
|
}
|
|
count = count + value.weights
|
|
}
|
|
panic("getRequest err")
|
|
}
|