Implement generic host

Directory.Packages.props

@@ -9,6 +9,7 @@
     <PackageVersion Include="CommandLineParser" Version="2.9.1" />
     <PackageVersion Include="JsonPath.Net" Version="1.1.6" />
     <PackageVersion Include="KeraLua" Version="1.4.1" />
+    <PackageVersion Include="Microsoft.Extensions.Hosting.Abstractions" Version="8.0.1" />
     <PackageVersion Include="NUnit" Version="4.1.0" />
     <PackageVersion Include="NUnit3TestAdapter" Version="4.6.0" />
     <PackageVersion Include="Microsoft.CodeAnalysis" Version="4.11.0" />

benchmark/BDN.benchmark/BDN.benchmark.csproj

@@ -29,4 +29,8 @@
     <Compile Include="..\..\main\GarnetServer\Extensions\SetIfPM.cs" Link="Custom\SetIfPM.cs" />
   </ItemGroup>
 
+  <ItemGroup>
+    <Folder Include="BenchmarkDotNet.Artifacts\" />
+  </ItemGroup>
+
 </Project>

benchmark/BDN.benchmark/Cluster/ClusterContext.cs

@@ -10,9 +10,9 @@
 
 namespace BDN.benchmark.Cluster
 {
-    unsafe class ClusterContext
+    class ClusterContext
     {
-        EmbeddedRespServer server;
+        GarnetEmbeddedApplication server;
         RespServerSession session;
         readonly BenchUtils benchUtils = new();
         readonly int port = 7000;
@@ -22,9 +22,11 @@ unsafe class ClusterContext
         public Request[] singleMGetMSet;
         public Request singleCTXNSET;
 
-        public void Dispose()
+        public async Task Dispose()
         {
             session.Dispose();
+            await server.StopAsync();
+            server.Dispose();
             server.Dispose();
         }
 
@@ -39,12 +41,17 @@ public void SetupSingleInstance(bool disableSlotVerification = false)
             };
             if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
                 opt.CheckpointDir = "/tmp";
-            server = new EmbeddedRespServer(opt);
+
+            var builder = GarnetEmbeddedApplication.CreateHostBuilder([], opt);
+
+            server = builder.Build();
+
             session = server.GetRespSession();
+
             _ = server.Register.NewTransactionProc(CustomTxnSet.CommandName, () => new CustomTxnSet(), new RespCommandsInfo { Arity = CustomTxnSet.Arity });
         }
 
-        public void AddSlotRange(List<(int, int)> slotRanges)
+        public unsafe void AddSlotRange(List<(int, int)> slotRanges)
         {
             foreach (var slotRange in slotRanges)
             {
@@ -56,7 +63,7 @@ public void AddSlotRange(List<(int, int)> slotRanges)
             }
         }
 
-        public void CreateGetSet(int keySize = 8, int valueSize = 32, int batchSize = 100)
+        public unsafe void CreateGetSet(int keySize = 8, int valueSize = 32, int batchSize = 100)
         {
             var pairs = new (byte[], byte[])[batchSize];
             for (var i = 0; i < batchSize; i++)
@@ -93,7 +100,7 @@ public void CreateGetSet(int keySize = 8, int valueSize = 32, int batchSize = 10
             singleGetSet = [getReq, setReq];
         }
 
-        public void CreateMGetMSet(int keySize = 8, int valueSize = 32, int batchSize = 100)
+        public unsafe void CreateMGetMSet(int keySize = 8, int valueSize = 32, int batchSize = 100)
         {
             var pairs = new (byte[], byte[])[batchSize];
             for (var i = 0; i < batchSize; i++)
@@ -134,7 +141,7 @@ public void CreateMGetMSet(int keySize = 8, int valueSize = 32, int batchSize =
             singleMGetMSet = [mGetReq, mSetReq];
         }
 
-        public void CreateCTXNSET(int keySize = 8, int batchSize = 100)
+        public unsafe void CreateCTXNSET(int keySize = 8, int batchSize = 100)
         {
             var keys = new byte[8][];
             for (var i = 0; i < 8; i++)
@@ -163,7 +170,7 @@ public void CreateCTXNSET(int keySize = 8, int batchSize = 100)
             singleCTXNSET = ctxnsetReq;
         }
 
-        public void Consume(byte* ptr, int length)
+        public unsafe void Consume(byte* ptr, int length)
             => session.TryConsumeMessages(ptr, length);
     }
 

benchmark/BDN.benchmark/Embedded/GarnetEmbeddedApplication.cs

@@ -0,0 +1,77 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+
+using System;
+using System.Linq;
+using System.Threading;
+using System.Threading.Tasks;
+using Garnet;
+using Garnet.common;
+using Garnet.server;
+using Microsoft.Extensions.DependencyInjection;
+using Tsavorite.core;
+
+namespace Embedded.server;
+
+internal sealed class GarnetEmbeddedApplication
+{
+    public RegisterApi Register => app.Register;
+
+    readonly StoreWrapper store;
+    readonly SubscribeBroker<SpanByte, SpanByte, IKeySerializer<SpanByte>> subscriberBroker;
+    readonly GarnetApplication app;
+
+    public GarnetEmbeddedApplication(GarnetApplication app)
+    {
+        this.app = app;
+        this.store = app.Services.GetRequiredService<StoreWrapper>();
+        this.subscriberBroker =
+            app.Services.GetRequiredService<SubscribeBroker<SpanByte, SpanByte, IKeySerializer<SpanByte>>>();
+    }
+
+    public static GarnetEmbeddedApplicationBuilder CreateHostBuilder(string[] args, GarnetServerOptions options)
+    {
+        return new(new GarnetApplicationOptions
+        {
+            Args = args,
+        }, options);
+    }
+
+    public async Task StopAsync(CancellationToken cancellationToken = default)
+    {
+        await app.StopAsync(cancellationToken);
+    }
+
+    /// <summary>
+    /// Dispose server
+    /// </summary>
+    public void Dispose() => app.Dispose();
+
+    public StoreWrapper StoreWrapper => store;
+
+    /// <summary>
+    /// Return a direct RESP session to this server
+    /// </summary>
+    /// <returns>A new RESP server session</returns>
+    internal RespServerSession GetRespSession()
+    {
+        return new RespServerSession(0,
+            new EmbeddedNetworkSender(),
+            store,
+            subscribeBroker: this.subscriberBroker,
+            null,
+            true);
+    }
+
+    internal EmbeddedNetworkHandler GetNetworkHandler()
+    {
+        var server = app.Services
+            .GetServices<IGarnetServer>()
+            .OfType<GarnetServerEmbedded>()
+            .FirstOrDefault();
+
+        Console.WriteLine(server);
+
+        return server?.CreateNetworkHandler();
+    }
+}

benchmark/BDN.benchmark/Embedded/GarnetEmbeddedApplicationBuilder.cs

@@ -0,0 +1,40 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+
+using System.Linq;
+using Garnet;
+using Garnet.server;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace Embedded.server;
+
+internal class GarnetEmbeddedApplicationBuilder : GarnetApplicationBuilder
+{
+    internal GarnetEmbeddedApplicationBuilder(GarnetApplicationOptions options, GarnetServerOptions garnetServerOptions)
+        : base(options, garnetServerOptions)
+    {
+    }
+
+    public new GarnetEmbeddedApplication Build()
+    {
+        {
+            var serviceDescriptor = base.Services
+                .FirstOrDefault(descriptor => descriptor.ServiceType == typeof(StoreWrapper));
+
+            base.Services.Remove(serviceDescriptor);
+        }
+
+        {
+            var serviceDescriptor = base.Services
+                .FirstOrDefault(descriptor => descriptor.ServiceType == typeof(IGarnetServer));
+
+            base.Services.Remove(serviceDescriptor);
+
+            base.Services.AddSingleton<IGarnetServer, GarnetServerEmbedded>();
+        }
+
+        var app = base.Build();
+
+        return new GarnetEmbeddedApplication(app);
+    }
+}