statistics

type AIConfig struct {
	APIEndpoint string // API地址
	APIKey      string // API Key
	Model       string // 模型名称
}

type AIMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type AIRequest struct {
	Model    string      `json:"model"`
	Messages []AIMessage `json:"messages"`
}

type AIResponse struct {
	Choices []struct {
		Message struct {
			Content string `json:"content"`
		} `json:"message"`
	} `json:"choices"`
	Error *struct {
		Message string `json:"message"`
	} `json:"error,omitempty"`
}

type ReportData struct {
	URL             string       // 压测URL
	Concurrency     uint64       // 并发数
	TotalRequests   uint64       // 总请求数
	SuccessNum      uint64       // 成功数
	FailureNum      uint64       // 失败数
	TotalTime       float64      // 总耗时(秒)
	QPS             float64      // QPS
	MaxTime         float64      // 最大响应时间(ms)
	MinTime         float64      // 最小响应时间(ms)
	AvgTime         float64      // 平均响应时间(ms)
	TP90            float64      // TP90(ms)
	TP95            float64      // TP95(ms)
	TP99            float64      // TP99(ms)
	ReceivedBytes   int64        // 下载字节数
	ErrorCodeMap    map[int]int  // 错误码分布
	TimeRecords     []TimeRecord // 时间序列记录
	RequestTimeList []uint64     // 请求时间列表
	StartTime       time.Time    // 开始时间
	EndTime         time.Time    // 结束时间
}

type ResponseTimeDistribution struct {
	Labels []string `json:"labels"`
	Values []int    `json:"values"`
}

type ScoreResult struct {
	TotalScore       int               `json:"total_score"`        // 总分 (0-100)
	Grade            string            `json:"grade"`              // 评级 (A/B/C/D/F)
	SuccessRateScore int               `json:"success_rate_score"` // 成功率得分
	QPSScore         int               `json:"qps_score"`          // QPS得分
	AvgTimeScore     int               `json:"avg_time_score"`     // 平均响应时间得分
	TP99Score        int               `json:"tp99_score"`         // TP99稳定性得分
	ErrorCodeScore   int               `json:"error_code_score"`   // 错误码得分
	Suggestions      []string          `json:"suggestions"`        // 改进建议
	Details          map[string]string `json:"details"`            // 详细说明
}

type TimeRecord struct {
	Timestamp   time.Time   // 记录时间点
	Elapsed     float64     // 耗时(秒)
	Concurrent  int         // 并发数
	Success     uint64      // 成功数
	Failure     uint64      // 失败数
	SuccessRate float64     // 成功率(%)
	QPS         float64     // QPS
	MaxTime     float64     // 最大响应时间(ms)
	MinTime     float64     // 最小响应时间(ms)
	AvgTime     float64     // 平均响应时间(ms)
	ErrorCodes  map[int]int // 错误码分布
}