Add Benchmark

atmina · Sep 6, 2024 · c817835 · c817835
1 parent 1decda7
commit c817835
Show file tree

Hide file tree

Showing 5 changed files with 119 additions and 5 deletions.
diff --git a/src/Benchmark/Bench.cs b/src/Benchmark/Bench.cs
@@ -0,0 +1,76 @@
+using BenchmarkDotNet.Attributes;
+using Fluss;
+using Fluss.Authentication;
+using Fluss.Events;
+using Fluss.ReadModel;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace Benchmark;
+
+[SimpleJob]
+[MemoryDiagnoser]
+public class Bench
+{
+    [IterationSetup]
+    public void Setup()
+    {
+        var sc = new ServiceCollection();
+        sc.AddEventSourcing();
+        sc.AddPoliciesFrom(typeof(Bench).Assembly);
+
+        _sp = sc.BuildServiceProvider();
+    }
+
+    ServiceProvider _sp = null!;
+
+    [Benchmark]
+    public async Task<int> PublishEventsAndRead()
+    {
+        var sum = 0;
+        for (var j = 0; j < 1000; j++)
+        {
+            await _sp.GetSystemUserUnitOfWorkFactory().Commit(async unitOfWork => {
+                for (var i = 0; i < 50; i++)
+                {
+                    await unitOfWork.Publish(new TestEvent(1));
+                    await unitOfWork.Publish(new TestEvent(2));
+                }
+            });
+
+            var unitOfWork = _sp.GetSystemUserUnitOfWork();
+            var readModel1 = await unitOfWork.GetReadModel<TestReadModel, int>(1);
+            var readModel2 = await unitOfWork.GetReadModel<TestReadModel, int>(2);
+            sum += readModel1.GotEvents + readModel2.GotEvents;
+        }
+
+        return sum;
+    }
+}
+
+public class AllowAllPolicy : Policy {
+    public ValueTask<bool> AuthenticateEvent(EventEnvelope envelope, IAuthContext authContext)
+    {
+        return ValueTask.FromResult(true);
+    }
+
+    public ValueTask<bool> AuthenticateReadModel(IReadModel readModel, IAuthContext authContext)
+    {
+        return ValueTask.FromResult(true);
+    }
+}
+
+public record TestEvent(int Id) : Event;
+
+public record TestReadModel : ReadModelWithKey<int>
+{
+    public int GotEvents { get; private init; }
+
+    protected override TestReadModel When(EventEnvelope envelope)
+    {
+        return envelope.Event switch
+        {
+            TestEvent testEvent when testEvent.Id == Id => this with { GotEvents = GotEvents + 1 },
+            _ => this,
+        };
+    }
+} 
diff --git a/src/Benchmark/Benchmark.csproj b/src/Benchmark/Benchmark.csproj
@@ -0,0 +1,19 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+    <PropertyGroup>
+        <OutputType>Exe</OutputType>
+        <TargetFramework>net8.0</TargetFramework>
+        <ImplicitUsings>enable</ImplicitUsings>
+        <Nullable>enable</Nullable>
+    </PropertyGroup>
+
+    <ItemGroup>
+      <PackageReference Include="BenchmarkDotNet" Version="0.14.0" />
+      <PackageReference Include="Enterprize1.BenchmarkDotNet.GitCompare" Version="0.0.1" />
+    </ItemGroup>
+
+    <ItemGroup>
+        <ProjectReference Include="..\Fluss\Fluss.csproj" />
+    </ItemGroup>
+
+</Project>
diff --git a/src/Benchmark/Program.cs b/src/Benchmark/Program.cs
@@ -0,0 +1,4 @@
+using Benchmark;
+using BenchmarkDotNet.Running;
+
+BenchmarkRunner.Run<Bench>();
diff --git a/src/Fluss.sln b/src/Fluss.sln
@@ -12,6 +12,8 @@ Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Fluss.Testing", "Fluss.Test
 EndProject
 Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Fluss.Regen", "Fluss.Regen\Fluss.Regen.csproj", "{6E250321-6993-4B94-8600-85E15BF713FA}"
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Benchmark", "Benchmark\Benchmark.csproj", "{970AB461-4D44-4B99-AACA-7DAA569D4C34}"
+EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|Any CPU = Debug|Any CPU
@@ -42,5 +44,9 @@ Global
 		{6E250321-6993-4B94-8600-85E15BF713FA}.Debug|Any CPU.Build.0 = Debug|Any CPU
 		{6E250321-6993-4B94-8600-85E15BF713FA}.Release|Any CPU.ActiveCfg = Release|Any CPU
 		{6E250321-6993-4B94-8600-85E15BF713FA}.Release|Any CPU.Build.0 = Release|Any CPU
+		{970AB461-4D44-4B99-AACA-7DAA569D4C34}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{970AB461-4D44-4B99-AACA-7DAA569D4C34}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{970AB461-4D44-4B99-AACA-7DAA569D4C34}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{970AB461-4D44-4B99-AACA-7DAA569D4C34}.Release|Any CPU.Build.0 = Release|Any CPU
 	EndGlobalSection
 EndGlobal
diff --git a/src/Fluss/UnitOfWork/UnitOfWork.Authorization.cs b/src/Fluss/UnitOfWork/UnitOfWork.Authorization.cs
@@ -1,9 +1,6 @@
 using Fluss.Authentication;
 using Fluss.Events;
 using Fluss.ReadModel;
-#if !DEBUG
-using Fluss.Extensions;
-#endif
 
 namespace Fluss;
 
@@ -21,7 +18,7 @@ private async ValueTask<bool> AuthorizeUsage(EventEnvelope envelope)
 
         return accepted.Count > 0;
 #else
-        return await _policies.Select(p => p.AuthenticateEvent(envelope, ac)).AnyAsync();
+        return await AnyAsync(_policies.Select(p => p.AuthenticateEvent(envelope, ac)));
 #endif
     }
 
@@ -37,7 +34,19 @@ private async ValueTask<bool> AuthorizeUsage(IReadModel eventListener)
 
         return accepted.Count > 0;
 #else
-        return await _policies.Select(p => p.AuthenticateReadModel(eventListener, ac)).AnyAsync();
+        return await AnyAsync(_policies.Select(p => p.AuthenticateReadModel(eventListener, ac)));
 #endif
     }
+
+    private static async ValueTask<bool> AnyAsync(IEnumerable<ValueTask<bool>> valueTasks)
+    {
+        foreach (var valueTask in valueTasks)
+        {
+            if (await valueTask)
+            {
+                return true;
+            }
+        }
+        return false;
+    }
 }