【测试平台系列】第一章 手撸压力机（十一）-初步实现性能测试

上一章节我们组合了场景，它是一个list结构。今天我们实现性能测试计划的数据结构及其方法。
首先，我们在model目录新建test_plan.go文件：

// Package model -----------------------------
// @file      : test_plan.go
// @author    : 被测试耽误的大厨
// @contact   : 13383088061@163.com
// @time      : 2023/8/14 12:06
// -------------------------------------------
package model

import (
        "kitchen-engine/global/constant"
        "time"
)

type TestPlan struct {
        Id         string      `json:"id"`      // 唯一id
        Name       string      `json:"name"`    // 场景名称
        ItemId     string      `json:"item_id"` // 项目Id
        TeamId     string      `json:"team_id"` // 团队Id
        Task       Task        `json:"task"`       // 具体的任务
        TestScenes []TestScene `json:"test_scenes"` // 执行的场景列表
}

我们需要给测试计划配上具体的测试任务。
上述代码中的Task为测试任务结构，在model下新建task.go：

// Package model -----------------------------
// @file      : task.go
// @author    : 被测试耽误的大厨
// @contact   : 13383088061@163.com
// @time      : 2023/8/14 13:48
// -------------------------------------------
package model

/* 
        测试任务，我们先实现根据时长并发数模式
**/


type Task struct {
        ConcurrentUsers int64  `json:"concurrent_users"` // 并发用户数
        TimeType        string `json:"time_type"`        // 时间的类型 时：h   分：m   秒：s
        Duration        int64  `json:"duration"`         // 持续时长
}

我们实现一个计划可以运行多个场景，所以使用[]TestScene列表的机构，这样我们一个计划就包含了多个场景。下面我们优化一下，上一章节中的TestScene结构体的Dispose()方法，优化后如下：

// 优化前
func (testScene TestScene) Dispose() {
        // 从testScene.TestObjects的最外层开始循环
        for _, testObject := range testScene.TestObjects {
                // 声明一个waitGroup控制等待内层循环请求内部完成
                wg := sync.WaitGroup{}
                // 从一层级的list中读取每个TestObject对象
                wg.Add(1)
                go func(object TestObject) {
                        defer wg.Done()
                        response := TestObjectResponse{
                                Name:       object.Name,
                                Id:         object.Id,
                                SceneId:    testScene.Id,
                                ItemId:     object.ItemId,
                                ObjectType: object.ObjectType,
                        }
                        // 开始进行请求处理
                        object.Dispose(&response)
                }(testObject)
                // 等待内层的list完成后，再开始下一次外层循环
                wg.Wait()
        }

}


// 优化后
func (testScene TestScene) Dispose() {
        // 从testScene.TestObjects的最外层开始循环
        for _, testObject := range testScene.TestObjects {
                response := &TestObjectResponse{
                        Name:       testObject.Name,
                        Id:         testObject.Id,
                        SceneId:    testScene.Id,
                        ItemId:     testObject.ItemId,
                        ObjectType: testObject.ObjectType,
                }
                testObject.Dispose(response)
                // 从一层级的list中读取每个TestObject对象
        }

}

下面，我们实现一下TestPlan的Dispose函数， test_plan.go：

// 处理测试计划
func (testPlan TestPlan) Dispose() {

        duration := time.Duration(testPlan.Task.Duration)

        // 将时分转换为秒， 默认为秒
        switch testPlan.Task.TimeType {
        case constant.H:
                duration = duration * 3600
        case constant.M:
                duration = duration * 60
        }

        users := testPlan.Task.ConcurrentUsers
        testScenes := testPlan.TestScenes

        // 使用协程同时处理计划中的所有场景
        for _, testScene := range testScenes {
                go disposePlan(users, duration, testScene)
        }

}



// 开始进行并发处理
func disposePlan(users int64, duration time.Duration, testScene TestScene) {

        // 启动所有的用户，每个协程代表一个用户
        for i := int64(0); i < users; i++ {
                // 每个协程都配备一个定时器，时间到后，自动停止任务
                timer := time.NewTimer(duration * time.Second)
                go func() {
                        // 使用for循环和select进行定时控制
                        for {
                                select {
                                case <-timer.C: // 当到达持续时间后，停止任务
                                        log.Logger.Debug("定时器结束。。。。。。。。。。。。。。。。。")
                                        timer.Stop() // 先把定时器停止掉，防止内泄露
                                        return       // 退出
                                default:
                                        testScene.Dispose()
                                }
                        }

                }()
        }
}

这样，我们就可以简单的实现的性能任务测试了，下面我们实现一下testPlan的api，方便我们通过接口进行调用。
在service目录下新建stress_test_plan.go

// Package service -----------------------------
// @file      : stress_test_plan.go
// @author    : 被测试耽误的大厨
// @contact   : 13383088061@163.com
// @time      : 2023/8/14 14:23
// -------------------------------------------
package service

import (
        "encoding/json"
        "fmt"
        "github.com/gin-gonic/gin"
        "github.com/google/uuid"
        "kitchen-engine/global/common"
        "kitchen-engine/global/log"
        "kitchen-engine/model"
        "net/http"
)

func StressRunTestPlan(c *gin.Context) {
        // 声明一个TO对象
        var testPlan model.TestPlan

        // 接收json格式的请求数据
        err := c.ShouldBindJSON(&testPlan)
        id := uuid.New().String()
        // 如果请求格式错误
        if err != nil {
                log.Logger.Error("请求数据格式错误", err.Error())
                common.ReturnResponse(c, http.StatusBadRequest, id, "请求数据格式错误！", err.Error())
                return
        }

        // 使用json包解析以下TO对象, 解析出来为[]byte类型
        requestJson, _ := json.Marshal(testPlan)
        // 打印以下日志, 使用fmt.Sprintf包格式花数据，%s 表示string(requestJson)为字符串类型，如果不确定类型，可以使用%v表示
        log.Logger.Debug(fmt.Sprintf("测试对象： %s", string(requestJson)))

        // 如果场景为空，直接返回
        if testPlan.Task.ConcurrentUsers <= 0 {
                common.ReturnResponse(c, http.StatusBadRequest, id, "请求错误！", "并发数不能小于等于0")
                return
        }
        if testPlan.Task.Duration <= 0 {
                common.ReturnResponse(c, http.StatusBadRequest, id, "请求错误！", "运行时间不能小于等于0")
                return
        }

        if len(testPlan.TestScenes) <= 0 || len(testPlan.TestScenes[0].TestObjects) <= 0 {
                common.ReturnResponse(c, http.StatusBadRequest, id, "请求错误！", "计划中的场景不能为空")
                return
        }

        // 开始处理TP
        testPlan.Dispose()
        // 返回响应消息
        common.ReturnResponse(c, http.StatusOK, id, "请求成功！", "success")
        return
}

router.go
// Package routers -----------------------------
// @file      : router.go
// @author    : 被测试耽误的大厨
// @contact   : 13383088061@163.com
// @time      : 2023/6/29 13:54
// -------------------------------------------
package routers

import (
        "github.com/gin-gonic/gin"
        "kitchen-engine/service"
)

/*
  路由配置
*/
func initRouters(groups *gin.RouterGroup) {
        {
                groups.POST("/run/testObject/", service.RunTestObject)          //运行测试对象接口， url: http://*****/engine/run/testObject/
                groups.POST("/run/testScene/", service.RunTestScene)            //运行场景接口， url: http://*****/engine/run/testObject/
                groups.POST("/run/stress/testPlan/", service.StressRunTestPlan) //运行性能测试计划接口， url: http://*****/engine/run/stress/testPlan/

        }

}

我们在根目录下新建expample-json目录，这里存放我们请求本服务的json文件。
运行对象调试的json： object.json

接口： http://127.0.0.1:8002/engine/run/testObject/

{
    "name": "百度",
    "id": "12312312312312",
    "object_type": "HTTP1.1",
    "item_id": "12312312312312",
    "team_id": "1231231231231",
    "http_request": {
        "url": "http://www.baidu.com",
        "method": "GET",
        "request_timeout": 5,
        "headers": [],
        "querys": [],
        "cookies": [],
        "http_client_settings": {}
    }
}

运行场景调试的json：scene.json

接口：http://127.0.0.1:8002/engine/run/testScene/


{
    "name": "调试百度",
    "id": "dlsjflksdjflks",
    "item_id": "12312312312312",
    "team_id": "1231231231231",
    "test_objects": [
        {
            "name": "百度",
            "id": "12312312312312",
            "object_type": "HTTP1.1",
            "item_id": "12312312312312",
            "team_id": "1231231231231",
            "http_request": {
                "url": "http://www.baidu.com",
                "method": "GET",
                "request_timeout": 5,
                "headers": [],
                "querys": [],
                "cookies": [],
                "http_client_settings": {}
            }

        }
    ]
}

运行性能测试计划json: plan.json

接口： http://127.0.0.1:8002/engine/run/stress/testPlan/


{
    "id": "lkjflksjlfjsjlflsfjfskldj",
    "name": "性能测试百度接口",
    "item_id": "12312312312312",
    "team_id": "1231231231231",
    "task": {
        "concurrent_users": 5,
        "time_type": "s",
        "duration": 5
    },
    "test_scenes": [
        {
            "name": "调试百度",
            "id": "dlsjflksdjflks",
            "item_id": "12312312312312",
            "team_id": "1231231231231",
            "test_objects": [
                {
                    "name": "百度",
                    "id": "12312312312312",
                    "object_type": "HTTP1.1",
                    "item_id": "12312312312312",
                    "team_id": "1231231231231",
                    "http_request": {
                        "url": "http://www.baidu.com",
                        "method": "GET",
                        "request_timeout": 5,
                        "headers": [],
                        "querys": [],
                        "cookies": [],
                        "http_client_settings": {}
                    }

                }
            ]
        }
    ]
}

大家，可以在自己电脑运行一下，使用访问一下这几个接口试试。