feat: 新增近24小时每小时与今日模型使用量统计接口及实现

Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/UsageStatisticsService.cs

@@ -57,7 +57,7 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
         // 从Message表统计近7天的token消耗
         var dailyUsage = await _messageRepository._DbQueryable
             .Where(x => x.UserId == userId)
-            .Where(x => x.Role == "assistant" || x.Role == "system")
+            .Where(x => x.Role == "system")
             .Where(x => x.CreationTime >= startDate && x.CreationTime < endDate.AddDays(1))
             .WhereIF(input.TokenId.HasValue,x => x.TokenId == input.TokenId)
             .GroupBy(x => x.CreationTime.Date)
@@ -228,4 +228,106 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
 
         return result;
     }
+
+    /// <summary>
+    /// 获取当前用户近24小时每小时Token消耗统计（柱状图）
+    /// </summary>
+    /// <returns>每小时Token使用量列表，包含各模型堆叠数据</returns>
+    public async Task<List<HourlyTokenUsageDto>> GetLast24HoursTokenUsageAsync([FromQuery]UsageStatisticsGetInput input)
+    {
+        var userId = CurrentUser.GetId();
+        var now = DateTime.Now;
+        var startTime = now.AddHours(-23); // 滚动24小时，从23小时前到现在
+        var startHour = new DateTime(startTime.Year, startTime.Month, startTime.Day, startTime.Hour, 0, 0);
+
+        // 从Message表查询近24小时的数据，只选择需要的字段
+        var messages = await _messageRepository._DbQueryable
+            .Where(x => x.UserId == userId)
+            .Where(x => x.Role == "system")
+            .Where(x => x.CreationTime >= startHour)
+            .WhereIF(input.TokenId.HasValue, x => x.TokenId == input.TokenId)
+            .Select(x => new
+            {
+                x.CreationTime,
+                x.ModelId,
+                x.TokenUsage.TotalTokenCount
+            })
+            .ToListAsync();
+
+        // 在内存中按小时和模型分组统计
+        var hourlyGrouped = messages
+            .GroupBy(x => new
+            {
+                Hour = new DateTime(x.CreationTime.Year, x.CreationTime.Month, x.CreationTime.Day, x.CreationTime.Hour, 0, 0),
+                x.ModelId
+            })
+            .Select(g => new
+            {
+                g.Key.Hour,
+                g.Key.ModelId,
+                Tokens = g.Sum(x => x.TotalTokenCount)
+            })
+            .ToList();
+
+        // 生成完整的24小时数据
+        var result = new List<HourlyTokenUsageDto>();
+        for (int i = 0; i < 24; i++)
+        {
+            var hour = startHour.AddHours(i);
+            var hourData = hourlyGrouped.Where(x => x.Hour == hour).ToList();
+
+            var modelBreakdown = hourData.Select(x => new ModelTokenBreakdownDto
+            {
+                ModelId = x.ModelId,
+                Tokens = x.Tokens
+            }).ToList();
+
+            result.Add(new HourlyTokenUsageDto
+            {
+                Hour = hour,
+                TotalTokens = modelBreakdown.Sum(x => x.Tokens),
+                ModelBreakdown = modelBreakdown
+            });
+        }
+
+        return result;
+    }
+
+    /// <summary>
+    /// 获取当前用户今日各模型使用量统计（卡片列表）
+    /// </summary>
+    /// <returns>模型今日使用量列表，包含使用次数和总Token</returns>
+    public async Task<List<ModelTodayUsageDto>> GetTodayModelUsageAsync([FromQuery]UsageStatisticsGetInput input)
+    {
+        var userId = CurrentUser.GetId();
+        var todayStart = DateTime.Today; // 今天凌晨0点
+        var tomorrowStart = todayStart.AddDays(1);
+
+        // 从Message表查询今天的数据，只选择需要的字段
+        var messages = await _messageRepository._DbQueryable
+            .Where(x => x.UserId == userId)
+            .Where(x => x.Role == "system")
+            .Where(x => x.CreationTime >= todayStart && x.CreationTime < tomorrowStart)
+            .WhereIF(input.TokenId.HasValue, x => x.TokenId == input.TokenId)
+            .Select(x => new
+            {
+                x.ModelId,
+                x.TokenUsage.TotalTokenCount
+            })
+            .ToListAsync();
+
+        // 在内存中按模型分组统计
+        var modelStats = messages
+            .GroupBy(x => x.ModelId)
+            .Select(g => new ModelTodayUsageDto
+            {
+                ModelId = g.Key,
+                UsageCount = g.Count(),
+                TotalTokens = g.Sum(x => x.TotalTokenCount)
+            })
+            .OrderByDescending(x => x.TotalTokens)
+            .ToList();
+
+        return modelStats;
+    }
 }