EnumToDictJob.cs 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151
  1. // Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
  2. //
  3. // 本项目主要遵循 MIT 许可证和 Apache 许可证(版本 2.0)进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
  4. //
  5. // 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动!任何基于本项目二次开发而产生的一切法律纠纷和责任,我们不承担任何责任!
  6. namespace Admin.NET.Core;
  7. /// <summary>
  8. /// 枚举转字典
  9. /// </summary>
  10. [JobDetail("job_EnumToDictJob", Description = "枚举转字典", GroupName = "default", Concurrent = false)]
  11. [PeriodSeconds(1, TriggerId = "trigger_EnumToDictJob", Description = "枚举转字典", MaxNumberOfRuns = 1, RunOnStart = true)]
  12. public class EnumToDictJob : IJob
  13. {
  14. private readonly IServiceScopeFactory _scopeFactory;
  15. private readonly IJsonSerializerProvider _jsonSerializer;
  16. public EnumToDictJob(IServiceScopeFactory scopeFactory, IJsonSerializerProvider jsonSerializer)
  17. {
  18. _scopeFactory = scopeFactory;
  19. _jsonSerializer = jsonSerializer;
  20. }
  21. public async Task ExecuteAsync(JobExecutingContext context, CancellationToken stoppingToken)
  22. {
  23. using var serviceScope = _scopeFactory.CreateScope();
  24. var sysEnumService = serviceScope.ServiceProvider.GetRequiredService<SysEnumService>();
  25. var db = serviceScope.ServiceProvider.GetRequiredService<ISqlSugarClient>().CopyNew();
  26. var enumTypeList = sysEnumService.GetEnumTypeList();
  27. var enumCodeList = enumTypeList.Select(x => x.TypeName);
  28. var sysDictTypeCodeList = await db.Queryable<SysDictType>().Where(x => enumCodeList.Contains(x.Code)).Select(x => x.Code).ToListAsync(stoppingToken);
  29. // 更新的
  30. var uEnumType = enumTypeList.Where(x => sysDictTypeCodeList.Contains(x.TypeName)).ToList();
  31. var waitUpdateSysDictType = await db.Queryable<SysDictType>().Where(x => uEnumType.Any(y => y.TypeName == x.Code)).ToListAsync(stoppingToken);
  32. var waitUpdateSysDictTypeDict = waitUpdateSysDictType.ToDictionary(x => x.Code, x => x);
  33. var waitUpdateSysDictData = await db.Queryable<SysDictData>().Where(x => uEnumType.Any(y => y.TypeName == x.DictType.Code)).ToListAsync(stoppingToken);
  34. var uSysDictType = new List<SysDictType>();
  35. var uSysDictData = new List<SysDictData>();
  36. if (uEnumType.Count > 0)
  37. {
  38. uEnumType.ForEach(e =>
  39. {
  40. if (waitUpdateSysDictTypeDict.TryGetValue(e.TypeName, out SysDictType value))
  41. {
  42. var uDictType = value;
  43. uDictType.Name = e.TypeDescribe;
  44. uDictType.Remark = e.TypeRemark;
  45. var uDictData = waitUpdateSysDictData.Where(x => x.DictTypeId == uDictType.Id).ToList();
  46. if (uDictData.Count > 0)
  47. {
  48. uDictData.ForEach(d =>
  49. {
  50. var enumData = e.EnumEntities.Where(x => d.Code == x.Name).FirstOrDefault();
  51. if (enumData != null)
  52. {
  53. d.Value = enumData.Value.ToString();
  54. d.Code = enumData.Name;
  55. uSysDictData.Add(d);
  56. }
  57. });
  58. }
  59. if (!uSysDictType.Any(x => x.Id == uDictType.Id))
  60. {
  61. uSysDictType.Add(uDictType);
  62. }
  63. }
  64. });
  65. try
  66. {
  67. db.Ado.BeginTran();
  68. if (uSysDictType.Count > 0)
  69. {
  70. await db.Updateable(uSysDictType).ExecuteCommandAsync(stoppingToken);
  71. }
  72. if (uSysDictData.Count > 0)
  73. {
  74. await db.Updateable(uSysDictData).ExecuteCommandAsync(stoppingToken);
  75. }
  76. db.Ado.CommitTran();
  77. }
  78. catch (Exception error)
  79. {
  80. db.Ado.RollbackTran();
  81. Log.Error($"{context.Trigger.Description}错误:" + _jsonSerializer.Serialize(error));
  82. throw new Exception($"{context.Trigger.Description}错误");
  83. }
  84. }
  85. // 添加的
  86. var iEnumType = enumTypeList.Where(x => !sysDictTypeCodeList.Contains(x.TypeName)).ToList();
  87. if (iEnumType.Count > 0)
  88. {
  89. // 需要新增的字典类型
  90. var iDictType = iEnumType.Select(x => new SysDictType
  91. {
  92. Id = YitIdHelper.NextId(),
  93. Code = x.TypeName,
  94. Name = x.TypeDescribe,
  95. Remark = x.TypeRemark,
  96. Status = StatusEnum.Enable,
  97. OrderNo = 100
  98. }).ToList();
  99. // 需要新增的字典数据
  100. var dictData = iEnumType.Join(iDictType, t1 => t1.TypeName, t2 => t2.Code, (t1, t2) => new
  101. {
  102. data = t1.EnumEntities.Select(x => new SysDictData
  103. {
  104. // 性能优化,使用BulkCopyAsync必须手动获取Id
  105. Id = YitIdHelper.NextId(),
  106. DictTypeId = t2.Id,
  107. Name = x.Describe,
  108. Value = x.Value.ToString(),
  109. Code = x.Name,
  110. Remark = t2.Remark,
  111. OrderNo = 100,
  112. TagType = "info"
  113. }).ToList()
  114. }).ToList();
  115. var iDictData = new List<SysDictData>();
  116. dictData.ForEach(item =>
  117. {
  118. iDictData.AddRange(item.data);
  119. });
  120. try
  121. {
  122. db.Ado.BeginTran();
  123. if (iDictType.Count > 0)
  124. {
  125. await db.Insertable(iDictType).ExecuteCommandAsync(stoppingToken);
  126. }
  127. if (iDictData.Count > 0)
  128. {
  129. await db.Insertable(iDictData).ExecuteCommandAsync(stoppingToken);
  130. }
  131. db.Ado.CommitTran();
  132. }
  133. catch (Exception error)
  134. {
  135. db.Ado.RollbackTran();
  136. Log.Error($"{context.Trigger.Description}错误:" + _jsonSerializer.Serialize(error));
  137. throw new Exception($"{context.Trigger.Description}错误");
  138. }
  139. }
  140. var originColor = Console.ForegroundColor;
  141. Console.ForegroundColor = ConsoleColor.Green;
  142. Console.WriteLine($"【{DateTime.Now}】系统枚举转换字典");
  143. Console.ForegroundColor = originColor;
  144. }
  145. }