added Benchmark for Channel and Ply

PragmaticFlow · Nov 21, 2022 · 2ae14df · 2ae14df
1 parent c5c0961
commit 2ae14df
Show file tree

Hide file tree

Showing 7 changed files with 118 additions and 5 deletions.
diff --git a/NBomber.Performance.sln b/NBomber.Performance.sln
@@ -2,6 +2,8 @@
 Microsoft Visual Studio Solution File, Format Version 12.00
 Project("{F2A71F9B-5D33-465A-A702-920D77279786}") = "NBomber.Benchmarks", "performance\NBomber.Benchmarks\NBomber.Benchmarks.fsproj", "{60140186-F81E-4C62-ACD9-1A327D297399}"
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "NBomber.CSharpImpl", "performance\NBomber.CSharpImpl\NBomber.CSharpImpl.csproj", "{C4632132-9C00-4A87-A235-F7A830F5853E}"
+EndProject
 Global
  GlobalSection(SolutionConfigurationPlatforms) = preSolution
  Debug|Any CPU = Debug|Any CPU
@@ -12,5 +14,9 @@ Global
  {60140186-F81E-4C62-ACD9-1A327D297399}.Debug|Any CPU.Build.0 = Debug|Any CPU
  {60140186-F81E-4C62-ACD9-1A327D297399}.Release|Any CPU.ActiveCfg = Release|Any CPU
  {60140186-F81E-4C62-ACD9-1A327D297399}.Release|Any CPU.Build.0 = Release|Any CPU
+ {C4632132-9C00-4A87-A235-F7A830F5853E}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+ {C4632132-9C00-4A87-A235-F7A830F5853E}.Debug|Any CPU.Build.0 = Debug|Any CPU
+ {C4632132-9C00-4A87-A235-F7A830F5853E}.Release|Any CPU.ActiveCfg = Release|Any CPU
+ {C4632132-9C00-4A87-A235-F7A830F5853E}.Release|Any CPU.Build.0 = Release|Any CPU
  EndGlobalSection
 EndGlobal
diff --git a/global.json b/global.json
@@ -0,0 +1,7 @@
+{
+ "sdk": {
+ "version": "7.0.0",
+ "rollForward": "latestMajor",
+ "allowPrerelease": true
+ }
+}
diff --git a/performance/NBomber.Benchmarks/Actors/Benchmark.fs b/performance/NBomber.Benchmarks/Actors/Benchmark.fs
@@ -3,6 +3,7 @@ namespace NBomber.Benchmarks.Actors
 open System.Threading.Tasks
 open System.Threading.Tasks.Dataflow
 open BenchmarkDotNet.Attributes
+open NBomber.CSharpImpl
 
 type ActorMessage =
  | Inc
@@ -26,7 +27,7 @@ type ActorState =
 type ActorsBenchmark() as this =
 
  //[<Params(100_000, 1_000_000, 10_000_000)>]
- [<Params(1_00_000)>]
+ [<Params(50_000)>]
  member val ThreadIteration = 0 with get, set
 
  [<Params(10)>]
@@ -49,6 +50,38 @@ type ActorsBenchmark() as this =
  tcs.Task.Wait()
 
  [<Benchmark>]
+ member _.ChannelActorPly() =
+ let actor = ChannelActorPly<ActorState, ActorMessage>(ActorState.init(), ActorState.handle)
+
+ [| 1..this.ThreadCount |]
+ |> Array.Parallel.iter (fun _ ->
+ for i = 0 to this.ThreadIteration do
+ let isAdd = i % 2 = 0
+ if isAdd then actor.Publish(ActorMessage.Inc)
+ else actor.Publish(ActorMessage.Dec)
+ )
+
+ let tcs = TaskCompletionSource<int>()
+ actor.Publish(ActorMessage.GetValue tcs)
+ tcs.Task.Wait()
+
+ [<Benchmark>]
+ member _.ChannelActorCSharp() =
+ let actor = ChannelActorCSharp<ActorState, ActorMessage>(ActorState.init(), ActorState.handle)
+
+ [| 1..this.ThreadCount |]
+ |> Array.Parallel.iter (fun _ ->
+ for i = 0 to this.ThreadIteration do
+ let isAdd = i % 2 = 0
+ if isAdd then actor.Publish(ActorMessage.Inc)
+ else actor.Publish(ActorMessage.Dec)
+ )
+
+ let tcs = TaskCompletionSource<int>()
+ actor.Publish(ActorMessage.GetValue tcs)
+ tcs.Task.Wait()
+
+ //[<Benchmark>]
  member _.FastActor() =
  let actor = Actor.FastActor<ActorState, ActorMessage>(ActorState.init(), ActorState.handle)
 
@@ -69,7 +102,10 @@ type ActorsBenchmark() as this =
  let mutable state = ActorState.init()
 
  let actor = ActionBlock(fun msg ->
- state <- ActorState.handle state msg
+ backgroundTask {
+ state <- ActorState.handle state msg
+ }
+ :> Task
  )
 
  [| 1..this.ThreadCount |]
@@ -84,7 +120,7 @@ type ActorsBenchmark() as this =
  actor.Post(ActorMessage.GetValue tcs) |> ignore
  tcs.Task.Wait()
 
- [<Benchmark>]
+ //[<Benchmark>]
  member _.MailBox() =
  let mutable state = ActorState.init()
 

diff --git a/performance/NBomber.Benchmarks/Actors/ChannelActor.fs b/performance/NBomber.Benchmarks/Actors/ChannelActor.fs
@@ -1,14 +1,33 @@
 namespace NBomber.Benchmarks.Actors
 
+open System
 open System.Threading.Channels
+open FSharp.Control.Tasks.NonAffine
 
 type ChannelActor<'TState,'TMessage>(initialState: 'TState, handler: 'TState -> 'TMessage -> 'TState) =
 
  let _channel = Channel.CreateUnbounded<'TMessage>()
  let mutable _currentState = initialState
  let mutable _stop = false
 
- let loop () = task {
+ let loop () = backgroundTask {
+ while not _stop do
+ let! msg = _channel.Reader.ReadAsync()
+ _currentState <- handler _currentState msg
+ }
+
+ do loop() |> ignore
+
+ member _.Publish(message: 'TMessage) =
+ _channel.Writer.TryWrite(message) |> ignore
+
+type ChannelActorPly<'TState,'TMessage>(initialState: 'TState, handler: 'TState -> 'TMessage -> 'TState) =
+
+ let _channel = Channel.CreateUnbounded<'TMessage>()
+ let mutable _currentState = initialState
+ let mutable _stop = false
+
+ let loop () = vtask {
  while not _stop do
  let! msg = _channel.Reader.ReadAsync()
  _currentState <- handler _currentState msg

diff --git a/performance/NBomber.Benchmarks/NBomber.Benchmarks.fsproj b/performance/NBomber.Benchmarks/NBomber.Benchmarks.fsproj
@@ -2,7 +2,7 @@
 
  <PropertyGroup>
  <OutputType>Exe</OutputType>
- <TargetFramework>netcoreapp3.1</TargetFramework>
+ <TargetFramework>net7.0</TargetFramework>
  </PropertyGroup>
 
  <ItemGroup>
@@ -16,7 +16,13 @@
  <ItemGroup>
  <PackageReference Include="BenchmarkDotNet" Version="0.12.1" />
  <PackageReference Include="FSharpx.Collections" Version="2.1.2" />
+ <PackageReference Include="Ply" Version="0.3.1" />
  <PackageReference Include="Streams" Version="0.5.0" />
+ <PackageReference Update="FSharp.Core" Version="7.0.0" />
+ </ItemGroup>
+
+ <ItemGroup>
+ <ProjectReference Include="..\NBomber.CSharpImpl\NBomber.CSharpImpl.csproj" />
  </ItemGroup>
 
 </Project>
diff --git a/performance/NBomber.CSharpImpl/ChannelActorCSharp.cs b/performance/NBomber.CSharpImpl/ChannelActorCSharp.cs
@@ -0,0 +1,30 @@
+using System.Threading.Channels;
+
+namespace NBomber.CSharpImpl;
+
+public class ChannelActorCSharp<TState,TMessage>
+{
+ TState _currentState;
+ private bool stop = false;
+ private readonly Func<TState, TMessage, TState> _handler;
+ private readonly Channel<TMessage> _channel = Channel.CreateUnbounded<TMessage>();
+
+ public ChannelActorCSharp(TState initialState, Func<TState, TMessage, TState> handler)
+ {
+ _currentState = initialState;
+ _handler = handler;
+
+ Loop();
+ }
+
+ async Task Loop()
+ {
+ while (!stop)
+ {
+ var msg = await _channel.Reader.ReadAsync();
+ _currentState = _handler(_currentState, msg);
+ }
+ }
+
+ public void Publish(TMessage message) => _channel.Writer.TryWrite(message);
+}
diff --git a/performance/NBomber.CSharpImpl/NBomber.CSharpImpl.csproj b/performance/NBomber.CSharpImpl/NBomber.CSharpImpl.csproj
@@ -0,0 +1,9 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+ <PropertyGroup>
+ <TargetFramework>net7.0</TargetFramework>
+ <ImplicitUsings>enable</ImplicitUsings>
+ <Nullable>enable</Nullable>
+ </PropertyGroup>
+
+</Project>