mirror of https://github.com/Squidex/squidex.git
7 changed files with 199 additions and 13 deletions
@ -0,0 +1,173 @@ |
|||
// ==========================================================================
|
|||
// HandleEvents.cs
|
|||
// Squidex Headless CMS
|
|||
// ==========================================================================
|
|||
// Copyright (c) Squidex Group
|
|||
// All rights reserved.
|
|||
// ==========================================================================
|
|||
|
|||
using System; |
|||
using System.Collections.Generic; |
|||
using System.Threading.Tasks; |
|||
using MongoDB.Driver; |
|||
using Newtonsoft.Json; |
|||
using Squidex.Infrastructure; |
|||
using Squidex.Infrastructure.CQRS.Events; |
|||
using Squidex.Infrastructure.Json; |
|||
using Squidex.Infrastructure.Log; |
|||
using Squidex.Infrastructure.MongoDb; |
|||
using Squidex.Infrastructure.MongoDb.EventStore; |
|||
using Squidex.Infrastructure.Tasks; |
|||
|
|||
// ReSharper disable InvertIf
|
|||
|
|||
namespace Benchmarks.Tests |
|||
{ |
|||
public sealed class HandleEvents : IBenchmark |
|||
{ |
|||
[TypeName("MyEvent")] |
|||
public sealed class MyEvent : IEvent |
|||
{ |
|||
public int EventNumber { get; set; } |
|||
} |
|||
|
|||
public sealed class MyEventConsumer : IEventConsumer |
|||
{ |
|||
private readonly TaskCompletionSource<object> completion = new TaskCompletionSource<object>(); |
|||
private readonly int numEvents; |
|||
|
|||
public List<int> EventNumbers { get; } = new List<int>(); |
|||
|
|||
public string Name |
|||
{ |
|||
get { return typeof(MyEventConsumer).Name; } |
|||
} |
|||
|
|||
public string EventsFilter |
|||
{ |
|||
get { return string.Empty; } |
|||
} |
|||
|
|||
public MyEventConsumer(int numEvents) |
|||
{ |
|||
this.numEvents = numEvents; |
|||
} |
|||
|
|||
public Task ClearAsync() |
|||
{ |
|||
return TaskHelper.Done; |
|||
} |
|||
|
|||
public void Wait() |
|||
{ |
|||
completion.Task.Wait(); |
|||
} |
|||
|
|||
public Task On(Envelope<IEvent> @event) |
|||
{ |
|||
if (@event.Payload is MyEvent myEvent) |
|||
{ |
|||
EventNumbers.Add(myEvent.EventNumber); |
|||
|
|||
if (myEvent.EventNumber == numEvents) |
|||
{ |
|||
completion.SetResult(true); |
|||
} |
|||
} |
|||
|
|||
return TaskHelper.Done; |
|||
} |
|||
} |
|||
|
|||
private readonly TypeNameRegistry typeNameRegistry = new TypeNameRegistry().Map(typeof(MyEvent)); |
|||
private readonly EventDataFormatter formatter; |
|||
private readonly JsonSerializerSettings serializerSettings = new JsonSerializerSettings(); |
|||
private const int NumEvents = 1000; |
|||
private IMongoClient mongoClient; |
|||
private IMongoDatabase mongoDatabase; |
|||
private IEventStore eventStore; |
|||
private IEventNotifier eventNotifier; |
|||
private IEventConsumerInfoRepository eventConsumerInfos; |
|||
private EventReceiver eventReceiver; |
|||
private MyEventConsumer eventConsumer; |
|||
|
|||
public string Id |
|||
{ |
|||
get { return "handleEvents"; } |
|||
} |
|||
|
|||
public string Name |
|||
{ |
|||
get { return "HandleEvents"; } |
|||
} |
|||
|
|||
public HandleEvents() |
|||
{ |
|||
serializerSettings.Converters.Add(new PropertiesBagConverter()); |
|||
|
|||
formatter = new EventDataFormatter(typeNameRegistry, serializerSettings); |
|||
} |
|||
|
|||
public void Initialize() |
|||
{ |
|||
mongoClient = new MongoClient("mongodb://localhost"); |
|||
} |
|||
|
|||
public void RunInitialize() |
|||
{ |
|||
mongoDatabase = mongoClient.GetDatabase(Guid.NewGuid().ToString()); |
|||
|
|||
var log = new SemanticLog(new ILogChannel[0], new ILogAppender[0], () => new JsonLogWriter(Formatting.Indented, true)); |
|||
|
|||
eventConsumerInfos = new MongoEventConsumerInfoRepository(mongoDatabase); |
|||
eventNotifier = new DefaultEventNotifier(new InMemoryPubSub()); |
|||
eventStore = new MongoEventStore(mongoDatabase, eventNotifier); |
|||
eventConsumer = new MyEventConsumer(NumEvents); |
|||
|
|||
eventReceiver = new EventReceiver(formatter, eventStore, eventNotifier, eventConsumerInfos, log); |
|||
eventReceiver.Subscribe(eventConsumer); |
|||
} |
|||
|
|||
public long Run() |
|||
{ |
|||
var streamName = Guid.NewGuid().ToString(); |
|||
|
|||
for (var eventId = 0; eventId < NumEvents; eventId++) |
|||
{ |
|||
var eventData = formatter.ToEventData(new Envelope<IEvent>(new MyEvent { EventNumber = eventId + 1 }), Guid.NewGuid()); |
|||
|
|||
eventStore.AppendEventsAsync(Guid.NewGuid(), streamName, eventId - 1, new [] { eventData }).Wait(); |
|||
} |
|||
|
|||
eventConsumer.Wait(); |
|||
|
|||
if (eventConsumer.EventNumbers.Count != NumEvents) |
|||
{ |
|||
throw new InvalidOperationException($"Only {eventConsumer.EventNumbers.Count} have been handled"); |
|||
} |
|||
|
|||
for (var i = 0; i < eventConsumer.EventNumbers.Count; i++) |
|||
{ |
|||
var value = eventConsumer.EventNumbers[i]; |
|||
|
|||
if (value != i + 1) |
|||
{ |
|||
throw new InvalidOperationException($"Event[{i}] != value"); |
|||
} |
|||
} |
|||
|
|||
return NumEvents; |
|||
} |
|||
|
|||
public void RunCleanup() |
|||
{ |
|||
mongoClient.DropDatabase(mongoDatabase.DatabaseNamespace.DatabaseName); |
|||
|
|||
eventReceiver.Dispose(); |
|||
} |
|||
|
|||
public void Cleanup() |
|||
{ |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue