mirror of https://github.com/Squidex/squidex.git
11 changed files with 261 additions and 4 deletions
@ -0,0 +1,18 @@ |
|||||
|
// ==========================================================================
|
||||
|
// Squidex Headless CMS
|
||||
|
// ==========================================================================
|
||||
|
// Copyright (c) Squidex UG (haftungsbeschraenkt)
|
||||
|
// All rights reserved. Licensed under the MIT license.
|
||||
|
// ==========================================================================
|
||||
|
|
||||
|
using System; |
||||
|
using System.IO; |
||||
|
using System.Threading.Tasks; |
||||
|
|
||||
|
namespace Squidex.Infrastructure.Log |
||||
|
{ |
||||
|
public interface ILogStore |
||||
|
{ |
||||
|
Task ReadLockAsync(string key, DateTime from, DateTime to, Stream stream); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,62 @@ |
|||||
|
// ==========================================================================
|
||||
|
// Squidex Headless CMS
|
||||
|
// ==========================================================================
|
||||
|
// Copyright (c) Squidex UG (haftungsbeschraenkt)
|
||||
|
// All rights reserved. Licensed under the MIT license.
|
||||
|
// ==========================================================================
|
||||
|
|
||||
|
using System; |
||||
|
using System.IO; |
||||
|
using System.Threading; |
||||
|
using System.Threading.Tasks; |
||||
|
using Orleans; |
||||
|
using Squidex.Infrastructure.Orleans; |
||||
|
|
||||
|
namespace Squidex.Infrastructure.Log |
||||
|
{ |
||||
|
public sealed class LockingLogStore : ILogStore |
||||
|
{ |
||||
|
private static readonly TimeSpan LockWaitingTime = TimeSpan.FromMinutes(10); |
||||
|
private readonly ILogStore inner; |
||||
|
private readonly ILockGrain lockGrain; |
||||
|
|
||||
|
public LockingLogStore(ILogStore inner, IGrainFactory grainFactory) |
||||
|
{ |
||||
|
Guard.NotNull(inner, nameof(inner)); |
||||
|
Guard.NotNull(grainFactory, nameof(grainFactory)); |
||||
|
|
||||
|
this.inner = inner; |
||||
|
|
||||
|
lockGrain = grainFactory.GetGrain<ILockGrain>(SingleGrain.Id); |
||||
|
} |
||||
|
|
||||
|
public async Task ReadLockAsync(string key, DateTime from, DateTime to, Stream stream) |
||||
|
{ |
||||
|
string releaseToken = null; |
||||
|
|
||||
|
using (var cts = new CancellationTokenSource(LockWaitingTime)) |
||||
|
{ |
||||
|
while (!cts.IsCancellationRequested) |
||||
|
{ |
||||
|
releaseToken = await lockGrain.AcquireLockAsync(key); |
||||
|
|
||||
|
if (releaseToken != null) |
||||
|
{ |
||||
|
break; |
||||
|
} |
||||
|
|
||||
|
await Task.Delay(2000); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
try |
||||
|
{ |
||||
|
await inner.ReadLockAsync(key, from, to, stream); |
||||
|
} |
||||
|
finally |
||||
|
{ |
||||
|
await lockGrain.ReleaseLockAsync(releaseToken); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,19 @@ |
|||||
|
// ==========================================================================
|
||||
|
// Squidex Headless CMS
|
||||
|
// ==========================================================================
|
||||
|
// Copyright (c) Squidex UG (haftungsbeschraenkt)
|
||||
|
// All rights reserved. Licensed under the MIT license.
|
||||
|
// ==========================================================================
|
||||
|
|
||||
|
using System.Threading.Tasks; |
||||
|
using Orleans; |
||||
|
|
||||
|
namespace Squidex.Infrastructure.Orleans |
||||
|
{ |
||||
|
public interface ILockGrain : IGrainWithStringKey |
||||
|
{ |
||||
|
Task<string> AcquireLockAsync(string key); |
||||
|
|
||||
|
Task ReleaseLockAsync(string releaseToken); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,46 @@ |
|||||
|
// ==========================================================================
|
||||
|
// Squidex Headless CMS
|
||||
|
// ==========================================================================
|
||||
|
// Copyright (c) Squidex UG (haftungsbeschraenkt)
|
||||
|
// All rights reserved. Licensed under the MIT license.
|
||||
|
// ==========================================================================
|
||||
|
|
||||
|
using System; |
||||
|
using System.Collections.Generic; |
||||
|
using System.Linq; |
||||
|
using System.Threading.Tasks; |
||||
|
using Squidex.Infrastructure.Tasks; |
||||
|
|
||||
|
namespace Squidex.Infrastructure.Orleans |
||||
|
{ |
||||
|
public sealed class LockGrain : GrainOfString, ILockGrain |
||||
|
{ |
||||
|
private readonly Dictionary<string, string> locks = new Dictionary<string, string>(); |
||||
|
|
||||
|
public Task<string> AcquireLockAsync(string key) |
||||
|
{ |
||||
|
string releaseToken = null; |
||||
|
|
||||
|
if (!locks.ContainsKey(key)) |
||||
|
{ |
||||
|
releaseToken = Guid.NewGuid().ToString(); |
||||
|
|
||||
|
locks.Add(key, releaseToken); |
||||
|
} |
||||
|
|
||||
|
return Task.FromResult(releaseToken); |
||||
|
} |
||||
|
|
||||
|
public Task ReleaseLockAsync(string releaseToken) |
||||
|
{ |
||||
|
var key = locks.FirstOrDefault(x => x.Value == releaseToken).Key; |
||||
|
|
||||
|
if (!string.IsNullOrWhiteSpace(key)) |
||||
|
{ |
||||
|
locks.Remove(key); |
||||
|
} |
||||
|
|
||||
|
return TaskHelper.Done; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,60 @@ |
|||||
|
// ==========================================================================
|
||||
|
// Squidex Headless CMS
|
||||
|
// ==========================================================================
|
||||
|
// Copyright (c) Squidex UG (haftungsbeschraenkt)
|
||||
|
// All rights reserved. Licensed under the MIT license.
|
||||
|
// ==========================================================================
|
||||
|
|
||||
|
using System; |
||||
|
using System.IO; |
||||
|
using System.Threading.Tasks; |
||||
|
using FakeItEasy; |
||||
|
using Orleans; |
||||
|
using Squidex.Infrastructure.Orleans; |
||||
|
using Xunit; |
||||
|
|
||||
|
namespace Squidex.Infrastructure.Log |
||||
|
{ |
||||
|
public class LockingLogStoreTests |
||||
|
{ |
||||
|
private readonly IGrainFactory grainFactory = A.Fake<IGrainFactory>(); |
||||
|
private readonly ILockGrain lockGrain = A.Fake<ILockGrain>(); |
||||
|
private readonly ILogStore inner = A.Fake<ILogStore>(); |
||||
|
private readonly LockingLogStore sut; |
||||
|
|
||||
|
public LockingLogStoreTests() |
||||
|
{ |
||||
|
A.CallTo(() => grainFactory.GetGrain<ILockGrain>(SingleGrain.Id, null)) |
||||
|
.Returns(lockGrain); |
||||
|
|
||||
|
sut = new LockingLogStore(inner, grainFactory); |
||||
|
} |
||||
|
|
||||
|
[Fact] |
||||
|
public async Task Should_lock_and_call_inner() |
||||
|
{ |
||||
|
var stream = new MemoryStream(); |
||||
|
|
||||
|
var dateFrom = DateTime.Today; |
||||
|
var dateTo = dateFrom.AddDays(2); |
||||
|
|
||||
|
var key = "MyKey"; |
||||
|
|
||||
|
var releaseToken = Guid.NewGuid().ToString(); |
||||
|
|
||||
|
A.CallTo(() => lockGrain.AcquireLockAsync(key)) |
||||
|
.Returns(releaseToken); |
||||
|
|
||||
|
await sut.ReadLockAsync(key, dateFrom, dateTo, stream); |
||||
|
|
||||
|
A.CallTo(() => lockGrain.AcquireLockAsync(key)) |
||||
|
.MustHaveHappened(); |
||||
|
|
||||
|
A.CallTo(() => lockGrain.ReleaseLockAsync(releaseToken)) |
||||
|
.MustHaveHappened(); |
||||
|
|
||||
|
A.CallTo(() => inner.ReadLockAsync(key, dateFrom, dateTo, stream)) |
||||
|
.MustHaveHappened(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,50 @@ |
|||||
|
// ==========================================================================
|
||||
|
// Squidex Headless CMS
|
||||
|
// ==========================================================================
|
||||
|
// Copyright (c) Squidex UG (haftungsbeschraenkt)
|
||||
|
// All rights reserved. Licensed under the MIT license.
|
||||
|
// ==========================================================================
|
||||
|
|
||||
|
using System.Threading.Tasks; |
||||
|
using Xunit; |
||||
|
|
||||
|
namespace Squidex.Infrastructure.Orleans |
||||
|
{ |
||||
|
public class LockGrainTests |
||||
|
{ |
||||
|
private readonly LockGrain sut = new LockGrain(); |
||||
|
|
||||
|
[Fact] |
||||
|
public async Task Should_not_acquire_lock_when_locked() |
||||
|
{ |
||||
|
var releaseLock1 = await sut.AcquireLockAsync("Key1"); |
||||
|
var releaseLock2 = await sut.AcquireLockAsync("Key1"); |
||||
|
|
||||
|
Assert.NotNull(releaseLock1); |
||||
|
Assert.Null(releaseLock2); |
||||
|
} |
||||
|
|
||||
|
[Fact] |
||||
|
public async Task Should_acquire_lock_with_other_key() |
||||
|
{ |
||||
|
var releaseLock1 = await sut.AcquireLockAsync("Key1"); |
||||
|
var releaseLock2 = await sut.AcquireLockAsync("Key2"); |
||||
|
|
||||
|
Assert.NotNull(releaseLock1); |
||||
|
Assert.NotNull(releaseLock2); |
||||
|
} |
||||
|
|
||||
|
[Fact] |
||||
|
public async Task Should_acquire_lock_after_released() |
||||
|
{ |
||||
|
var releaseLock1 = await sut.AcquireLockAsync("Key1"); |
||||
|
|
||||
|
await sut.ReleaseLockAsync(releaseLock1); |
||||
|
|
||||
|
var releaseLock2 = await sut.AcquireLockAsync("Key1"); |
||||
|
|
||||
|
Assert.NotNull(releaseLock1); |
||||
|
Assert.NotNull(releaseLock2); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue