feat: 完成token功能

Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Chat/MessageAggregateRoot.cs

@@ -20,10 +20,11 @@ public class MessageAggregateRoot : FullAuditedAggregateRoot<Guid>
     }
 
     public MessageAggregateRoot(Guid? userId, Guid? sessionId, string content, string role, string modelId,
-        ThorUsageResponse? tokenUsage)
+        ThorUsageResponse? tokenUsage, Guid? tokenId = null)
     {
         UserId = userId;
         SessionId = sessionId;
+        TokenId = tokenId ?? Guid.Empty;
         //如果没有会话，不存储对话内容
         Content = sessionId is null ? null : content;
         Role = role;
@@ -59,6 +60,11 @@ public class MessageAggregateRoot : FullAuditedAggregateRoot<Guid>
     public Guid? UserId { get; set; }
     public Guid? SessionId { get; set; }
 
+    /// <summary>
+    /// Token密钥Id（通过API调用时记录，Web调用为Guid.Empty）
+    /// </summary>
+    public Guid TokenId { get; set; }
+
     [SugarColumn(ColumnDataType = StaticConfig.CodeFirst_BigString)]
     public string? Content { get; set; }
 

Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/OpenApi/TokenAggregateRoot.cs

@@ -5,27 +5,84 @@ using Volo.Abp.Domain.Entities.Auditing;
 namespace Yi.Framework.AiHub.Domain.Entities.OpenApi;
 
 [SugarTable("Ai_Token")]
+[SugarIndex($"index_{{table}}_{nameof(UserId)}", nameof(UserId), OrderByType.Asc)]
 public class TokenAggregateRoot : FullAuditedAggregateRoot<Guid>
 {
     public TokenAggregateRoot()
     {
     }
 
-    public TokenAggregateRoot(Guid userId)
+    public TokenAggregateRoot(Guid userId, string name)
     {
-        this.UserId = userId;
-        this.Token = GenerateToken();
+        UserId = userId;
+        Name = name;
+        Token = GenerateToken();
+        IsDisabled = false;
     }
 
+    /// <summary>
+    /// Token密钥
+    /// </summary>
     public string Token { get; set; }
+
+    /// <summary>
+    /// 用户Id
+    /// </summary>
     public Guid UserId { get; set; }
 
     /// <summary>
-    /// 重置token
+    /// 名称
     /// </summary>
-    public void ResetToken()
+    [SugarColumn(Length = 100)]
+    public string Name { get; set; }
+
+    /// <summary>
+    /// 过期时间（空为永不过期）
+    /// </summary>
+    public DateTime? ExpireTime { get; set; }
+
+    /// <summary>
+    /// 尊享包额度限制（空为不限制）
+    /// </summary>
+    public long? PremiumQuotaLimit { get; set; }
+
+    /// <summary>
+    /// 是否禁用
+    /// </summary>
+    public bool IsDisabled { get; set; }
+
+    /// <summary>
+    /// 检查Token是否可用
+    /// </summary>
+    public bool IsAvailable()
     {
-        this.Token = GenerateToken();
+        if (IsDisabled)
+        {
+            return false;
+        }
+
+        if (ExpireTime.HasValue && ExpireTime.Value < DateTime.Now)
+        {
+            return false;
+        }
+
+        return true;
+    }
+
+    /// <summary>
+    /// 禁用Token
+    /// </summary>
+    public void Disable()
+    {
+        IsDisabled = true;
+    }
+
+    /// <summary>
+    /// 启用Token
+    /// </summary>
+    public void Enable()
+    {
+        IsDisabled = false;
     }
 
     private string GenerateToken(int length = 36)

Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/UsageStatisticsAggregateRoot.cs

@@ -7,16 +7,22 @@ namespace Yi.Framework.AiHub.Domain.Entities;
 /// 用量统计
 /// </summary>
 [SugarTable("Ai_UsageStatistics")]
+[SugarIndex($"index_{{table}}_{nameof(UserId)}_{nameof(ModelId)}_{nameof(TokenId)}",
+    nameof(UserId), OrderByType.Asc,
+    nameof(ModelId), OrderByType.Asc,
+    nameof(TokenId), OrderByType.Asc
+)]
 public class UsageStatisticsAggregateRoot : FullAuditedAggregateRoot<Guid>
 {
     public UsageStatisticsAggregateRoot()
     {
     }
 
-    public UsageStatisticsAggregateRoot(Guid? userId, string modelId)
+    public UsageStatisticsAggregateRoot(Guid? userId, string modelId, Guid tokenId)
     {
         UserId = userId;
         ModelId = modelId;
+        TokenId = tokenId;
     }
 
     /// <summary>
@@ -29,6 +35,11 @@ public class UsageStatisticsAggregateRoot : FullAuditedAggregateRoot<Guid>
     /// </summary>
     public string ModelId { get; set; }
 
+    /// <summary>
+    /// Token密钥Id（通过API调用时记录，Web调用为Guid.Empty）
+    /// </summary>
+    public Guid TokenId { get; set; }
+
     /// <summary>
     /// 对话次数
     /// </summary>