Merge pull request #188 from Mpdreamz/patch-1

Implement IAsyncDisposable on CompositeDisposableServiceProvider

Author: neuecc

src/ConsoleAppFramework/Emitter.cs

@@ -838,7 +838,7 @@ public void EmitAsConsoleAppBuilder(SourceBuilder sb, DllReference dllReference)
 internal static class ConsoleAppHostBuilderExtensions
 {
     class CompositeDisposableServiceProvider(IDisposable host, IServiceProvider serviceServiceProvider, IDisposable scope, IServiceProvider serviceProvider)
-        : IServiceProvider, IKeyedServiceProvider, IDisposable
+        : IServiceProvider, IKeyedServiceProvider, IDisposable, IAsyncDisposable
     {
         public object? GetService(Type serviceType)
         {
@@ -868,6 +868,25 @@ public void Dispose()
             }
             host.Dispose();
         }
+        
+        public async ValueTask DisposeAsync()
+        {
+            await CastAndDispose(host);
+            await CastAndDispose(scope);
+            await CastAndDispose(serviceProvider);
+            await CastAndDispose(serviceServiceProvider);
+            GC.SuppressFinalize(this);
+            
+            return;
+            
+            static async ValueTask CastAndDispose<T>(T resource)
+            {
+                if (resource is IAsyncDisposable resourceAsyncDisposable)
+                    await resourceAsyncDisposable.DisposeAsync();
+                else if (resource is IDisposable resourceDisposable)
+                    resourceDisposable.Dispose();
+            }
+        }
     }
 
     internal static ConsoleApp.ConsoleAppBuilder ToConsoleAppBuilder(this IHostBuilder hostBuilder)