临时提交
This commit is contained in:
@@ -2,28 +2,30 @@ using System.Diagnostics;
|
||||
using AutoMapper;
|
||||
using DMS.Infrastructure.Entities;
|
||||
using DMS.Core.Enums;
|
||||
using DMS.Helper;
|
||||
using DMS.Models;
|
||||
using iNKORE.UI.WPF.Modern.Common.IconKeys;
|
||||
using DMS.Extensions;
|
||||
using DMS.Core.Helper;
|
||||
using DMS.Core.Models;
|
||||
using DMS.Infrastructure.Data;
|
||||
using SqlSugar;
|
||||
|
||||
namespace DMS.Infrastructure.Repositories;
|
||||
|
||||
public class DeviceRepository
|
||||
public class DeviceRepository : IDeviceRepository
|
||||
{
|
||||
private readonly IMapper _mapper;
|
||||
private readonly MenuRepository _menuRepository;
|
||||
private readonly VarTableRepository _varTableRepository;
|
||||
private readonly IMenuRepository _menuRepository;
|
||||
private readonly IVarTableRepository _varTableRepository;
|
||||
private readonly IUnitOfWork _unitOfWork;
|
||||
|
||||
public DeviceRepository(IMapper mapper,MenuRepository menuRepository,VarTableRepository varTableRepository)
|
||||
public DeviceRepository(IMapper mapper, IMenuRepository menuRepository, IVarTableRepository varTableRepository, IUnitOfWork unitOfWork)
|
||||
{
|
||||
_mapper = mapper;
|
||||
_menuRepository = menuRepository;
|
||||
_varTableRepository = varTableRepository;
|
||||
_unitOfWork = unitOfWork;
|
||||
}
|
||||
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// 编辑设备信息。
|
||||
/// </summary>
|
||||
@@ -33,25 +35,7 @@ public class DeviceRepository
|
||||
{
|
||||
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 db = _unitOfWork.GetInstance();
|
||||
var result = await db.Updateable<DbDevice>(_mapper.Map<DbDevice>(device))
|
||||
.ExecuteCommandAsync();
|
||||
stopwatch.Stop();
|
||||
@@ -59,6 +43,8 @@ public class DeviceRepository
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// 获取设备列表
|
||||
@@ -73,13 +59,13 @@ public class DeviceRepository
|
||||
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 )
|
||||
.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);
|
||||
var devices = _mapper.Map<List<Device>>(dlist);
|
||||
return devices;
|
||||
}
|
||||
}
|
||||
@@ -111,59 +97,23 @@ public class DeviceRepository
|
||||
/// <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)
|
||||
{
|
||||
var db = _unitOfWork.GetInstance();
|
||||
Stopwatch stopwatch = new Stopwatch();
|
||||
stopwatch.Start();
|
||||
var result = await db.Deleteable<DbDevice>(new DbDevice { Id = device.Id })
|
||||
.ExecuteCommandAsync();
|
||||
// 删除变量表
|
||||
await _varTableRepository.DeleteAsync(device.VariableTables, db);
|
||||
await _varTableRepository.DeleteAsync(device.VariableTables);
|
||||
|
||||
// 删除菜单
|
||||
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>
|
||||
@@ -172,86 +122,25 @@ public class DeviceRepository
|
||||
{
|
||||
Stopwatch stopwatch = new Stopwatch();
|
||||
stopwatch.Start();
|
||||
var db = DbContext.GetInstance();
|
||||
try
|
||||
{
|
||||
// 开启事务
|
||||
await db.BeginTranAsync();
|
||||
//查询设备的名字是否存在
|
||||
var db = _unitOfWork.GetInstance();
|
||||
|
||||
var exist = await db.Queryable<DbDevice>()
|
||||
.Where(d => d.Name == device.Name)
|
||||
.FirstAsync();
|
||||
if (exist != null)
|
||||
throw new InvalidOperationException("设备名称已经存在。");
|
||||
//查询设备的名字是否存在
|
||||
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)
|
||||
{
|
||||
;
|
||||
// 添加设备
|
||||
// 2. 将设备添加到数据库
|
||||
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);
|
||||
}
|
||||
var addDeviceMenuId = await _menuRepository.AddAsync(addDevice);
|
||||
|
||||
await _menuRepository.AddVarTableMenuAsync(addDevice, addDeviceMenuId, db);
|
||||
return _mapper.Map<Device>(addDevice);
|
||||
stopwatch.Stop();
|
||||
NlogHelper.Info($"添加设备 '{device.Name}' 及相关菜单耗时：{stopwatch.ElapsedMilliseconds}ms");
|
||||
}
|
||||
|
||||
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user