support IAsyncDIsposable

Cysharp · Apr 7, 2021 · 8932db8 · 8932db8
1 parent 22a073b
commit 8932db8
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/ConsoleAppFramework/ConsoleAppEngineService.cs b/src/ConsoleAppFramework/ConsoleAppEngineService.cs
@@ -85,7 +85,14 @@ public async Task StopAsync(CancellationToken cancellationToken)
             }
             finally
             {
-                scope.Dispose();
+                if (scope is IAsyncDisposable asyncDisposable)
+                {
+                    await asyncDisposable.DisposeAsync();
+                }
+                else
+                {
+                    scope.Dispose();
+                }
             }
         }
     }