using System.Collections.ObjectModel;
using System.Windows;
using AutoMapper;
using CommunityToolkit.Mvvm.ComponentModel;
using CommunityToolkit.Mvvm.Messaging;
using DMS.Application.DTOs;
using DMS.Application.DTOs.Events;
using DMS.Core.Models;
using DMS.Application.Interfaces;
using DMS.Core.Enums;
using DMS.Core.Models;
using DMS.Message;
using DMS.WPF.ViewModels.Items;
using Microsoft.IdentityModel.Protocols.OpenIdConnect;
using ObservableCollections;
namespace DMS.WPF.Services;
///
/// 数据服务类,负责从数据库加载和管理各种数据,并提供数据变更通知。
/// 继承自ObservableRecipient,可以接收消息;实现IRecipient,处理加载消息。
///
public partial class DataServices : ObservableObject, IRecipient, IDisposable
{
private readonly IMapper _mapper;
private readonly IAppDataCenterService _appDataCenterService;
private readonly IMqttAppService _mqttAppService;
// 设备列表,使用ObservableProperty特性,当值改变时会自动触发属性变更通知。
[ObservableProperty]
private ObservableCollection _devices;
// 变量表列表。
[ObservableProperty]
private ObservableCollection _variableTables;
// 变量数据列表。
[ObservableProperty]
private ObservableCollection _variables;
// 菜单树列表。
[ObservableProperty]
private ObservableCollection _menus;
// 菜单树列表。
[ObservableProperty]
private ObservableCollection _menuTrees;
// MQTT配置列表。
[ObservableProperty]
private ObservableCollection _mqttServers;
// 日志列表。
[ObservableProperty]
private ObservableCollection _nlogs;
// 设备列表变更事件,当设备列表数据更新时触发。
public event Action> OnDeviceListChanged;
// 菜单树列表变更事件,当菜单树数据更新时触发。
public event Action> OnMenuTreeListChanged;
// MQTT列表变更事件,当MQTT配置数据更新时触发。
public event Action> OnMqttListChanged;
///
/// 处理变量值变更事件
///
private void OnVariableValueChanged(object sender, VariableValueChangedEventArgs e)
{
// 在UI线程上更新变量值
App.Current.Dispatcher.BeginInvoke(new Action(() =>
{
// 查找并更新对应的变量
var variableToUpdate = Variables.FirstOrDefault(v => v.Id == e.VariableId);
if (variableToUpdate != null)
{
variableToUpdate.DataValue = e.NewValue;
variableToUpdate.DisplayValue = e.NewValue;
variableToUpdate.UpdatedAt = e.UpdateTime;
}
}));
}
///
/// 处理日志变更事件
///
private void OnNlogChanged(object sender, NlogChangedEventArgs e)
{
// 在UI线程上更新日志
App.Current.Dispatcher.BeginInvoke(new Action(() =>
{
switch (e.ChangeType)
{
case DataChangeType.Added:
Nlogs.Add(_mapper.Map(e.Nlog));
break;
case DataChangeType.Updated:
var existingLog = Nlogs.FirstOrDefault(l => l.Id == e.Nlog.Id);
if (existingLog != null)
{
_mapper.Map(e.Nlog, existingLog);
}
break;
case DataChangeType.Deleted:
var logToRemove = Nlogs.FirstOrDefault(l => l.Id == e.Nlog.Id);
if (logToRemove != null)
{
Nlogs.Remove(logToRemove);
}
break;
}
}));
}
///
/// DataServices类的构造函数。
/// 初始化各个数据仓库。
///
/// AutoMapper 实例。
/// 数据服务中心实例。
/// MQTT应用服务实例。
public DataServices(IMapper mapper, IAppDataCenterService appDataCenterService, IMqttAppService mqttAppService)
{
_mapper = mapper;
_appDataCenterService = appDataCenterService;
_mqttAppService = mqttAppService;
Devices = new ObservableCollection();
VariableTables = new ObservableCollection();
Variables = new ObservableCollection();
Menus = new ObservableCollection();
MenuTrees = new ObservableCollection();
MqttServers = new ObservableCollection();
Nlogs = new ObservableCollection();
// 监听变量值变更事件
_appDataCenterService.VariableValueChanged += OnVariableValueChanged;
_appDataCenterService.OnLoadDataCompleted += OnLoadDataCompleted;
// 监听日志变更事件
_appDataCenterService.NlogChanged += OnNlogChanged;
// 注册消息接收
// WeakReferenceMessenger.Register(this, (r, m) => r.Receive(m));
}
private void OnLoadDataCompleted(object? sender, DataLoadCompletedEventArgs e)
{
if (e.IsSuccess)
{
LoadAllDatas();
}
}
///
/// 异步加载设备数据,并以高效的方式更新UI集合。
/// 此方法会比较新旧数据,只对有变化的设备进行更新、添加或删除,避免不必要的UI刷新。
///
private void LoadAllDatas()
{
foreach (var deviceDto in _appDataCenterService.Devices.Values)
{
Devices.Add(_mapper.Map(deviceDto));
}
foreach (var device in Devices)
{
foreach (var variableTable in device.VariableTables)
{
VariableTables.Add(variableTable);
foreach (var variable in variableTable.Variables)
{
Variables.Add(variable);
}
}
}
Menus = _mapper.Map>(_appDataCenterService.Menus.Values);
// 加载MQTT服务器数据
MqttServers = _mapper.Map>(_appDataCenterService.MqttServers.Values);
// 加载日志数据
Nlogs = _mapper.Map>(_appDataCenterService.Nlogs.Values);
BuildMenuTrees();
}
public async Task AddDevice(CreateDeviceWithDetailsDto dto)
{
var addDto = await _appDataCenterService.CreateDeviceWithDetailsAsync(dto);
//更新当前界面
Devices.Add(_mapper.Map(addDto.Device));
AddMenuItem(_mapper.Map(addDto.DeviceMenu));
await AddVariableTable(addDto.VariableTable);
AddMenuItem(_mapper.Map(addDto.VariableTableMenu));
//更新数据中心
_appDataCenterService.AddDeviceToMemory(addDto.Device);
_appDataCenterService.AddVariableTableToMemory(addDto.VariableTable);
_appDataCenterService.AddMenuToMemory(addDto.DeviceMenu);
_appDataCenterService.AddMenuToMemory(addDto.VariableTableMenu);
BuildMenuTrees();
return addDto;
}
public async Task DeleteDevice(DeviceItemViewModel device)
{
if (!await _appDataCenterService.DeleteDeviceByIdAsync(device.Id))
{
return false;
}
_appDataCenterService.RemoveDeviceFromMemory(device.Id);
// 1. 删除与设备关联的所有变量表及其变量
foreach (var variableTable in device.VariableTables)
{
// 删除与当前变量表关联的所有变量
DeleteVariableTable(variableTable);
}
// 2. 删除设备
Devices.Remove(device);
// 3. 删除与设备关联的菜单项
var deviceMenu = Menus.FirstOrDefault(m => m.MenuType == MenuType.DeviceMenu && m.TargetId == device.Id);
DeleteMenuItem(deviceMenu);
return true;
}
public async Task UpdateDevice(DeviceItemViewModel device)
{
if (!_appDataCenterService.Devices.TryGetValue(device.Id, out var deviceDto))
{
return false;
}
_mapper.Map(device, deviceDto);
if (await _appDataCenterService.UpdateDeviceAsync(deviceDto) > 0)
{
var menu = Menus.FirstOrDefault(m =>
m.MenuType == MenuType.DeviceMenu &&
m.TargetId == device.Id);
if (menu != null)
{
menu.Header = device.Name;
}
}
return true;
}
public async Task AddVariableTable(VariableTableDto variableTableDto,
MenuBeanDto menuDto = null, bool isAddDb = false)
{
if (variableTableDto == null)
return false;
if (isAddDb && menuDto != null)
{
CreateVariableTableWithMenuDto createDto = new CreateVariableTableWithMenuDto();
createDto.VariableTable = variableTableDto;
createDto.DeviceId = variableTableDto.DeviceId;
createDto.Menu = menuDto;
var resDto = await _appDataCenterService.CreateVariableTableAsync(createDto);
_mapper.Map(resDto.VariableTable, variableTableDto);
AddMenuItem(_mapper.Map(resDto.Menu));
}
_appDataCenterService.AddVariableTableToMemory(variableTableDto);
var device = Devices.FirstOrDefault(d => d.Id == variableTableDto.DeviceId);
if (device != null)
{
var variableTableItemViewModel = _mapper.Map(variableTableDto);
variableTableItemViewModel.Device = device;
device.VariableTables.Add(variableTableItemViewModel);
VariableTables.Add(variableTableItemViewModel);
}
return true;
}
public async Task UpdateVariableTable(VariableTableItemViewModel variableTable)
{
if (variableTable==null)
{
return false;
}
var variableTableDto = _mapper.Map(variableTable);
if (await _appDataCenterService.UpdateVariableTableAsync(variableTableDto) > 0)
{
_appDataCenterService.UpdateVariableTableInMemory(variableTableDto);
var menu = Menus.FirstOrDefault(m =>
m.MenuType == MenuType.VariableTableMenu &&
m.TargetId == variableTable.Id);
if (menu != null)
{
menu.Header = variableTable.Name;
}
return true;
}
return false;
}
public async Task DeleteVariableTable(VariableTableItemViewModel variableTable, bool isDeleteDb = false)
{
if (variableTable == null)
{
return false;
}
if (isDeleteDb)
{
if (!await _appDataCenterService.DeleteVariableTableAsync(variableTable.Id))
{
return false;
}
}
// 删除与当前变量表关联的所有变量
foreach (var variable in variableTable.Variables)
{
Variables.Remove(variable);
}
_appDataCenterService.RemoveVariableTableFromMemory(variableTable.Id);
var variableTableMenu
= Menus.FirstOrDefault(m => m.MenuType == MenuType.VariableTableMenu && m.TargetId == variableTable.Id);
DeleteMenuItem(variableTableMenu);
// 删除变量表
VariableTables.Remove(variableTable);
variableTable.Device.VariableTables.Remove(variableTable);
return true;
}
private void BuildMenuTrees()
{
MenuTrees.Clear();
// 遍历所有菜单项,构建树形结构
foreach (var menu in Menus)
{
var parentMenu = Menus.FirstOrDefault(m => m.Id == menu.ParentId);
// 检查是否有父ID,并且父ID不为0(通常0或null表示根节点)
if (parentMenu != null && menu.ParentId != 0)
{
// 将当前菜单添加到父菜单的Children列表中
if (!parentMenu.Children.Contains(menu))
{
parentMenu.Children.Add(menu);
}
}
else
{
// 如果没有父ID,则这是一个根菜单
MenuTrees.Add(menu);
}
}
}
public void AddMenuItem(MenuItemViewModel menuItemViewModel)
{
if (menuItemViewModel == null)
{
return;
}
var deviceMenu = Menus.FirstOrDefault(m => m.Id == menuItemViewModel.ParentId);
if (deviceMenu != null)
{
deviceMenu.Children.Add(menuItemViewModel);
Menus.Add(menuItemViewModel);
}
}
public void AddVariable(VariableItemViewModel variableItem)
{
if (variableItem == null)
{
return;
}
var variableTable = VariableTables.FirstOrDefault(d => d.Id == variableItem.VariableTableId);
if (variableTable != null)
{
variableTable.Variables.Add(variableItem);
Variables.Add(variableItem);
}
}
private void DeleteMenuItem(MenuItemViewModel? menuItemViewModel)
{
if (menuItemViewModel == null)
{
return;
}
// 从扁平菜单列表中移除
Menus.Remove(menuItemViewModel);
// 从树形结构中移除
if (menuItemViewModel.ParentId.HasValue && menuItemViewModel.ParentId.Value != 0)
{
// 如果有父菜单,从父菜单的Children中移除
var parentMenu = Menus.FirstOrDefault(m => m.Id == menuItemViewModel.ParentId.Value);
parentMenu?.Children.Remove(menuItemViewModel);
}
else
{
// 如果是根菜单,从MenuTrees中移除
MenuTrees.Remove(menuItemViewModel);
}
}
public void DeleteVariable(int id)
{
var variableItem = Variables.FirstOrDefault(v => v.Id == id);
if (variableItem == null)
{
return;
}
var variableTable = VariableTables.FirstOrDefault(vt => vt.Id == variableItem.VariableTableId);
variableTable.Variables.Remove(variableItem);
Variables.Remove(variableItem);
}
///
/// 异步加载所有MQTT服务器数据
///
public async Task LoadMqttServers(IMqttAppService mqttAppService)
{
try
{
var mqttServerDtos = await mqttAppService.GetAllMqttServersAsync();
MqttServers = _mapper.Map>(mqttServerDtos);
OnMqttListChanged?.Invoke(mqttServerDtos);
}
catch (Exception ex)
{
// 记录异常或处理错误
Console.WriteLine($"加载MQTT服务器数据时发生错误: {ex.Message}");
}
}
///
/// 添加MQTT服务器
///
public async Task AddMqttServer(IMqttAppService mqttAppService, MqttServerItemViewModel mqttServer)
{
var dto = _mapper.Map(mqttServer);
var id = await mqttAppService.CreateMqttServerAsync(dto);
dto.Id = id;
var mqttServerItem = _mapper.Map(dto);
MqttServers.Add(mqttServerItem);
return mqttServerItem;
}
///
/// 更新MQTT服务器
///
public async Task UpdateMqttServer(IMqttAppService mqttAppService, MqttServerItemViewModel mqttServer)
{
var dto = _mapper.Map(mqttServer);
await mqttAppService.UpdateMqttServerAsync(dto);
return true;
}
///
/// 删除MQTT服务器
///
public async Task DeleteMqttServer(IMqttAppService mqttAppService, MqttServerItemViewModel mqttServer)
{
await mqttAppService.DeleteMqttServerAsync(mqttServer.Id);
MqttServers.Remove(mqttServer);
return true;
}
///
/// 处理LoadMessage消息
///
/// 加载消息
public void Receive(LoadMessage message)
{
switch (message.LoadType)
{
case LoadTypes.Devices:
// 设备数据已在IDataCenterService中处理
break;
case LoadTypes.Menu:
// 菜单数据已在IDataCenterService中处理
break;
case LoadTypes.Mqtts:
_ = Task.Run(async () => await LoadMqttServers(_mqttAppService));
break;
case LoadTypes.Logs:
// 日志数据已在IDataCenterService中处理
break;
case LoadTypes.All:
// 加载所有数据
LoadAllDatas();
_ = Task.Run(async () => await LoadMqttServers(_mqttAppService));
break;
}
}
///
/// 释放资源
///
public void Dispose()
{
// 取消事件订阅
if (_appDataCenterService != null)
{
_appDataCenterService.VariableValueChanged -= OnVariableValueChanged;
}
}
}