public Task UpdateAtomicAsync(Func <Task <TDocument> > updateAsync, Func <TDocument, Task> afterUpdateAsync = null)
{
if (updateAsync == null)
{
return(Task.CompletedTask);
}
_updateDelegateAsync += () => updateAsync();
if (afterUpdateAsync != null)
{
_afterUpdateDelegateAsync += document => afterUpdateAsync(document);
}
DocumentStore.AfterCommitSuccess <TDocument>(async() =>
{
(var locker, var locked) = await _distributedLock.TryAcquireLockAsync(
_options.CacheKey + "_LOCK",
TimeSpan.FromMilliseconds(_options.LockTimeout),
TimeSpan.FromMilliseconds(_options.LockExpiration));
if (!locked)
{
return;
}
await using var acquiredLock = locker;
TDocument document = null;
foreach (var d in _updateDelegateAsync.GetInvocationList())
{
document = await((UpdateDelegate)d)();
}
document.Identifier ??= IdGenerator.GenerateId();
await SetInternalAsync(document);
if (_afterUpdateDelegateAsync != null)
{
foreach (var d in _afterUpdateDelegateAsync.GetInvocationList())
{
await((AfterUpdateDelegate)d)(document);
}
}
});
return(Task.CompletedTask);
}
public Task UpdateAtomicAsync(Func <Task <TDocument> > updateAsync, Func <TDocument, Task> afterUpdateAsync = null,
TimeSpan?lockAcquireTimeout = null, TimeSpan?lockExpirationTime = null)
{
if (updateAsync == null)
{
return(Task.CompletedTask);
}
_updateDelegateAsync += () => updateAsync();
if (afterUpdateAsync != null)
{
_afterUpdateDelegateAsync += document => afterUpdateAsync(document);
}
_documentStore.AfterCommitSuccess <TDocument>(async() =>
{
var timeout = lockAcquireTimeout ?? DefaultLockTimeout;
var expiration = lockExpirationTime ?? DefaultLockExpiration;
(var locker, var locked) = await _distributedLock.TryAcquireLockAsync(_lockKey, timeout, expiration);
if (!locked)
{
return;
}
await using var acquiredLock = locker;
TDocument document = null;
foreach (var d in _updateDelegateAsync.GetInvocationList())
{
document = await((UpdateDelegate)d)();
}
document.Identifier ??= IdGenerator.GenerateId();
await SetInternalAsync(document);
if (_afterUpdateDelegateAsync != null)
{
foreach (var d in _afterUpdateDelegateAsync.GetInvocationList())
{
await((AfterUpdateDelegate)d)(document);
}
}
});
return(Task.CompletedTask);
}
