37 lines
1.3 KiB
C#
37 lines
1.3 KiB
C#
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
|
|
{
|
|
} |