// 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")
}