sunpengfei
2025-09-02 465f14d18c1ee52efffd7095bf7535a6bbb2e06f
FlexJobApi.UserServer.Application/Users/Commands/UserInfoCommandHandler.cs
@@ -1,5 +1,6 @@
using FlexJobApi.Core;
using Furion.DatabaseAccessor;
using Furion.FriendlyException;
using MediatR;
using Microsoft.EntityFrameworkCore;
using System;
@@ -15,8 +16,9 @@
    /// </summary>
    public class UserInfoCommandHandler(
            IRepository<User> rep
        ) :
        IRequestHandler<SetUserInfoStatusCommand, int>
        ) :
        IRequestHandler<SetUserInfoStatusCommand, int>,
        IRequestHandler<SetUserInfoRolesCommand, int>
    {
        private readonly IRepository<User> rep = rep;
@@ -37,5 +39,29 @@
            }
            return entities.Count;
        }
        /// <summary>
        /// 设置用户信息角色
        /// </summary>
        /// <param name="request"></param>
        /// <param name="cancellationToken"></param>
        /// <returns></returns>
        public async Task<int> Handle(SetUserInfoRolesCommand request, CancellationToken cancellationToken)
        {
            var entity = await rep.AsQueryable()
                .Include(it => it.UserRoles)
                .Where(it => it.Id == request.UserInfoId)
                .FirstOrDefaultAsync(cancellationToken);
            if (entity == null) throw Oops.Oh(EnumErrorCodeType.s404, "用户");
            entity.UserRoles = request.RoleIds
                .Select(it => new UserRole
                {
                    RoleId = it
                })
                .ToList();
            await rep.UpdateAsync(entity);
            return entity.UserRoles.Count;
        }
    }
}