-
zhangwei
3 天以前 89b94f3cc1aa492b3223b97f3312d8eca004032b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
//
// 本项目主要遵循 MIT 许可证和 Apache 许可证(版本 2.0)进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
//
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动!任何基于本项目二次开发而产生的一切法律纠纷和责任,我们不承担任何责任!
 
using Admin.NET.Plugin.DingTalk;
using Furion.Schedule;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
 
namespace Admin.NET.Plugin.Job;
 
/// <summary>
/// 同步钉钉角色job,自动同步触发器请在web页面按需求设置
/// </summary>
[JobDetail("SyncDingTalkRoleJob", Description = "同步钉钉角色", GroupName = "default", Concurrent = false)]
public class SyncDingTalkRoleJob : IJob
{
    private readonly IServiceScopeFactory _scopeFactory;
    private readonly IDingTalkApi _dingTalkApi;
    private readonly ILogger _logger;
 
    public SyncDingTalkRoleJob(IServiceScopeFactory scopeFactory, IDingTalkApi dingTalkApi, ILoggerFactory loggerFactory)
    {
        _scopeFactory = scopeFactory;
        _dingTalkApi = dingTalkApi;
        _logger = loggerFactory.CreateLogger(CommonConst.SysLogCategoryName);
    }
 
    public async Task ExecuteAsync(JobExecutingContext context, CancellationToken stoppingToken)
    {
        using var serviceScope = _scopeFactory.CreateScope();
        var _dingTalkRoleRepo = serviceScope.ServiceProvider.GetRequiredService<SqlSugarRepository<DingTalkRoleUser>>();
        var _dingTalkOptions = serviceScope.ServiceProvider.GetRequiredService<IOptions<DingTalkOptions>>();
 
        // 获取Token
        var tokenRes = await _dingTalkApi.GetDingTalkToken(_dingTalkOptions.Value.ClientId, _dingTalkOptions.Value.ClientSecret);
        if (tokenRes.ErrCode != 0)
            throw Oops.Oh(tokenRes.ErrMsg);
 
        var dingTalkRoleUserList = new List<DingTalkRoleUser>();
        // 获取角色列表
        var roleIdsRes = await _dingTalkApi.GetDingTalkRoleList(tokenRes.AccessToken, new GetDingTalkCurrentRoleListInput
        { });
        if (roleIdsRes.Success)
        {
            _logger.LogError(roleIdsRes.ErrMsg);
            throw Oops.Oh(roleIdsRes.ErrMsg);
        }
        foreach (var item in roleIdsRes.Result.list)
        {
            foreach (var role_item in item.roles)
            {
                // 根据角色id获取指定角色的员工列表
                var role_user = await _dingTalkApi.GetDingTalkRoleSimplelist(
                    tokenRes.AccessToken,
                    new GetDingTalkCurrentRoleSimplelistInput()
                    {
                        role_id = role_item.id,
                    }
                );
 
                if (role_user.Success)
                {
                    _logger.LogError(role_user.ErrMsg);
                    break;
                }
                var tempList = role_user.Result.list.Select(u => new DingTalkRoleUser
                {
                    DingTalkUserId = u.userid,
                    groupId = item.groupId,
                    groupName = item.name,
                    roleId = role_item.id,
                    roleName = role_item.name
                }).ToList();
                if (tempList?.Count > 0)
                {
                    dingTalkRoleUserList.AddRange(tempList);
                }
            }
        }
 
        // 判断新增还是更新
        var sysDingTalkRoleList = await _dingTalkRoleRepo.AsQueryable().ToListAsync();
        // 需要更新的用户Id
        var uDingTalkRole = dingTalkRoleUserList.Where(u => sysDingTalkRoleList.Any(m => m.DingTalkUserId == u.DingTalkUserId && m.groupId == u.groupId));
        // 需要新增的用户Id
        var iDingTalkRole = dingTalkRoleUserList.Where(u => !sysDingTalkRoleList.Any(m => m.DingTalkUserId == u.DingTalkUserId && m.groupId == u.groupId));
        // 需要删除的数据
        var dDingTalkRole = sysDingTalkRoleList.Where(u => !dingTalkRoleUserList.Any(m => m.DingTalkUserId == u.DingTalkUserId && m.groupId == u.groupId)).ToList();
        // 新增钉钉角色
        var iUser = iDingTalkRole.Select(res => new DingTalkRoleUser
        {
            DingTalkUserId = res.DingTalkUserId,
            groupId = res.groupId,
            groupName = res.groupName,
            roleId = res.roleId,
            roleName = res.roleName,
        }).ToList();
        if (iUser.Count > 0)
        {
            await _dingTalkRoleRepo.CopyNew().AsInsertable(iUser).ExecuteCommandAsync();
        }
 
        // 更新钉钉角色
        var uUser = uDingTalkRole.Select(res => new DingTalkRoleUser
        {
            Id = sysDingTalkRoleList.Where(u => u.DingTalkUserId == res.DingTalkUserId).Select(u => u.Id).FirstOrDefault(),
            DingTalkUserId = res.DingTalkUserId,
            groupId = res.groupId,
            groupName = res.groupName,
            roleId = res.roleId,
            roleName = res.roleName
        }).ToList();
        //添加需要删除的数据
        Parallel.ForEach(dDingTalkRole, user => user.IsDelete = true);
        uUser.AddRange(dDingTalkRole);
        if (uUser.Count > 0)
        {
            await _dingTalkRoleRepo.CopyNew().AsUpdateable(uUser).UpdateColumns(u => new
            {
                u.DingTalkUserId,
                u.groupId,
                u.groupName,
                u.roleId,
                u.roleName,
                u.UpdateTime,
                u.UpdateUserName,
                u.UpdateUserId,
                u.IsDelete
            }).ExecuteCommandAsync();
        }
 
        var originColor = Console.ForegroundColor;
        Console.ForegroundColor = ConsoleColor.Blue;
        Console.WriteLine("【" + DateTime.Now + "】同步钉钉角色");
        Console.ForegroundColor = originColor;
    }
}