feat: 完成尊享服务

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

@@ -0,0 +1,147 @@
+using SqlSugar;
+using Volo.Abp.Domain.Entities.Auditing;
+
+namespace Yi.Framework.AiHub.Domain.Entities;
+
+/// <summary>
+/// 尊享包聚合根
+/// 用于给VIP扩展额外购买尊享token包
+/// </summary>
+[SugarTable("Ai_PremiumPackage")]
+[SugarIndex($"index_{nameof(UserId)}", nameof(UserId), OrderByType.Asc)]
+public class PremiumPackageAggregateRoot : FullAuditedAggregateRoot<Guid>
+{
+    public PremiumPackageAggregateRoot()
+    {
+    }
+
+    public PremiumPackageAggregateRoot(Guid userId, long totalTokens, string packageName)
+    {
+        UserId = userId;
+        TotalTokens = totalTokens;
+        RemainingTokens = totalTokens;
+        PackageName = packageName;
+        IsActive = true;
+    }
+
+    /// <summary>
+    /// 用户ID
+    /// </summary>
+    public Guid UserId { get; set; }
+
+    /// <summary>
+    /// 包名称
+    /// </summary>
+    public string PackageName { get; set; }
+
+    /// <summary>
+    /// 总用量（总token数）
+    /// </summary>
+    public long TotalTokens { get; set; }
+
+    /// <summary>
+    /// 剩余用量（剩余token数）
+    /// </summary>
+    public long RemainingTokens { get; set; }
+
+    /// <summary>
+    /// 已使用token数
+    /// </summary>
+    public long UsedTokens { get; set; }
+
+    /// <summary>
+    /// 到期时间
+    /// </summary>
+    public DateTime? ExpireDateTime { get; set; }
+
+    /// <summary>
+    /// 是否激活
+    /// </summary>
+    public bool IsActive { get; set; }
+
+    /// <summary>
+    /// 购买金额
+    /// </summary>
+    public decimal PurchaseAmount { get; set; }
+
+    /// <summary>
+    /// 备注
+    /// </summary>
+    public string? Remark { get; set; }
+
+    /// <summary>
+    /// 消耗token
+    /// </summary>
+    /// <param name="tokenCount">消耗的token数量</param>
+    /// <returns>是否消耗成功</returns>
+    public bool ConsumeTokens(long tokenCount)
+    {
+        if (RemainingTokens < tokenCount)
+        {
+            return false;
+        }
+
+        if (!IsActive)
+        {
+            return false;
+        }
+
+        if (ExpireDateTime.HasValue && ExpireDateTime.Value < DateTime.Now)
+        {
+            return false;
+        }
+
+        RemainingTokens -= tokenCount;
+        UsedTokens += tokenCount;
+        return true;
+    }
+
+    /// <summary>
+    /// 检查是否可用
+    /// </summary>
+    /// <returns>是否可用</returns>
+    public bool IsAvailable()
+    {
+        if (!IsActive)
+        {
+            return false;
+        }
+
+        if (RemainingTokens <= 0)
+        {
+            return false;
+        }
+
+        if (ExpireDateTime.HasValue && ExpireDateTime.Value < DateTime.Now)
+        {
+            return false;
+        }
+
+        return true;
+    }
+
+    /// <summary>
+    /// 停用尊享包
+    /// </summary>
+    public void Deactivate()
+    {
+        IsActive = false;
+    }
+
+    /// <summary>
+    /// 激活尊享包
+    /// </summary>
+    public void Activate()
+    {
+        IsActive = true;
+    }
+
+    /// <summary>
+    /// 设置到期时间
+    /// </summary>
+    /// <param name="expireDateTime">到期时间</param>
+    public void SetExpireDateTime(DateTime expireDateTime)
+    {
+        ExpireDateTime = expireDateTime;
+    }
+}