|
|
|
|
@ -1,5 +1,6 @@
|
|
|
|
|
using System;
|
|
|
|
|
using Expedience.Infrastructure;
|
|
|
|
|
using Expedience.Infrastructure.Concurrency;
|
|
|
|
|
using Expedience.Infrastructure.Models;
|
|
|
|
|
using MassTransit;
|
|
|
|
|
|
|
|
|
|
@ -9,12 +10,14 @@ namespace Expedience.Api.Consumers
|
|
|
|
|
{
|
|
|
|
|
private readonly ILogger<DutyCompletionResultConsumer> _logger;
|
|
|
|
|
private readonly IServiceScopeFactory _serviceScopeFactory;
|
|
|
|
|
|
|
|
|
|
private readonly IDistributedLock _distributedLock;
|
|
|
|
|
public DutyCompletionResultConsumer(ILogger<DutyCompletionResultConsumer> logger,
|
|
|
|
|
IDistributedLock distributedLock,
|
|
|
|
|
IServiceScopeFactory serviceScopeFactory)
|
|
|
|
|
{
|
|
|
|
|
_logger = logger;
|
|
|
|
|
_serviceScopeFactory = serviceScopeFactory;
|
|
|
|
|
_distributedLock = distributedLock;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public async Task Consume(ConsumeContext<Models.DutyCompletionResult> context)
|
|
|
|
|
@ -33,12 +36,11 @@ namespace Expedience.Api.Consumers
|
|
|
|
|
|
|
|
|
|
using var scope = _serviceScopeFactory.CreateScope();
|
|
|
|
|
using var dbContext = scope.ServiceProvider.GetRequiredService<ExpedienceContext>();
|
|
|
|
|
var user = dbContext.Users.FirstOrDefault(x => x.UserHash == userHash && x.WorldId == worldId);
|
|
|
|
|
var user = dbContext.Users.FirstOrDefault(x => x.UserHash == userHash && x.WorldId == worldId) ??
|
|
|
|
|
await CreateUser(dbContext, worldId, userHash, CancellationToken.None);
|
|
|
|
|
|
|
|
|
|
if (user == null)
|
|
|
|
|
{
|
|
|
|
|
_logger.LogError("No user found for World {worldId} and Hash {userHash}", worldId, userHash);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var completionResult = new DutyCompletionResult
|
|
|
|
|
{
|
|
|
|
|
@ -89,5 +91,58 @@ namespace Expedience.Api.Consumers
|
|
|
|
|
|
|
|
|
|
_logger.LogInformation("Consumed message {uploadId} from user {userHash}", message.UploadId, userHash);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private async Task<User?> CreateUser(ExpedienceContext dbContext, int worldId, string userHash, CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
var lockKey = $"{worldId}-{userHash}";
|
|
|
|
|
if (_distributedLock.AcquireLock(lockKey, TimeSpan.FromSeconds(10), TimeSpan.FromMinutes(4)))
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
var user = dbContext.Users.FirstOrDefault(x => x.UserHash == userHash && x.WorldId == worldId);
|
|
|
|
|
if (user == null)
|
|
|
|
|
{
|
|
|
|
|
string userName;
|
|
|
|
|
var isDuplicate = false;
|
|
|
|
|
do
|
|
|
|
|
{
|
|
|
|
|
userName = UserNameGenerator.Generate();
|
|
|
|
|
isDuplicate = dbContext.Users.Any(x => x.WorldId == worldId && x.UserName == userName);
|
|
|
|
|
await Task.Delay(20, cancellationToken); // Don't hog the CPU
|
|
|
|
|
}
|
|
|
|
|
while (isDuplicate == true);
|
|
|
|
|
|
|
|
|
|
user = new User
|
|
|
|
|
{
|
|
|
|
|
UserHash = userHash,
|
|
|
|
|
WorldId = worldId,
|
|
|
|
|
UserName = userName,
|
|
|
|
|
CreatedAt = DateTime.UtcNow,
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
dbContext.Users.Add(user);
|
|
|
|
|
await dbContext.SaveChangesAsync(cancellationToken);
|
|
|
|
|
_logger.LogInformation("Created user for World {worldId} and Hash {userHash}: {userName}", worldId, userHash, user.UserName);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return user;
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
catch (Exception ex)
|
|
|
|
|
{
|
|
|
|
|
_logger.LogError(ex, "Error obtaining user name for World Id {worldId} and hash {userHash}: {errorMessage}", worldId, userHash, ex.Message);
|
|
|
|
|
}
|
|
|
|
|
finally
|
|
|
|
|
{
|
|
|
|
|
_distributedLock.ReleaseLock(lockKey);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
_logger.LogError("Could not acquire lock for {lockKey}", lockKey);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|