diff --git a/modules/permission-management/src/Volo.Abp.PermissionManagement.Domain/Volo/Abp/PermissionManagement/AbpPermissionManagementDomainModule.cs b/modules/permission-management/src/Volo.Abp.PermissionManagement.Domain/Volo/Abp/PermissionManagement/AbpPermissionManagementDomainModule.cs index 72d2bb9eef..feb11fdee5 100644 --- a/modules/permission-management/src/Volo.Abp.PermissionManagement.Domain/Volo/Abp/PermissionManagement/AbpPermissionManagementDomainModule.cs +++ b/modules/permission-management/src/Volo.Abp.PermissionManagement.Domain/Volo/Abp/PermissionManagement/AbpPermissionManagementDomainModule.cs @@ -1,4 +1,5 @@ using System; +using System.Threading; using System.Threading.Tasks; using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.Hosting; @@ -39,7 +40,8 @@ public class AbpPermissionManagementDomainModule : AbpModule private static void SaveStaticPermissionsToDatabase(ApplicationInitializationContext context) { var rootServiceProvider = context.ServiceProvider.GetRequiredService(); - var hostApplicationLifetime = context.ServiceProvider.GetRequiredService(); + var hostApplicationLifetime = context.ServiceProvider.GetService(); + var cancellationToken = hostApplicationLifetime?.ApplicationStopping ?? CancellationToken.None; Task.Run(async () => { @@ -48,7 +50,7 @@ public class AbpPermissionManagementDomainModule : AbpModule try { - using (cancellationTokenProvider.Use(hostApplicationLifetime.ApplicationStopping)) + using (cancellationTokenProvider.Use(cancellationToken)) { await Policy .Handle()