feat: 完成token功能

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

@@ -9,7 +9,9 @@ using Yi.Framework.AiHub.Application.Contracts.Dtos.UsageStatistics;
 using Yi.Framework.AiHub.Application.Contracts.IServices;
 using Yi.Framework.AiHub.Domain.Entities;
 using Yi.Framework.AiHub.Domain.Entities.Chat;
+using Yi.Framework.AiHub.Domain.Entities.OpenApi;
 using Yi.Framework.AiHub.Domain.Extensions;
+using Yi.Framework.AiHub.Domain.Shared.Consts;
 using Yi.Framework.Ddd.Application.Contracts;
 using Yi.Framework.SqlSugarCore.Abstractions;
 
@@ -24,15 +26,18 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
     private readonly ISqlSugarRepository<MessageAggregateRoot> _messageRepository;
     private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _usageStatisticsRepository;
     private readonly ISqlSugarRepository<PremiumPackageAggregateRoot> _premiumPackageRepository;
+    private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
 
     public UsageStatisticsService(
         ISqlSugarRepository<MessageAggregateRoot> messageRepository,
         ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository,
-        ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository)
+        ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository,
+        ISqlSugarRepository<TokenAggregateRoot> tokenRepository)
     {
         _messageRepository = messageRepository;
         _usageStatisticsRepository = usageStatisticsRepository;
         _premiumPackageRepository = premiumPackageRepository;
+        _tokenRepository = tokenRepository;
     }
 
     /// <summary>
@@ -83,13 +88,14 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
     {
         var userId = CurrentUser.GetId();
 
-        // 从UsageStatistics表获取各模型的token消耗统计
+        // 从UsageStatistics表获取各模型的token消耗统计（按ModelId聚合，因为同一模型可能有多个TokenId的记录）
         var modelUsages = await _usageStatisticsRepository._DbQueryable
             .Where(x => x.UserId == userId)
+            .GroupBy(x => x.ModelId)
             .Select(x => new
             {
-                x.ModelId,
-                x.TotalTokenCount
+                ModelId = x.ModelId,
+                TotalTokenCount = SqlFunc.AggregateSum(x.TotalTokenCount)
             })
             .ToListAsync();
 
@@ -164,4 +170,54 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
         return new PagedResultDto<PremiumTokenUsageGetListOutput>(total,
             entities.Adapt<List<PremiumTokenUsageGetListOutput>>());
     }
+
+    /// <summary>
+    /// 获取当前用户尊享包不同Token用量占比（饼图）
+    /// </summary>
+    /// <returns>各Token的尊享模型用量及占比</returns>
+    [HttpGet("usage-statistics/premium-token-usage/by-token")]
+    public async Task<List<TokenPremiumUsageDto>> GetPremiumTokenUsageByTokenAsync()
+    {
+        var userId = CurrentUser.GetId();
+        var premiumModelIds = PremiumPackageConst.ModeIds;
+
+        // 从UsageStatistics表获取尊享模型的token消耗统计（按TokenId聚合）
+        var tokenUsages = await _usageStatisticsRepository._DbQueryable
+            .Where(x => x.UserId == userId && premiumModelIds.Contains(x.ModelId))
+            .GroupBy(x => x.TokenId)
+            .Select(x => new
+            {
+                TokenId = x.TokenId,
+                TotalTokenCount = SqlFunc.AggregateSum(x.TotalTokenCount)
+            })
+            .ToListAsync();
+
+        if (!tokenUsages.Any())
+        {
+            return new List<TokenPremiumUsageDto>();
+        }
+
+        // 获取用户的所有Token信息用于名称映射
+        var tokenIds = tokenUsages.Select(x => x.TokenId).ToList();
+        var tokens = await _tokenRepository._DbQueryable
+            .Where(x => x.UserId == userId && tokenIds.Contains(x.Id))
+            .Select(x => new { x.Id, x.Name })
+            .ToListAsync();
+
+        var tokenNameDict = tokens.ToDictionary(x => x.Id, x => x.Name);
+
+        // 计算总token数
+        var totalTokens = tokenUsages.Sum(x => x.TotalTokenCount);
+
+        // 计算各Token占比
+        var result = tokenUsages.Select(x => new TokenPremiumUsageDto
+        {
+            TokenId = x.TokenId,
+            TokenName = x.TokenId == Guid.Empty ? "Web端" : (tokenNameDict.TryGetValue(x.TokenId, out var name) ? name : "未知Token"),
+            Tokens = x.TotalTokenCount,
+            Percentage = totalTokens > 0 ? Math.Round((decimal)x.TotalTokenCount / totalTokens * 100, 2) : 0
+        }).OrderByDescending(x => x.Tokens).ToList();
+
+        return result;
+    }
 }