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