From 2796c05934024c9e55a9e1fac05a8594d832b6b7 Mon Sep 17 00:00:00 2001 From: tbphp Date: Sat, 12 Jul 2025 13:01:35 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=88=86=E7=BB=84=E7=BB=9F=E8=AE=A1?= =?UTF-8?q?=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/handler/group_handler.go | 179 ++++++++++++++++++++++++++++++ internal/router/router.go | 2 +- 2 files changed, 180 insertions(+), 1 deletion(-) diff --git a/internal/handler/group_handler.go b/internal/handler/group_handler.go index c69a4df..4d891d3 100644 --- a/internal/handler/group_handler.go +++ b/internal/handler/group_handler.go @@ -5,6 +5,7 @@ import ( "encoding/json" "fmt" "net/url" + "sync" app_errors "gpt-load/internal/errors" "gpt-load/internal/models" @@ -532,3 +533,181 @@ func (s *Server) GetGroupConfigOptions(c *gin.Context) { response.Success(c, options) } + +// KeyStats defines the statistics for API keys in a group. +type KeyStats struct { + TotalKeys int64 `json:"total_keys"` + ActiveKeys int64 `json:"active_keys"` + InvalidKeys int64 `json:"invalid_keys"` +} + +// RequestStats defines the statistics for requests over a period. +type RequestStats struct { + TotalRequests int64 `json:"total_requests"` + FailedRequests int64 `json:"failed_requests"` + FailureRate float64 `json:"failure_rate"` +} + +// GroupStatsResponse defines the complete statistics for a group. +type GroupStatsResponse struct { + KeyStats KeyStats `json:"key_stats"` + HourlyStats RequestStats `json:"hourly_stats"` // 1 hour + DailyStats RequestStats `json:"daily_stats"` // 24 hours + WeeklyStats RequestStats `json:"weekly_stats"` // 7 days +} + +// calculateRequestStats is a helper to compute request statistics. +func calculateRequestStats(total, failed int64) RequestStats { + stats := RequestStats{ + TotalRequests: total, + FailedRequests: failed, + } + if total > 0 { + stats.FailureRate, _ = strconv.ParseFloat(fmt.Sprintf("%.4f", float64(failed)/float64(total)), 64) + } + return stats +} + +// GetGroupStats handles retrieving detailed statistics for a specific group. +func (s *Server) GetGroupStats(c *gin.Context) { + id, err := strconv.Atoi(c.Param("id")) + if err != nil { + response.Error(c, app_errors.NewAPIError(app_errors.ErrBadRequest, "Invalid group ID format")) + return + } + groupID := uint(id) + + // 1. 验证分组是否存在 + var group models.Group + if err := s.DB.First(&group, groupID).Error; err != nil { + response.Error(c, app_errors.ParseDBError(err)) + return + } + + var resp GroupStatsResponse + var wg sync.WaitGroup + var mu sync.Mutex + var errors []error + + // 并发执行所有统计查询 + + // 2. Key 统计 + wg.Add(1) + go func() { + defer wg.Done() + var totalKeys, activeKeys int64 + + if err := s.DB.Model(&models.APIKey{}).Where("group_id = ?", groupID).Count(&totalKeys).Error; err != nil { + mu.Lock() + errors = append(errors, fmt.Errorf("failed to get total keys: %w", err)) + mu.Unlock() + return + } + if err := s.DB.Model(&models.APIKey{}).Where("group_id = ? AND status = ?", groupID, models.KeyStatusActive).Count(&activeKeys).Error; err != nil { + mu.Lock() + errors = append(errors, fmt.Errorf("failed to get active keys: %w", err)) + mu.Unlock() + return + } + + mu.Lock() + resp.KeyStats = KeyStats{ + TotalKeys: totalKeys, + ActiveKeys: activeKeys, + InvalidKeys: totalKeys - activeKeys, + } + mu.Unlock() + }() + + // 3. 1小时请求统计 (查询 request_logs 表) + wg.Add(1) + go func() { + defer wg.Done() + var total, failed int64 + now := time.Now() + oneHourAgo := now.Add(-1 * time.Hour) + + if err := s.DB.Model(&models.RequestLog{}).Where("group_id = ? AND timestamp BETWEEN ? AND ?", groupID, oneHourAgo, now).Count(&total).Error; err != nil { + mu.Lock() + errors = append(errors, fmt.Errorf("failed to get hourly total requests: %w", err)) + mu.Unlock() + return + } + if err := s.DB.Model(&models.RequestLog{}).Where("group_id = ? AND timestamp BETWEEN ? AND ? AND is_success = ?", groupID, oneHourAgo, now, false).Count(&failed).Error; err != nil { + mu.Lock() + errors = append(errors, fmt.Errorf("failed to get hourly failed requests: %w", err)) + mu.Unlock() + return + } + + mu.Lock() + resp.HourlyStats = calculateRequestStats(total, failed) + mu.Unlock() + }() + + // 4. 24小时和7天统计 (查询 group_hourly_stats 表) + // 辅助函数，用于从 group_hourly_stats 查询 + queryHourlyStats := func(duration time.Duration) (RequestStats, error) { + var result struct { + SuccessCount int64 + FailureCount int64 + } + now := time.Now() + // 结束时间为当前小时的整点，查询时不包含该小时 + // 开始时间为结束时间减去统计周期 + endTime := now.Truncate(time.Hour) + startTime := endTime.Add(-duration) + + err := s.DB.Model(&models.GroupHourlyStat{}). + Select("SUM(success_count) as success_count, SUM(failure_count) as failure_count"). + Where("group_id = ? AND time >= ? AND time < ?", groupID, startTime, endTime). + Scan(&result).Error + if err != nil { + return RequestStats{}, err + } + return calculateRequestStats(result.SuccessCount+result.FailureCount, result.FailureCount), nil + } + + // 24小时统计 + wg.Add(1) + go func() { + defer wg.Done() + stats, err := queryHourlyStats(24 * time.Hour) + if err != nil { + mu.Lock() + errors = append(errors, fmt.Errorf("failed to get daily stats: %w", err)) + mu.Unlock() + return + } + mu.Lock() + resp.DailyStats = stats + mu.Unlock() + }() + + // 7天统计 + wg.Add(1) + go func() { + defer wg.Done() + stats, err := queryHourlyStats(7 * 24 * time.Hour) + if err != nil { + mu.Lock() + errors = append(errors, fmt.Errorf("failed to get weekly stats: %w", err)) + mu.Unlock() + return + } + mu.Lock() + resp.WeeklyStats = stats + mu.Unlock() + }() + + wg.Wait() + + if len(errors) > 0 { + // 只记录第一个错误，但表明可能存在多个错误 + logrus.WithContext(c.Request.Context()).WithError(errors[0]).Error("Errors occurred while fetching group stats") + response.Error(c, app_errors.NewAPIError(app_errors.ErrDatabase, "Failed to retrieve some statistics")) + return + } + + response.Success(c, resp) +} diff --git a/internal/router/router.go b/internal/router/router.go index 0894659..ed66c2b 100644 --- a/internal/router/router.go +++ b/internal/router/router.go @@ -113,7 +113,7 @@ func registerProtectedAPIRoutes(api *gin.RouterGroup, serverHandler *handler.Ser groups.GET("/config-options", serverHandler.GetGroupConfigOptions) groups.PUT("/:id", serverHandler.UpdateGroup) groups.DELETE("/:id", serverHandler.DeleteGroup) - + groups.GET("/:id/stats", serverHandler.GetGroupStats) } // Key Management Routes