提取数据连接数据库方法修改

performance/Performance.Api/Controllers/TemplateController.cs

@@ -18,6 +18,7 @@
 using System.Collections.Generic;
 using System.IO;
 using System.Linq;
+using System.Threading;
 using System.Threading.Tasks;
 
 namespace Performance.Api.Controllers
@@ -230,14 +231,18 @@ public ApiResponse NewExtractData([CustomizeValidator, FromBody] ExtractRequest
 
                 if (isSingle)
                 {
+                    logger.LogInformation("同一项目中进行提取");
                     Task.Run(() =>
                     {
                         using (var scope = serviceScopeFactory.CreateScope())
                         {
                             var scopedServices = scope.ServiceProvider.GetRequiredService<ExtractService>();
+                            logger.LogInformation("提取绩效数据参数:" + JsonHelper.Serialize(new { allotId = allot.ID, hospitalId = allot.HospitalId, userId = claim.GetUserId() }));
                             string extractFilePath = scopedServices.Main(allot.ID, allot.HospitalId, email, "User" + claim.GetUserId(), filePath, isSingle);
                         }
                     });
+
+                    Thread.Sleep(1000);
                 }
                 else
                 {

performance/Performance.Services/ExtractExcelService/ExtractService.cs

@@ -75,10 +75,10 @@ public string Main(int allotId, int hospitalId, string email, string groupName,
             IWorkbook workbook = null;
             try
             {
-                logService.ReturnTheLog(allotId, groupName, 2, "等待提取", $"确认配置信息是否可完成数据提取...", 1, isSingle);
-
                 logService.ClearExtractLog(allotId);
 
+                logService.ReturnTheLog(allotId, groupName, 2, "等待提取", $"确认配置信息是否可完成数据提取...", 1, isSingle);
+
                 var allots = perallotRepository.GetEntities(t => t.HospitalId == hospitalId);
                 if (allots == null || !allots.Any(t => t.ID == allotId)) throw new Exception("绩效不存在");
 

performance/Performance.Services/ExtractExcelService/QueryService.cs

 using Dapper;
+using Microsoft.AspNetCore.Mvc.ViewFeatures;
 using Microsoft.Extensions.Logging;
 using Performance.DtoModels;
 using Performance.EntityModels;
+using Performance.Infrastructure;
 using Performance.Repository;
 using System;
 using System.Collections.Generic;
+using System.Data;
 using System.Linq;
 using System.Text;
 using System.Text.RegularExpressions;
@@ -191,36 +194,52 @@ private List<ex_result> ExtractModuleData(per_allot allot, string groupName, boo
                     logService.ReturnTheLog(allot.ID, groupName, 3, "", ratio > 20 ? 20 : ratio, 1, isSingle);
                     logService.ReturnTheLog(allot.ID, groupName, 2, "提取数据", $"开始提取模块“{string.Join("、", thisModules.Select(t => t.ModuleName))}”的数据", 1, isSingle);
 
+                    Dictionary<int, IDbConnection> pools = new Dictionary<int, IDbConnection>();
+
                     foreach (var script in scripts.Where(t => t.TypeId == typeId))
                     {
                         var config = configs.FirstOrDefault(t => t.Id == script.ConfigId) ?? configs.FirstOrDefault(t => t.DataBaseType == script.DatabaseType);
                         if (config == null) continue;
 
-                        var querydata = QueryData(config, allot, script.ExecScript);
-                        if (querydata != null && querydata.Any())
+                        try
                         {
-                            thisModules.ForEach(f =>
+                            if(!pools.ContainsKey(config.Id))
+                               pools.Add(config.Id, ConnectionBuilder.Create((DatabaseType)config.DataBaseType, config.DbSource, config.DbName, config.DbUser, config.DbPassword));
+
+                            IDbConnection connection = pools[config.Id];
+
+                            var parameters = GetParameters(allot);
+                            var querydata = QueryData(connection, parameters, script.ExecScript);
+
+                            if (querydata != null && querydata.Any())
                             {
-                                var result = querydata.Select(t => new ex_result
+                                thisModules.ForEach(f =>
                                 {
-                                    Department = t.Department,
-                                    Category = t.Category?.Trim(),
-                                    Fee = t.Value,
-                                    DoctorName = t.DoctorName,
-                                    PersonnelNumber = t.PersonnelNumber,
-                                    Source = f.ModuleName,
-                                    DatabaseType = config.DataBaseType,
-                                    ConfigId = config.Id,
-                                    AllotId = allot.ID,
-                                    CreateTime = CreateTime,
-                                }).ToList();
-                                exresultRepository.AddRange(result.ToArray());
-                                data.AddRange(result);
-                            });
+                                    var result = querydata.Select(t => new ex_result
+                                    {
+                                        Department = t.Department,
+                                        Category = t.Category?.Trim(),
+                                        Fee = t.Value,
+                                        DoctorName = t.DoctorName,
+                                        PersonnelNumber = t.PersonnelNumber,
+                                        Source = f.ModuleName,
+                                        DatabaseType = config.DataBaseType,
+                                        ConfigId = config.Id,
+                                        AllotId = allot.ID,
+                                        CreateTime = CreateTime,
+                                    }).ToList();
+                                    exresultRepository.AddRange(result.ToArray());
+                                    data.AddRange(result);
+                                });
+                            }
+
+                            logService.ReturnTheLog(allot.ID, groupName, 2, "提取数据", $"模块“{string.Join("、", thisModules.Select(t => t.ModuleName))}”的数据已完成提取", 1, isSingle);
+                        }
+                        catch (Exception)
+                        {
+                            logService.ReturnTheLog(allot.ID, groupName, 2, "数据库连接", $"数据库“{config.DbName}”连接失败", 3, isSingle);
                         }
                     }
-
-                    logService.ReturnTheLog(allot.ID, groupName, 2, "提取数据", $"模块“{string.Join("、", thisModules.Select(t => t.ModuleName))}”的数据已完成提取", 1, isSingle);
                 }
             }
 
@@ -372,17 +391,30 @@ public IEnumerable<ExtractDto> QueryData(sys_hospitalconfig config, per_allot al
             var parameters = GetParameters(allot);
             using (var connection = ConnectionBuilder.Create((DatabaseType)config.DataBaseType, config.DbSource, config.DbName, config.DbUser, config.DbPassword))
             {
-                foreach (var item in parameters)
-                {
-                    execsql = Regex.Replace(execsql, item.Key, item.Value, RegexOptions.IgnoreCase);
-                }
+                return QueryData(connection, parameters, execsql);
+            }
+        }
+        /// <summary>
+        /// 查询数据
+        /// </summary>
+        /// <param name="config"></param>
+        /// <param name="allot"></param>
+        /// <param name="execsql"></param>
+        /// <param name="source"></param>
+        /// <param name="category"></param>
+        /// <returns></returns>
+        public IEnumerable<ExtractDto> QueryData(IDbConnection connection, Dictionary<string, string> parameters, string execsql)
+        {
+            foreach (var item in parameters)
+            {
+                execsql = Regex.Replace(execsql, item.Key, item.Value, RegexOptions.IgnoreCase);
+            }
 
-                logger.LogInformation($"提取绩效数据SQL脚本{execsql}");
-                var result = connection.Query<ExtractDto>(execsql, commandTimeout: 20000);
-                logger.LogInformation($"提取绩效数据执行脚本获取数据{result?.Count() ?? 0}条记录");
+            logger.LogInformation($"提取绩效数据SQL脚本{execsql}");
+            var result = connection.Query<ExtractDto>(execsql, commandTimeout: 20000);
+            logger.LogInformation($"提取绩效数据执行脚本获取数据{result?.Count() ?? 0}条记录");
 
-                return result;
-            }
+            return result;
         }
 
         /// <summary>