开启自定义提取、强制刷新报表

performance/Performance.Api/Controllers/AllotController.cs

@@ -263,17 +263,17 @@ public ApiResponse GenerateAsync([CustomizeValidator(RuleSet = "Delete"), FromBo
             //}
             //else
             //{
-                _backgroundTaskQueue.QueueBackgroundWorkItem(async token =>
+            _backgroundTaskQueue.QueueBackgroundWorkItem(async token =>
+            {
+                using (var scope = _serviceScopeFactory.CreateScope())
                 {
-                    using (var scope = _serviceScopeFactory.CreateScope())
-                    {
-                        var scopedServices = scope.ServiceProvider.GetRequiredService<AllotService>();
-                        scopedServices.Generate(allot);
-                        await Task.Delay(TimeSpan.FromSeconds(5), token);
-                    }
-                });
-
-                //_publishEndpoint.Publish(allot).Wait();
+                    var scopedServices = scope.ServiceProvider.GetRequiredService<AllotService>();
+                    scopedServices.Generate(allot);
+                    await Task.Delay(TimeSpan.FromSeconds(5), token);
+                }
+            });
+
+            //_publishEndpoint.Publish(allot).Wait();
             //}
 
             _logManageService.WriteMsg("等待绩效生成", $"等待绩效生成{allot.Year}-{allot.Month.ToString().PadLeft(2, '0')}月份绩效!", 1, allot.ID, "ReceiveMessage");
@@ -297,7 +297,15 @@ public ApiResponse GenerateReport([CustomizeValidator(RuleSet = "Delete"), FromB
             if (null == allot || !states.Contains(allot.States))
                 throw new PerformanceException("当前绩效暂未生成，无法统计报表数据。");
 
-            //_publishEndpoint.Publish(allot).Wait();
+            _backgroundTaskQueue.QueueBackgroundWorkItem(async token =>
+            {
+                using (var scope = _serviceScopeFactory.CreateScope())
+                {
+                    var scopedServices = scope.ServiceProvider.GetRequiredService<AllotService>();
+                    scopedServices.GenerateReport(allot);
+                    await Task.Delay(TimeSpan.FromSeconds(5), token);
+                }
+            });
 
             return new ApiResponse(ResponseType.OK, "统计报表数据任务开始");
         }

performance/Performance.Api/Controllers/ModExtractController.cs

@@ -20,6 +20,7 @@ public class ModExtractController : Controller
         private readonly AllotService _allotService;
         private readonly CustomExtractService _extractService;
         private readonly IServiceScopeFactory _serviceScopeFactory;
+        private readonly IBackgroundTaskQueue _backgroundTaskQueue;
         private readonly IHubNotificationQueue _notificationQueue;
 
         public ModExtractController(
@@ -27,12 +28,14 @@ public class ModExtractController : Controller
             AllotService allotService,
             CustomExtractService extractService,
             IServiceScopeFactory serviceScopeFactory,
+            IBackgroundTaskQueue backgroundTaskQueue,
             IHubNotificationQueue notificationQueue)
         {
             _claim = claim;
             _allotService = allotService;
             _extractService = extractService;
             _serviceScopeFactory = serviceScopeFactory;
+            _backgroundTaskQueue = backgroundTaskQueue;
             _notificationQueue = notificationQueue;
         }
 
@@ -43,27 +46,27 @@ public ApiResponse CustomExtract(int allotId)
             if (!_extractService.CheckConfigScript(userId, allotId))
                 return new ApiResponse(ResponseType.Fail, "未配置自定义抽取，请联系绩效管理人员。");
 
-            //_backgroundTaskQueue.QueueBackgroundWorkItem(async token =>
-            //{
-            //    using (var scope = _serviceScopeFactory.CreateScope())
-            //    {
-            //        var scopedServices = scope.ServiceProvider.GetRequiredService<CustomExtractService>();
-            //        var scopedAllotService = scope.ServiceProvider.GetRequiredService<AllotService>();
-            //        var scopedQueue = scope.ServiceProvider.GetRequiredService<IHubNotificationQueue>();
+            _backgroundTaskQueue.QueueBackgroundWorkItem(async token =>
+            {
+                using (var scope = _serviceScopeFactory.CreateScope())
+                {
+                    var scopedServices = scope.ServiceProvider.GetRequiredService<CustomExtractService>();
+                    var scopedAllotService = scope.ServiceProvider.GetRequiredService<AllotService>();
+                    var scopedQueue = scope.ServiceProvider.GetRequiredService<IHubNotificationQueue>();
 
-            //        if (scopedServices.ExtractData(userId, allotId, out string resultFilePath))
-            //        {
-            //            scopedAllotService.UpdateAllotCustomExtractPath(allotId, resultFilePath);
-            //            scopedQueue.Send(new Notification(allotId, "CustomDowoload", new CustomDownloadContent("自定义数据提取数据成功，是否立即下载", allotId)));
-            //        }
-            //        else
-            //        {
-            //            scopedQueue.Send(new Notification(allotId, "Notification", new TextContent("自定义数据提取数据失败", NotificationLevel.ERR)));
-            //        }
+                    if (scopedServices.ExtractData(userId, allotId, out string resultFilePath))
+                    {
+                        scopedAllotService.UpdateAllotCustomExtractPath(allotId, resultFilePath);
+                        scopedQueue.Send(new Notification(allotId, "CustomDowoload", new CustomDownloadContent("自定义数据提取数据成功，是否立即下载", allotId)));
+                    }
+                    else
+                    {
+                        scopedQueue.Send(new Notification(allotId, "Notification", new TextContent("自定义数据提取数据失败", NotificationLevel.ERR)));
+                    }
 
-            //        await Task.Delay(TimeSpan.FromSeconds(5), token);
-            //    }
-            //});
+                    await Task.Delay(TimeSpan.FromSeconds(5), token);
+                }
+            });
             _notificationQueue.Send(new Notification(allotId, "Notification", new TextContent("自定义数据提取任务开始执行")));