added WebSockets example

examples/Demo/Demo.csproj

@@ -39,12 +39,13 @@
         <PackageReference Include="Bogus" Version="34.0.2" />
         <PackageReference Include="Dapper.Contrib" Version="2.0.78" />
         <PackageReference Include="LiteDB" Version="5.0.15" />
-        <PackageReference Include="MathNet.Numerics" Version="5.0.0" />
+        <PackageReference Include="MathNet.Numerics" Version="5.0.0" />        
         <PackageReference Include="NBomber" Version="5.5.0-beta.0" />
         <PackageReference Include="NBomber.Data" Version="5.0.0" />
         <PackageReference Include="NBomber.Http" Version="5.0.1" />
         <PackageReference Include="NBomber.Sinks.InfluxDB" Version="5.0.2" />
         <PackageReference Include="MQTTnet" Version="3.1.2" />
+        <PackageReference Include="NBomber.WebSockets" Version="0.1.0-beta.1" />
         <PackageReference Include="Serilog.Sinks.Elasticsearch" Version="9.0.0" />
         <PackageReference Include="StackExchange.Redis" Version="2.6.122" />
         <PackageReference Include="System.Data.SQLite.Core" Version="1.0.117" />
@@ -70,9 +71,6 @@
         <None Update="Features\DataFeed\users-feed-data.json">
           <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
         </None>
-        <None Update="Features\ClientPool\config.json">
-          <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
-        </None>
         <None Update="Features\CliArgs\config.json">
           <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
         </None>

examples/Demo/Features/ClientPool/ClientPoolMqttExample.cs → examples/Demo/MQTT/ClientPool/ClientPoolMqttExample.cs

@@ -102,7 +102,7 @@ public void Run()
 
         NBomberRunner
             .RegisterScenarios(scenario)
-            .LoadConfig("./Features/ClientPool/config.json")
+            .LoadConfig("./MQTT/ClientPool/config.json")
             .Run();
 
     }

examples/Demo/MQTT/docker-compose.yaml

@@ -1,8 +1,8 @@
 version: '3.4'
 services:
-    
+
   emqx:
-    image: "emqx/emqx:4.2.0"
+    image: "emqx/emqx:5.0.12"
     ports:
       - "1883:1883"
       - "18083:18083"

examples/Demo/Program.cs

@@ -17,6 +17,7 @@
 using Demo.MQTT;
 using Demo.HTTP.WebAppSimulator;
 using Demo.HTTP.SimpleBookstore;
+using Demo.WebSockets;
 
 // -------------------------------
 // ----- Hello World examples -----
@@ -73,6 +74,11 @@
 // new WebAppSimulatorExample().RunHttpUserExample();
 // new ExampleSimpleBookstore().Run();
 
+// ----------------
+// ----- WebSockets -----
+// ----------------
+// new PingPongWebSocketsTest().Run();
+
 // ----------------
 // ----- MQTT -----
 // ----------------

examples/Demo/WebSockets/PingPongWebSocketsTest.cs

@@ -0,0 +1,57 @@
+using NBomber.CSharp;
+using NBomber.Data;
+using NBomber.WebSockets;
+
+namespace Demo.WebSockets;
+
+public class PingPongWebSocketsTest
+{
+    // To run this example you need to spin up local server examples/simulators/WebAppSimulator
+    // The server should run on localhost:5233, you should run http profile that configured in WebAppSimulator/Properties/launchSettings.json
+
+    public void Run()
+    {
+        var payload = Data.GenerateRandomBytes(1_000_000); // 1MB
+
+        var scenario = Scenario.Create("ping_pong_websockets", async ctx =>
+        {
+            using var websocket = new WebSocket(new WebSocketConfig());
+
+            var connect = await Step.Run("connect", ctx, async () =>
+            {
+                await websocket.Connect("ws://localhost:5233/ws");
+                return Response.Ok();
+            });
+
+            var ping = await Step.Run("ping", ctx, async () =>
+            {
+                await websocket.Send(payload);
+                return Response.Ok(sizeBytes: payload.Length);
+            });
+
+            var pong = await Step.Run("pong", ctx, async () =>
+            {
+                using var response = await websocket.Receive();
+                // var str = Encoding.UTF8.GetString(response.Data.Span);
+                // var user = JsonSerializer.Deserialize<T>(response.Data.Span);
+                return Response.Ok(sizeBytes: response.Data.Length);
+            });
+
+            var disconnect = await Step.Run("disconnect", ctx, async () =>
+            {
+                await websocket.Close();
+                return Response.Ok();
+            });
+
+            return Response.Ok();
+        })
+        .WithoutWarmUp()
+        .WithLoadSimulations(
+            Simulation.KeepConstant(10, TimeSpan.FromSeconds(30))
+        );
+
+        NBomberRunner
+            .RegisterScenarios(scenario)
+            .Run();
+    }
+}

examples/WebAppSimulator/Controllers/WebSocketController.cs

@@ -0,0 +1,58 @@
+using System.Net.WebSockets;
+using Microsoft.AspNetCore.Mvc;
+using Microsoft.IO;
+
+namespace WebAppSimulator.Controllers;
+
+public class WebSocketController : ControllerBase
+{
+    private static readonly RecyclableMemoryStreamManager MsStreamManager = new();
+    private const int BufferSize = 1024 * 16;
+
+    [Route("/ws")]
+    public async Task Get()
+    {
+        if (HttpContext.WebSockets.IsWebSocketRequest)
+        {
+            using var webSocket = await HttpContext.WebSockets.AcceptWebSocketAsync();
+            using var ms = MsStreamManager.GetStream();
+
+            await Receive(webSocket, ms);
+            await Send(webSocket, ms);
+
+            await webSocket.CloseAsync(WebSocketCloseStatus.NormalClosure, "Closing", CancellationToken.None);
+        }
+        else
+        {
+            HttpContext.Response.StatusCode = StatusCodes.Status400BadRequest;
+        }
+    }
+
+    private async Task Receive(WebSocket webSocket, RecyclableMemoryStream ms)
+    {
+        var endOfMessage = false;
+
+        while (!endOfMessage)
+        {
+            var buffer = ms.GetMemory(BufferSize);
+            var message = await webSocket.ReceiveAsync(buffer, CancellationToken.None);
+
+            if (message.MessageType == WebSocketMessageType.Close)
+            {
+                break;
+            }
+
+            ms.Advance(message.Count);
+            endOfMessage = message.EndOfMessage;
+        }
+    }
+
+    private async Task Send(WebSocket webSocket, RecyclableMemoryStream ms)
+    {
+        if (ms.Length > 0)
+        {
+            var msg = ms.GetBuffer().AsMemory(0, (int) ms.Length);
+            await webSocket.SendAsync(msg, WebSocketMessageType.Binary, WebSocketMessageFlags.EndOfMessage, CancellationToken.None);
+        }
+    }
+}

examples/WebAppSimulator/Program.cs

@@ -1,3 +1,6 @@
+using System.Buffers;
+using System.IO.Pipelines;
+using System.Text;
 using WebAppSimulator.Infra.DAL;
 
 namespace WebAppSimulator
@@ -6,11 +9,13 @@ public class SQLiteSettings
     {
         public string ConnectionString { get; set; }
     }
+
     public class RedisSettings
     {
         public string ConnectionString { get; set; }
         public string ServerName { get; set; }
     }
+
     public class Program
     {
         public static void Main(string[] args)
@@ -37,16 +42,13 @@ public static void Main(string[] args)
                 var rep = new RedisRepository(settings);
                 builder.Services.AddSingleton<IUserRepository>(rep);
             }
-            else
-                throw new Exception("The base for the test is not specified in the file appsettings.json");
 
             var app = builder.Build();
 
             if (!app.Environment.IsDevelopment())
             {
                 app.UseExceptionHandler("/Error");
             }
-            app.UseHttpsRedirection();
 
             if (app.Environment.IsDevelopment())
             {
@@ -63,12 +65,14 @@ public static void Main(string[] args)
                 app.UseSwaggerUI();
             }
 
+            // app.UseHttpsRedirection();
             app.UseRouting();
 
             app.UseAuthorization();
 
             app.MapRazorPages();
             app.MapControllers();
+            app.UseWebSockets();
 
             app.Run();
         }

examples/WebAppSimulator/WebAppSimulator.csproj

@@ -17,6 +17,7 @@
     <ItemGroup>
         <PackageReference Include="Bogus" Version="34.0.2" />
         <PackageReference Include="LiteDB" Version="5.0.15" />
+        <PackageReference Include="Microsoft.IO.RecyclableMemoryStream" Version="3.0.0" />
         <PackageReference Include="Newtonsoft.Json" Version="13.0.3" />
         <PackageReference Include="StackExchange.Redis" Version="2.6.122" />
         <PackageReference Include="System.Data.SQLite.Core" Version="1.0.117" />

examples/WebAppSimulator/appsettings.json

@@ -7,7 +7,7 @@
     },
     "AllowedHosts": "*",
 
-    "DbUse": "SQLite",
+    "DbUse": "none",
 
     "SQLiteSettings": {
         "ConnectionString": "Data Source=DB/UsersDB.db"