jinko.yikatong/Src/Dlhis.Basic/Device/Services/Implements/DeviceGroupService.cs

using Dlhis.Basic.Device.Request;
using Dlhis.Basic.Device.Response;
using Dlhis.Entity.Basic.Entity;
using Dlhis.Entity.Device.Entities;
using Dlhis.Entity.Device.Enums;
using Kurisu.AspNetCore.Utils.Extensions;
using Kurisu.Core.DataAccess.Dto;
using Kurisu.SqlSugar.Attributes;
using Kurisu.SqlSugar.Extensions;
using Kurisu.SqlSugar.Services;
using Mapster;

namespace Dlhis.Basic.Device.Services.Implements;

public class DeviceGroupService : IDeviceGroupService
{
    private readonly IDbContext _dbContext;
    private readonly IDevicePermissionService _devicePermissionService;

    public DeviceGroupService(IDbContext dbContext, IDevicePermissionService devicePermissionService)
    {
        _dbContext = dbContext;
        _devicePermissionService = devicePermissionService;
    }

    public async Task<Pagination<DeviceGroupPageResponse>> GetPageAsync(DeviceGroupPageRequest request)
    {
        return await _dbContext.Queryable<DeviceGroupEntity>()
               .WhereIF(request.Search.IsPresent(), x => x.Name.Contains(request.Search))
               .WhereIF(request.Type.HasValue, x => x.Type == request.Type)
               .Select<DeviceGroupPageResponse>()
               .ToPageListAsync(request);
    }

    public async Task<List<DeviceGroupTreeResponse>> GetTreeAsync(bool showDevices, DeviceTypeEnumType? type)
    {
        var deviceGroups = await _dbContext.Queryable<DeviceGroupEntity>()
            .WhereIF(type.HasValue, x => x.Type == type)
            .Select<DeviceGroupTreeResponse>()
            .ToListAsync();

        var tops = deviceGroups.Where(x => !x.PCode.HasValue).ToList();
        tops.ForEach(x =>
        {
            RecursionGetNext(x, deviceGroups, showDevices);
        });

        return tops;
    }

    public async Task SaveAsync(DeviceGroupSaveRequest request)
    {
        //检查父级节点
        await CheckPCodeExistsDeviceAsync(request.PCode);

        if (!request.Id.HasValue)
        {
            var entity = request.Adapt<DeviceGroupEntity>();
            entity.Editable = true;
            entity.Code = Guid.NewGuid();

            await _dbContext.InsertAsync(entity);
        }
        else
        {
            var entity = await _dbContext.Queryable<DeviceGroupEntity>().Where(x => x.Id == request.Id).FirstAsync();
            entity.ThrowIfNull("设备组不存在,无法更新");

            entity.Editable.ThrowIfFalse("不可编辑当前组");

            var nextDevices = await GetDevicesAsync(entity.Code);
            if (!entity.Type.HasValue)
            {
                (request.Type.HasValue && nextDevices.Any(x => x.Type != request.Type)).ThrowIfTrue($"分组设备存在非{request.Type.Value.GetDescription()}类型的设备设备");
            }
            else
            {
                (request.Type.HasValue && request.Type != entity.Type && nextDevices.Any()).ThrowIfTrue("分组设备类型不匹配,无法修改类型");
            }

            await _dbContext.UpdateAsync(entity);
        }
    }

    public async Task DeleteAsync(long id)
    {
        var deviceGroup = await _dbContext.Queryable<DeviceGroupEntity>().Where(x => x.Id == id).FirstAsync();
        deviceGroup.ThrowIfNull("设备组不存在,无需删除");

        await _dbContext.DeleteAsync(deviceGroup);
    }


    [Transactional]
    public async Task AssignDevicesAsync(DeviceAssignRequest request)
    {
        var group = await _dbContext.Queryable<DeviceGroupEntity>().Where(x => x.Code == request.GroupCode).FirstAsync();
        group.ThrowIfNull("设备组不存在");

        await CheckNoNextGroupAsync(group.Code);
        await CheckGroupTypeMatchAsync(request, group);

        //下发任务
        await _devicePermissionService.GenerateDeviceTasksAsync(async () =>
        {
            var relationships = await _dbContext.Queryable<DeviceGroupRelationshipEntity>().Where(x => x.GroupCode == request.GroupCode).ToListAsync();
            await _dbContext.DeleteAsync(relationships);


            var entities = request.DeviceCodes.Select(x => new DeviceGroupRelationshipEntity
            {
                GroupCode = request.GroupCode,
                DeviceCode = x

            }).ToList();

            await _dbContext.InsertAsync(entities);

        }, null);
    }

    public async Task<List<DeviceUnderGroupResponse>> GetDevicesAsync(Guid code)
    {
        var relationships = await _dbContext.Queryable<DeviceGroupRelationshipEntity>().Where(x => x.GroupCode == code).ToListAsync();
        var deviceCodes = relationships.Select(x => x.DeviceCode);

        return await _dbContext.Queryable<DeviceEntity>()
               .Where(x => deviceCodes.Contains(x.Code))
               .Select<DeviceUnderGroupResponse>()
               .ToListAsync();

    }

    /****************************************************** help methods *********************************************************/

    /// <summary>
    /// 检查父级分组存在设备
    /// </summary>
    /// <param name="pCode"></param>
    /// <returns></returns>
    private async Task CheckPCodeExistsDeviceAsync(Guid? pCode)
    {
        if (pCode.HasValue)
        {
            var exists = await _dbContext.Queryable<DeviceGroupRelationshipEntity>().Where(x => x.GroupCode == pCode).AnyAsync();
            exists.ThrowIfTrue("父级分组存在设备");
        }
    }

    /// <summary>
    /// 
    /// </summary>
    /// <param name="currentCode"></param>
    /// <returns></returns>
    private async Task CheckNoNextGroupAsync(Guid currentCode)
    {
        var exists = await _dbContext.Queryable<DeviceGroupEntity>().Where(x => x.PCode == currentCode).AnyAsync();
        exists.ThrowIfTrue("存在下级设备分组");
    }

    /// <summary>
    /// 检查分组分类匹配
    /// </summary>
    /// <param name="request"></param>
    /// <param name="group"></param>
    /// <returns></returns>
    private async Task CheckGroupTypeMatchAsync(DeviceAssignRequest request, DeviceGroupEntity group)
    {
        var devices = await _dbContext.Queryable<DeviceEntity>().Where(x => request.DeviceCodes.Contains(x.Code)).ToListAsync();
        //检查分组的分类控制
        if (group.Type.HasValue)
        {
            devices.Any(x => x.Type != group.Type).ThrowIfTrue("存在设备类型有误,无法分配到当前分组");
        }

        //重新赋值到请求参数中
        request.DeviceCodes = devices.Select(x => x.Code).ToList();
    }


    /// <summary>
    /// 递归获取树
    /// </summary>
    /// <param name="current"></param>
    /// <param name="departments"></param>
    /// <param name="showDevices"></param>
    private void RecursionGetNext(DeviceGroupTreeResponse current, List<DeviceGroupTreeResponse> departments, bool showDevices)
    {
        var next = departments.Where(x => x.PCode == current.Code).ToList();
        current.Next = next;
        foreach (var item in next)
        {
            RecursionGetNext(item, departments, showDevices);
        }

        //最后层级附加设备信息
        if (!current.Next.Any() && showDevices)
        {
            current.Next = _dbContext.Queryable<DeviceGroupRelationshipEntity>()
                .LeftJoin<DeviceEntity>((relation, device) => relation.DeviceCode == device.Code)
                .Where((relation, device) => relation.GroupCode == current.Code)
                .Select((relation, device) => new DeviceGroupTreeResponse
                {
                    Code = device.Code,
                    Name = device.UniqueNo,
                    PCode = relation.GroupCode
                })
                .Distinct()
                .ToList();
        }
    }
}