Merge branch 'hotfix/v20211223'

performance/Performance.Api/Controllers/AllotController.cs

@@ -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("自定义数据提取任务开始执行")));