DMS/Data/Repositories/DeviceRepository.cs

using System.Diagnostics;
using AutoMapper;
using iNKORE.UI.WPF.Modern.Common.IconKeys;
using PMSWPF.Data.Entities;
using PMSWPF.Enums;
using PMSWPF.Extensions;
using PMSWPF.Helper;
using PMSWPF.Models;
using SqlSugar;

namespace PMSWPF.Data.Repositories;

public class DeviceRepository
{
    private readonly IMapper _mapper;
    private readonly MenuRepository _menuRepository;
    private readonly VarTableRepository _varTableRepository;

    public DeviceRepository(IMapper mapper,MenuRepository menuRepository,VarTableRepository varTableRepository)
    {
        _mapper = mapper;
        _menuRepository = menuRepository;
        _varTableRepository = varTableRepository;
    }


    /// <summary>
    /// 编辑设备信息。
    /// </summary>
    /// <param name="device">要编辑的设备对象。</param>
    /// <returns>受影响的行数。</returns>
    public async Task<int> UpdateAsync(Device device)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        using (var db = DbContext.GetInstance())
        {
            var result = await UpdateAsync(device, db);
            stopwatch.Stop();
            NlogHelper.Info($"编辑设备 '{device.Name}' 耗时：{stopwatch.ElapsedMilliseconds}ms");
            return result;
        }
    }

    /// <summary>
    /// 编辑设备信息。支持事务
    /// </summary>
    /// <param name="device">要编辑的设备对象。</param>
    /// <returns>受影响的行数。</returns>
    public async Task<int> UpdateAsync(Device device, SqlSugarClient db)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        
        var result = await db.Updateable<DbDevice>(_mapper.Map<DbDevice>(device))
                             .ExecuteCommandAsync();
        stopwatch.Stop();
        NlogHelper.Info($"编辑设备 '{device.Name}' 耗时：{stopwatch.ElapsedMilliseconds}ms");
        return result;
    }


    /// <summary>
    /// 获取设备列表
    /// </summary>
    /// <returns></returns>
    public async Task<List<Device>> GetAllAsync()
    {
        using (var db = DbContext.GetInstance())
        {
            Stopwatch stopwatch = new Stopwatch();
            stopwatch.Start();
            var dlist = await db.Queryable<DbDevice>()
                                .Includes(d => d.VariableTables, dv => dv.Device)
                                .Includes(d => d.VariableTables, dvd => dvd.Variables, data => data.VariableTable)
                                .Includes(d=>d.VariableTables,vt=>vt.Variables,v=>v.VariableMqtts )
                                .ToListAsync();
           

            stopwatch.Stop();
            NlogHelper.Info($"加载设备列表总耗时：{stopwatch.ElapsedMilliseconds}ms");
            var devices= _mapper.Map<List<Device>>(dlist);
            return devices;
        }
    }

    /// <summary>
    /// 根据ID获取设备信息。
    /// </summary>
    /// <param name="id">设备ID。</param>
    /// <returns>对应的DbDevice对象。</returns>
    public async Task<Device> GetByIdAsync(int id)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        using (var db = DbContext.GetInstance())
        {
            var result = await db.Queryable<DbDevice>()
                                 .FirstAsync(p => p.Id == id);
            stopwatch.Stop();
            NlogHelper.Info($"根据ID '{id}' 获取设备耗时：{stopwatch.ElapsedMilliseconds}ms");
            return _mapper.Map<Device>(result);
        }
    }


    /// <summary>
    /// 删除设备。
    /// </summary>
    /// <param name="id">要删除的设备ID。</param>
    /// <returns>受影响的行数。</returns>
    public async Task<int> DeleteAsync(Device device, List<MenuBean> menus)
    {
        using (var db = DbContext.GetInstance())
        {
            await db.BeginTranAsync();
            try
            {
                Stopwatch stopwatch = new Stopwatch();
                stopwatch.Start();
                var res = await DeleteAsync(device, menus, db);

                stopwatch.Stop();
                NlogHelper.Info($"删除设备:{device.Name},耗时：{stopwatch.ElapsedMilliseconds}ms");
                await db.CommitTranAsync();
                return res;
            }
            catch (Exception e)
            {
                await db.RollbackTranAsync();
                throw;
            }

            return 0;
        }
    }


    /// <summary>
    /// 删除设备。
    /// </summary>
    /// <param name="device"></param>
    /// <param name="menus"></param>
    /// <param name="db"></param>
    /// <param name="id">要删除的设备ID。</param>
    /// <returns>受影响的行数。</returns>
    public async Task<int> DeleteAsync(Device device, List<MenuBean> menus, SqlSugarClient db)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        var result = await db.Deleteable<DbDevice>(new DbDevice { Id = device.Id })
                             .ExecuteCommandAsync();
        // 删除变量表
        await _varTableRepository.DeleteAsync(device.VariableTables, db);

        // 删除菜单
        var menu = DataServicesHelper.FindMenusForDevice(device, menus);
        if (menu == null)
            throw new NullReferenceException($"没有找到设备:{device.Name},的菜单对象。");
        await _menuRepository.DeleteAsync(menu, db);
        stopwatch.Stop();
        NlogHelper.Info($"删除设备:{device.Name},耗时：{stopwatch.ElapsedMilliseconds}ms");
        return result;
    }


    /// <summary>
    /// 添加设备
    /// </summary>
    /// <param name="device"></param>
    public async Task AddAsync(Device device)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        var db = DbContext.GetInstance();
        try
        {
            // 开启事务
            await db.BeginTranAsync();
            //查询设备的名字是否存在

            var exist = await db.Queryable<DbDevice>()
                                .Where(d => d.Name == device.Name)
                                .FirstAsync();
            if (exist != null)
                throw new InvalidOperationException("设备名称已经存在。");

            // 2. 将设备添加到数据库
            var addDevice = await AddAsync(device, db);


            await db.CommitTranAsync();
            // 菜单也添加成功，通知 UI 更新
            MessageHelper.SendLoadMessage(LoadTypes.Menu);
            MessageHelper.SendLoadMessage(LoadTypes.Devices);
        }
        catch (Exception e)
        {
            // 中间出错了 回滚
            await db.RollbackTranAsync();
            // 捕获并记录所有未预期的异常
            NotificationHelper.ShowError("添加设备的过程中发生了不可预期的错误：" + e.Message, e);
        }
        finally
        {
            stopwatch.Stop();
            NlogHelper.Info($"添加设备 '{device.Name}' 及相关菜单耗时：{stopwatch.ElapsedMilliseconds}ms");
        }
    }

    /// <summary>
    /// 单独添加设备，不包括菜单和变量表
    /// </summary>
    /// <param name="device"></param>
    /// <param name="db"></param>
    /// <returns></returns>
    /// <exception cref="InvalidOperationException"></exception>
    private async Task<Device> AddAsync(Device device, SqlSugarClient db)
    {
        ;
        // 添加设备
        var addDevice = await db.Insertable<DbDevice>(_mapper.Map<DbDevice>(device))
                                .ExecuteReturnEntityAsync();

        // 4. 为新设备添加菜单
        var addDeviceMenuId = await _menuRepository.AddAsync(addDevice, db);
        
        if (device.IsAddDefVarTable)
        {
            // 添加默认变量表
            var varTable = new VariableTable();
            device.VariableTables = new ();
            varTable.IsActive = true;
            varTable.DeviceId = addDevice.Id;
            varTable.Name = "默认变量表";
            varTable.Description = "默认变量表";
            varTable.ProtocolType = device.ProtocolType;
            device.VariableTables.Add(varTable);
            var addVarTable = await _varTableRepository.AddAsync(varTable, db);
            // 添加添加变量表的菜单
            var varTableMenu = new MenuBean()
                               {
                                   Name = "默认变量表",
                                   Icon = SegoeFluentIcons.Tablet.Glyph,
                                   Type = MenuType.VariableTableMenu,
                                   ParentId = addDeviceMenuId,
                                   DataId = addVarTable.Id
                               };
            await _menuRepository.AddAsync(varTableMenu, db);
        }

        await _menuRepository.AddVarTableMenuAsync(addDevice, addDeviceMenuId, db);
        return _mapper.Map<Device>(addDevice);
    }

 
}