SyncDingTalkRoleJob.cs 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134
  1. using Admin.NET.Plugin.DingTalk;
  2. using Furion.Schedule;
  3. using Microsoft.Extensions.DependencyInjection;
  4. using Microsoft.Extensions.Logging;
  5. namespace Admin.NET.Plugin.Job;
  6. /// <summary>
  7. /// 同步钉钉角色job,自动同步触发器请在web页面按需求设置
  8. /// </summary>
  9. [JobDetail("SyncDingTalkRoleJob", Description = "同步钉钉角色", GroupName = "default", Concurrent = false)]
  10. public class SyncDingTalkRoleJob : IJob
  11. {
  12. private readonly IServiceScopeFactory _scopeFactory;
  13. private readonly IDingTalkApi _dingTalkApi;
  14. private readonly ILogger _logger;
  15. public SyncDingTalkRoleJob(IServiceScopeFactory scopeFactory, IDingTalkApi dingTalkApi, ILoggerFactory loggerFactory)
  16. {
  17. _scopeFactory = scopeFactory;
  18. _dingTalkApi = dingTalkApi;
  19. _logger = loggerFactory.CreateLogger(CommonConst.SysLogCategoryName);
  20. }
  21. public async Task ExecuteAsync(JobExecutingContext context, CancellationToken stoppingToken)
  22. {
  23. using var serviceScope = _scopeFactory.CreateScope();
  24. var _dingTalkRoleRepo = serviceScope.ServiceProvider.GetRequiredService<SqlSugarRepository<DingTalkRoleUser>>();
  25. var _dingTalkOptions = serviceScope.ServiceProvider.GetRequiredService<IOptions<DingTalkOptions>>();
  26. // 获取Token
  27. var tokenRes = await _dingTalkApi.GetDingTalkToken(_dingTalkOptions.Value.ClientId, _dingTalkOptions.Value.ClientSecret);
  28. if (tokenRes.ErrCode != 0)
  29. throw Oops.Oh(tokenRes.ErrMsg);
  30. var dingTalkRoleUserList = new List<DingTalkRoleUser>();
  31. // 获取角色列表
  32. var roleIdsRes = await _dingTalkApi.GetDingTalkRoleList(tokenRes.AccessToken, new GetDingTalkCurrentRoleListInput
  33. { });
  34. if (roleIdsRes.Success)
  35. {
  36. _logger.LogError(roleIdsRes.ErrMsg);
  37. throw Oops.Oh(roleIdsRes.ErrMsg);
  38. }
  39. foreach (var item in roleIdsRes.Result.list)
  40. {
  41. foreach (var role_item in item.roles)
  42. {
  43. // 根据角色id获取指定角色的员工列表
  44. var role_user = await _dingTalkApi.GetDingTalkRoleSimplelist(
  45. tokenRes.AccessToken,
  46. new GetDingTalkCurrentRoleSimplelistInput()
  47. {
  48. role_id = role_item.id,
  49. }
  50. );
  51. if (role_user.Success)
  52. {
  53. _logger.LogError(role_user.ErrMsg);
  54. break;
  55. }
  56. var tempList = role_user.Result.list.Select(u => new DingTalkRoleUser
  57. {
  58. DingTalkUserId = u.userid,
  59. groupId = item.groupId,
  60. groupName = item.name,
  61. roleId = role_item.id,
  62. roleName = role_item.name
  63. }).ToList();
  64. if (tempList?.Count > 0)
  65. {
  66. dingTalkRoleUserList.AddRange(tempList);
  67. }
  68. }
  69. }
  70. // 判断新增还是更新
  71. var sysDingTalkRoleList = await _dingTalkRoleRepo.AsQueryable().ToListAsync();
  72. // 需要更新的用户Id
  73. var uDingTalkRole = dingTalkRoleUserList.Where(u => sysDingTalkRoleList.Any(m => m.DingTalkUserId == u.DingTalkUserId && m.groupId == u.groupId));
  74. // 需要新增的用户Id
  75. var iDingTalkRole = dingTalkRoleUserList.Where(u => !sysDingTalkRoleList.Any(m => m.DingTalkUserId == u.DingTalkUserId && m.groupId == u.groupId));
  76. // 需要删除的数据
  77. var dDingTalkRole = sysDingTalkRoleList.Where(u => !dingTalkRoleUserList.Any(m => m.DingTalkUserId == u.DingTalkUserId && m.groupId == u.groupId)).ToList();
  78. // 新增钉钉角色
  79. var iUser = iDingTalkRole.Select(res => new DingTalkRoleUser
  80. {
  81. DingTalkUserId = res.DingTalkUserId,
  82. groupId = res.groupId,
  83. groupName = res.groupName,
  84. roleId = res.roleId,
  85. roleName = res.roleName,
  86. }).ToList();
  87. if (iUser.Count > 0)
  88. {
  89. await _dingTalkRoleRepo.CopyNew().AsInsertable(iUser).ExecuteCommandAsync();
  90. }
  91. // 更新钉钉角色
  92. var uUser = uDingTalkRole.Select(res => new DingTalkRoleUser
  93. {
  94. Id = sysDingTalkRoleList.Where(u => u.DingTalkUserId == res.DingTalkUserId).Select(u => u.Id).FirstOrDefault(),
  95. DingTalkUserId = res.DingTalkUserId,
  96. groupId = res.groupId,
  97. groupName = res.groupName,
  98. roleId = res.roleId,
  99. roleName = res.roleName
  100. }).ToList();
  101. //添加需要删除的数据
  102. Parallel.ForEach(dDingTalkRole, user => user.IsDelete = true);
  103. uUser.AddRange(dDingTalkRole);
  104. if (uUser.Count > 0)
  105. {
  106. await _dingTalkRoleRepo.CopyNew().AsUpdateable(uUser).UpdateColumns(u => new
  107. {
  108. u.DingTalkUserId,
  109. u.groupId,
  110. u.groupName,
  111. u.roleId,
  112. u.roleName,
  113. u.UpdateTime,
  114. u.UpdateUserName,
  115. u.UpdateUserId,
  116. u.IsDelete
  117. }).ExecuteCommandAsync();
  118. }
  119. var originColor = Console.ForegroundColor;
  120. Console.ForegroundColor = ConsoleColor.Blue;
  121. Console.WriteLine("【" + DateTime.Now + "】同步钉钉角色");
  122. Console.ForegroundColor = originColor;
  123. }
  124. }