feat:开发

sunpengfei

2025-08-12 209c09c20a7737094b660f4f35857e8e989a24c5

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
using FlexJobApi.Core;
using Furion.DatabaseAccessor;
using Furion.FriendlyException;
using Mapster;
using MediatR;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
 
namespace FlexJobApi.UserServer.Application
{
    /// <summary>
    /// 保存角色
    /// </summary>
    public class SaveRoleCommandHandler(
            IRepository<Role> rep
        ) : IRequestHandler<SaveRoleCommand, Guid>
    {
        private readonly IRepository<Role> rep = rep;
 
        /// <inheritdoc/>
        public async Task<Guid> Handle(SaveRoleCommand request, CancellationToken cancellationToken)
        {
            if (request.Id.HasValue)
            {
                var entity = await rep.AsQueryable()
                    .Include(it => it.RoleMenus)
                    .Include(it => it.RoleResources)
                    .FirstOrDefaultAsync(it => it.Id == request.Id, cancellationToken);
                if (entity == null) throw Oops.Oh(EnumErrorCodeType.s404, "角色");
                request.Adapt(entity);
                entity.RoleMenus = entity.RoleMenus.Where(it => request.MenuIds.Contains(it.MenuId)).ToList();
                foreach (var menuId in request.MenuIds)
                {
                    var entityRoleMenu = entity.RoleMenus.FirstOrDefault(it => it.MenuId == menuId);
                    if (entityRoleMenu == null)
                    {
                        entityRoleMenu = new RoleMenu
                        {
                            MenuId = menuId
                        };
                        entity.RoleMenus.Add(entityRoleMenu);
                    }
                }
                entity.RoleResources = entity.RoleResources.Where(it => request.Resources.Any(r => r.ResourceId == it.ResourceId)).ToList();
                foreach (var roleResource in request.Resources)
                {
                    var entityRoleResource = entity.RoleResources.FirstOrDefault(it => it.ResourceId == roleResource.ResourceId);
                    if (entityRoleResource == null)
                    {
                        entityRoleResource = new RoleResource
                        {
                            ResourceId = roleResource.ResourceId,
                            DataPower = roleResource.DataPower
                        };
                        entity.RoleResources.Add(entityRoleResource);
                    }
                    else
                    {
                        entityRoleResource.DataPower = roleResource.DataPower;
                    }
                }
                await rep.UpdateAsync(entity);
                return entity.Id;
            }
            else
            {
                var entity = new Role();
                request.Adapt(entity);
                entity.RoleMenus = request.MenuIds
                    .Select(it => new RoleMenu
                    {
                        MenuId = it
                    })
                    .ToList();
                entity.RoleResources = request.Resources
                    .Select(it => new RoleResource
                    {
                        ResourceId = it.ResourceId,
                        DataPower = it.DataPower
                    })
                    .ToList();
                await rep.InsertAsync(entity);
                return entity.Id;
            }
        }
    }
}