// Package server 压测启动
package server

import (
	"context"
	"fmt"
	"go_dreamfactory/stress/model"
	"go_dreamfactory/stress/robot"
	"strconv"

	"sync"
	"time"

	"go_dreamfactory/stress/server/golink"
	"go_dreamfactory/stress/server/statistics"
	"go_dreamfactory/stress/server/verify"
)

const (
	connectionMode = 2 // 1:顺序建立长链接 2:并发建立长链接
)

// init 注册验证器
func init() {

	// http
	//model.RegisterVerifyHTTP("statusCode", verify.HTTPStatusCode)
	//model.RegisterVerifyHTTP("json", verify.HTTPJson)

	// webSocket
	//model.RegisterVerifyWebSocket("json", verify.WebSocketJSON)
	model.RegisterVerifyWebSocket("pb", verify.WebSocketProto)
}

// Dispose 处理函数
func Dispose(ctx context.Context, concurrency, totalNumber uint64, request *model.Request) {
	// 设置接收数据缓存
	ch := make(chan *model.RequestResults, 1000)
	var (
		wg          sync.WaitGroup // 发送数据完成
		wgReceiving sync.WaitGroup // 数据处理完成
	)
	wgReceiving.Add(1)
	go statistics.ReceivingResults(concurrency, ch, &wgReceiving)

	for i := uint64(0); i < concurrency; i++ {
		wg.Add(1)
		switch request.Form {
		case model.FormTypeWebSocket:
			switch connectionMode {
			case 1:
				// 连接以后再启动协程
				r := robot.NewRobot(request.URL)
				//r.SetAccount("e51112" + strconv.Itoa(int(i)))
				r.SetAccount("user001")
				//head := &pb.UserMessage{MainType: "user", SubType: "login"}

				// 先登录
				//r.SendToClient(head, &pb.UserLoginReq{})
				//r.AddGrormetItem()
				//r.Create()
				go golink.WebSocket(ctx, i, ch, totalNumber, &wg, request, r)
			case 2:
				// 并发建立长链接
				go func(i uint64) {
					r := robot.NewRobot(request.URL)
					r.SetAccount("d71112" + strconv.Itoa(int(i)))
					//head := &pb.UserMessage{MainType: "user", SubType: "login"}

					// 先登录
					//r.SendToClient(head, &pb.UserLoginReq{})
					//r.AddGrormetItem()
					//r.Create()
					go golink.WebSocket(ctx, i, ch, totalNumber, &wg, request, r)
				}(i)
				// 注意:时间间隔太短会出现连接失败的报错 默认连接时长:20毫秒(公网连接)
				time.Sleep(5 * time.Millisecond)
			default:
				data := fmt.Sprintf("不支持的类型:%d", connectionMode)
				panic(data)
			}
		default:
			// 类型不支持
			wg.Done()
		}
	}
	// 等待所有的数据都发送完成
	wg.Wait()
	// 延时1毫秒 确保数据都处理完成了
	time.Sleep(1 * time.Millisecond)
	close(ch)
	// 数据全部处理完成了
	wgReceiving.Wait()
	return
}