feat: 完成用量统计功能模块

Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/UsageStatisticsManager.cs

@@ -0,0 +1,37 @@
+using Medallion.Threading;
+using Volo.Abp.Domain.Services;
+using Yi.Framework.AiHub.Domain.Entities;
+using Yi.Framework.SqlSugarCore.Abstractions;
+
+public class UsageStatisticsManager : DomainService
+{
+    private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _repository;
+
+    private IDistributedLockProvider DistributedLock =>
+        LazyServiceProvider.LazyGetRequiredService<IDistributedLockProvider>();
+
+    public async Task SetUsageAsync(Guid userId, string modelId, int inputTokenCount, int outputTokenCount)
+    {
+        await using (await DistributedLock.AcquireLockAsync($"UsageStatistics:{userId.ToString()}"))
+        {
+            var entity = await _repository._DbQueryable.FirstAsync(x => x.UserId == userId && x.ModelId == modelId);
+            //存在数据，更细
+            if (entity is not null)
+            {
+                entity.AddOnceChat(inputTokenCount, outputTokenCount);
+                await _repository.UpdateAsync(entity);
+            }
+            //不存在插入
+            else
+            {
+                var usage = new UsageStatisticsAggregateRoot(userId, modelId);
+                usage.AddOnceChat(inputTokenCount, outputTokenCount);
+                await _repository.InsertAsync(usage);
+            }
+        }
+    }
+}
+
+internal class LazyServiceProvider
+{
+}