making interactive mode a env / cli option

Author: thefringeninja

src/SqlStreamStore.HAL.DevServer/Program.cs

@@ -1,24 +1,132 @@
 namespace SqlStreamStore.HAL.DevServer
 {
+    using System;
+    using System.Linq;
     using System.Threading;
     using System.Threading.Tasks;
     using Microsoft.AspNetCore.Hosting;
+    using Microsoft.Extensions.Configuration;
+    using SqlStreamStore.Streams;
 
-    internal static class Program
+    internal class Program : IDisposable
     {
+        private static readonly Random s_random = new Random();
+        private readonly CancellationTokenSource _cts;
+        private readonly InMemoryStreamStore _streamStore;
+        private readonly IWebHost _host;
+        private readonly IConfigurationRoot _configuration;
+
+        private bool Interactive => _configuration.GetValue<bool>("interactive");
+
         public static async Task<int> Main(string[] args)
         {
-            using(var cts = new CancellationTokenSource())
-            using(var streamStore = new InMemoryStreamStore())
-            using(var host = new WebHostBuilder()
+            using(var program = new Program(args))
+            {
+                return await program.Run();
+            }
+        }
+
+        private Program(string[] args)
+        {
+            _cts = new CancellationTokenSource();
+            _streamStore = new InMemoryStreamStore();
+            _host = new WebHostBuilder()
                 .UseKestrel()
-                .UseStartup(new DevServerStartup(streamStore))
-                .Build())
+                .UseStartup(new DevServerStartup(_streamStore))
+                .Build();
+            _configuration = new ConfigurationBuilder()
+                .AddEnvironmentVariables()
+                .AddCommandLine(args)
+                .Build();
+        }
+
+        private async Task<int> Run()
+        {
+            try
+            {
+                var serverTask = _host.RunAsync(_cts.Token);
+
+                if(Interactive)
+                {
+                    DisplayMenu(_streamStore);
+                }
+
+                await serverTask;
+
+                return 0;
+            }
+            catch(Exception ex)
+            {
+                Console.Error.WriteLine(ex.Message);
+            }
+
+            return 1;
+        }
+
+        private static void DisplayMenu(IStreamStore streamStore, string url = null)
+        {
+            while(true)
             {
-                await host.RunAsync(cts.Token);
+                Console.WriteLine("Using stream store: {0}", streamStore.GetType().Name);
+                Console.WriteLine("Press w to write 10 messages each to 100 streams");
+                Console.WriteLine("Press t to write 100 messages each to 10 streams");
+                Console.WriteLine("Press ESC to exit");
+
+                var key = Console.ReadKey();
+
+                switch(key.Key)
+                {
+                    case ConsoleKey.Escape:
+                        return;
+                    case ConsoleKey.W:
+                        Write(streamStore, url, 10, 100);
+                        break;
+                    case ConsoleKey.T:
+                        Write(streamStore, url, 100, 10);
+                        break;
+                    default:
+                        Console.WriteLine("Computer says no");
+                        break;
+                }
             }
+        }
+
+        private static void Write(IStreamStore streamStore, string url, int messageCount, int streamCount)
+        {
+            var streams = Enumerable.Range(0, streamCount).Select(_ => $"test-{Guid.NewGuid():n}").ToList();
 
-            return 0;
+            var streamIds = string.Join("\n", streams.Select(streamid => $"{url}streams/{streamid}"));
+
+            Console.WriteLine("\nAbout to create the following streams: ");
+            Console.WriteLine(streamIds);
+
+            Task.Run(() => Task.WhenAll(
+                from streamId in streams
+                select streamStore.AppendToStream(streamId,
+                    ExpectedVersion.NoStream,
+                    GenerateMessages(messageCount))));
+        }
+
+        private static NewStreamMessage[] GenerateMessages(int messageCount)
+        {
+            return Enumerable.Range(0, messageCount)
+                .Select(_ => new NewStreamMessage(
+                    Guid.NewGuid(),
+                    "test",
+                    $@"{{ ""foo"": ""{Guid.NewGuid()}"", ""baz"": {{  }}, ""qux"": [ {
+                            string.Join(", ",
+                                Enumerable
+                                    .Range(0, messageCount).Select(max => s_random.Next(max)))
+                        } ] }}",
+                    "{}"))
+                .ToArray();
+        }
+
+        public void Dispose()
+        {
+            _host?.Dispose();
+            _streamStore?.Dispose();
+            _cts?.Dispose();
         }
     }
 }

src/SqlStreamStore.HAL.DevServer/SqlStreamStore.HAL.DevServer.csproj

@@ -8,6 +8,8 @@
     <PackageReference Include="Microsoft.AspNetCore.Hosting" Version="2.0.2" />
     <PackageReference Include="Microsoft.AspNetCore.Server.Kestrel" Version="2.0.2" />
     <PackageReference Include="Microsoft.AspNetCore.ResponseCompression" Version="2.0.2" />
+    <PackageReference Include="Microsoft.Extensions.Configuration.Binder" Version="2.0.2" />
+    <PackageReference Include="Microsoft.Extensions.Configuration.CommandLine" Version="2.0.2" />
   </ItemGroup>
   <ItemGroup>
     <ProjectReference Include="..\SqlStreamStore.HAL\SqlStreamStore.HAL.csproj" />