初步给Mqtt添加了别名功能

2025-09-06 16:12:30 +08:00
parent e4cb38cd1d
commit 31c4e77232
6 changed files with 163 additions and 98 deletions
--- a/DMS.Application/Interfaces/IMqttAliasAppService.cs
+++ b/DMS.Application/Interfaces/IMqttAliasAppService.cs
@@ -1,34 +1,37 @@
 using DMS.Application.DTOs;
+using System.Collections.Generic;
+using System.Threading.Tasks;

 namespace DMS.Application.Interfaces;

 /// <summary>
-/// 定义MQTT别名管理相关的应用服务操作。
+/// 定义了MQTT别名管理相关的应用服务操作。
 /// </summary>
 public interface IMqttAliasAppService
 {
    /// <summary>
-    /// 异步根据ID获取MQTT别名DTO。
+    /// 异步获取指定变量的所有MQTT别名关联。
    /// </summary>
-    Task<VariableMqttAliasDto> GetMqttAliasByIdAsync(int id);
+    Task<List<VariableMqttAliasDto>> GetAliasesForVariableAsync(int variableId);

    /// <summary>
-    /// 异步获取所有MQTT别名DTO列表。
+    /// 异步为变量分配或更新一个MQTT别名。
    /// </summary>
-    Task<List<VariableMqttAliasDto>> GetAllMqttAliasesAsync();
-
-    /// <summary>
-    /// 异步创建一个新的MQTT别名。
-    /// </summary>
-    Task<int> CreateMqttAliasAsync(VariableMqttAliasDto mqttAliasDto);
+    /// <param name="variableId">变量ID。</param>
+    /// <param name="mqttServerId">MQTT服务器ID。</param>
+    /// <param name="alias">要设置的别名。</param>
+    Task AssignAliasAsync(int variableId, int mqttServerId, string alias);

    /// <summary>
    /// 异步更新一个已存在的MQTT别名。
    /// </summary>
-    Task UpdateMqttAliasAsync(VariableMqttAliasDto mqttAliasDto);
+    /// <param name="aliasId">别名关联的ID。</param>
+    /// <param name="newAlias">新的别名字符串。</param>
+    Task UpdateAliasAsync(int aliasId, string newAlias);

    /// <summary>
-    /// 异步删除一个MQTT别名。
+    /// 异步移除一个MQTT别名关联。
    /// </summary>
-    Task DeleteMqttAliasAsync(int id);
+    /// <param name="aliasId">要移除的别名关联的ID。</param>
+    Task RemoveAliasAsync(int aliasId);
 }
--- a/DMS.Application/Services/DataCenterService.cs
+++ b/DMS.Application/Services/DataCenterService.cs
@@ -648,35 +648,64 @@ public class DataCenterService : IDataCenterService
            MenuTrees.Clear();
            MqttServers.Clear();

-            // 顺序加载所有数据，避免数据库连接并发问题
-            var devices = await LoadAllDevicesAsync();
-            var variableTables = await LoadAllVariableTablesAsync();
-            var variables = await LoadAllVariablesAsync();
-            var menus = await LoadAllMenusAsync();
+            // 加载所有设备
+            var devices = await _repositoryManager.Devices.GetAllAsync();
+            var deviceDtos = _mapper.Map<List<DeviceDto>>(devices);
+
+            // 加载所有变量表
+            var variableTables = await _repositoryManager.VariableTables.GetAllAsync();
+            var variableTableDtos = _mapper.Map<List<VariableTableDto>>(variableTables);
+
+            // 加载所有变量
+            var variables = await _repositoryManager.Variables.GetAllAsync();
+            var variableDtos = _mapper.Map<List<VariableDto>>(variables);
+
+            // 加载所有菜单
+            var menus = await _repositoryManager.Menus.GetAllAsync();
+            var menuDtos = _mapper.Map<List<MenuBeanDto>>(menus);
+            
            var mqttServers = await LoadAllMqttServersAsync();

-            // 加载设备数据到内存
-            foreach (var device in devices)
+            // 建立设备与变量表的关联
+            foreach (var deviceDto in deviceDtos)
            {
-                Devices.TryAdd(device.Id, device);
+                deviceDto.VariableTables = variableTableDtos
+                                           .Where(vt => vt.DeviceId == deviceDto.Id)
+                                           .ToList();
+
+                // 将设备添加到安全字典
+                Devices.TryAdd(deviceDto.Id, deviceDto);
            }

-            // 加载变量表数据到内存
-            foreach (var variableTable in variableTables)
+            // 建立变量表与变量的关联
+            foreach (var variableTableDto in variableTableDtos)
            {
-                VariableTables.TryAdd(variableTable.Id, variableTable);
+                variableTableDto.Variables = variableDtos
+                                             .Where(v => v.VariableTableId == variableTableDto.Id)
+                                             .ToList();
+                if (Devices.TryGetValue(variableTableDto.DeviceId, out var deviceDto))
+                {
+                    variableTableDto.Device = deviceDto;
+                }
+
+                // 将变量表添加到安全字典
+                VariableTables.TryAdd(variableTableDto.Id, variableTableDto);
            }

-            // 加载变量数据到内存
-            foreach (var variable in variables)
+            // 将变量添加到安全字典
+            foreach (var variableDto in variableDtos)
            {
-                Variables.TryAdd(variable.Id, variable);
+                if (VariableTables.TryGetValue(variableDto.VariableTableId, out var variableTable))
+                {
+                    variableDto.VariableTable = variableTable;
+                }
+                Variables.TryAdd(variableDto.Id, variableDto);
            }

-            // 加载菜单数据到内存
-            foreach (var menu in menus)
+            // 将菜单添加到安全字典
+            foreach (var menuDto in menuDtos)
            {
-                Menus.TryAdd(menu.Id, menu);
+                Menus.TryAdd(menuDto.Id, menuDto);
            }
            
            // 加载MQTT服务器数据到内存
--- a/DMS.Application/Services/MqttAliasAppService.cs
+++ b/DMS.Application/Services/MqttAliasAppService.cs
@@ -1,14 +1,16 @@
 using AutoMapper;
-using DMS.Core.Interfaces;
-using DMS.Core.Models;
 using DMS.Application.DTOs;
 using DMS.Application.Interfaces;
+using DMS.Core.Interfaces;
+using DMS.Core.Models;
+using System.Collections.Generic;
+using System.Threading.Tasks;
+using System;

 namespace DMS.Application.Services;

 /// <summary>
-/// MQTT别名应用服务，负责处理MQTT别名相关的业务逻辑。
-/// 实现 <see cref="IMqttAliasAppService"/> 接口。
+/// IMqttAliasAppService 的实现，负责管理变量与MQTT服务器的别名关联。
 /// </summary>
 public class MqttAliasAppService : IMqttAliasAppService
 {
@@ -16,10 +18,8 @@ public class MqttAliasAppService : IMqttAliasAppService
    private readonly IMapper _mapper;

    /// <summary>
-    /// 构造函数，通过依赖注入获取仓储管理器和AutoMapper实例。
+    /// 构造函数。
    /// </summary>
-    /// <param name="repoManager">仓储管理器实例。</param>
-    /// <param name="mapper">AutoMapper 实例。</param>
    public MqttAliasAppService(IRepositoryManager repoManager, IMapper mapper)
    {
        _repoManager = repoManager;
@@ -27,94 +27,93 @@ public class MqttAliasAppService : IMqttAliasAppService
    }

    /// <summary>
-    /// 异步根据ID获取MQTT别名数据传输对象。
+    /// 异步获取指定变量的所有MQTT别名关联。
    /// </summary>
-    /// <param name="id">MQTT别名ID。</param>
-    /// <returns>MQTT别名数据传输对象。</returns>
-    public async Task<VariableMqttAliasDto> GetMqttAliasByIdAsync(int id)
+    public async Task<List<VariableMqttAliasDto>> GetAliasesForVariableAsync(int variableId)
    {
-        var mqttAlias = await _repoManager.VariableMqttAliases.GetByIdAsync(id);
-        return _mapper.Map<VariableMqttAliasDto>(mqttAlias);
+        // 从仓储获取别名，并确保加载了关联的MqttServer信息
+        var aliases = await _repoManager.VariableMqttAliases.GetAliasesForVariableAsync(variableId);
+        return _mapper.Map<List<VariableMqttAliasDto>>(aliases);
    }

    /// <summary>
-    /// 异步获取所有MQTT别名数据传输对象列表。
+    /// 异步为变量分配或更新一个MQTT别名。
    /// </summary>
-    /// <returns>MQTT别名数据传输对象列表。</returns>
-    public async Task<List<VariableMqttAliasDto>> GetAllMqttAliasesAsync()
-    {
-        var mqttAliases = await _repoManager.VariableMqttAliases.GetAllAsync();
-        return _mapper.Map<List<VariableMqttAliasDto>>(mqttAliases);
-    }
-
-    /// <summary>
-    /// 异步创建一个新MQTT别名（事务性操作）。
-    /// </summary>
-    /// <param name="mqttAliasDto">要创建的MQTT别名数据传输对象。</param>
-    /// <returns>新创建MQTT别名的ID。</returns>
-    /// <exception cref="ApplicationException">如果创建MQTT别名时发生错误。</exception>
-    public async Task<int> CreateMqttAliasAsync(VariableMqttAliasDto mqttAliasDto)
+    public async Task AssignAliasAsync(int variableId, int mqttServerId, string alias)
    {
        try
        {
            await _repoManager.BeginTranAsync();
-            var mqttAlias = _mapper.Map<VariableMqttAlias>(mqttAliasDto);
-            await _repoManager.VariableMqttAliases.AddAsync(mqttAlias);
-            await _repoManager.CommitAsync();
-            return mqttAlias.Id;
-        }
-        catch (Exception ex)
-        {
-            await _repoManager.RollbackAsync();
-            throw new ApplicationException("创建MQTT别名时发生错误，操作已回滚。", ex);
-        }
-    }

-    /// <summary>
-    /// 异步更新一个已存在的MQTT别名（事务性操作）。
-    /// </summary>
-    /// <param name="mqttAliasDto">要更新的MQTT别名数据传输对象。</param>
-    /// <returns>表示异步操作的任务。</returns>
-    /// <exception cref="ApplicationException">如果找不到MQTT别名或更新MQTT别名时发生错误。</exception>
-    public async Task UpdateMqttAliasAsync(VariableMqttAliasDto mqttAliasDto)
-    {
-        try
-        {
-            await _repoManager.BeginTranAsync();
-            var mqttAlias = await _repoManager.VariableMqttAliases.GetByIdAsync(mqttAliasDto.Id);
-            if (mqttAlias == null)
+            // 检查是否已存在该变量与该服务器的关联
+            var existingAlias = await _repoManager.VariableMqttAliases.GetByVariableAndServerAsync(variableId, mqttServerId);
+
+            if (existingAlias != null)
            {
-                throw new ApplicationException($"MQTT Alias with ID {mqttAliasDto.Id} not found.");
+                // 如果存在，则更新别名
+                existingAlias.Alias = alias;
+                await _repoManager.VariableMqttAliases.UpdateAsync(existingAlias);
            }
-            _mapper.Map(mqttAliasDto, mqttAlias);
-            await _repoManager.VariableMqttAliases.UpdateAsync(mqttAlias);
+            else
+            {
+                // 如果不存在，则创建新的关联
+                var newAlias = new VariableMqttAlias
+                {
+                    VariableId = variableId,
+                    MqttServerId = mqttServerId,
+                    Alias = alias
+                };
+                await _repoManager.VariableMqttAliases.AddAsync(newAlias);
+            }
+
            await _repoManager.CommitAsync();
        }
        catch (Exception ex)
        {
            await _repoManager.RollbackAsync();
-            throw new ApplicationException("更新MQTT别名时发生错误，操作已回滚。", ex);
+            throw new ApplicationException("分配/更新MQTT别名失败。", ex);
        }
    }

    /// <summary>
-    /// 异步删除一个MQTT别名（事务性操作）。
+    /// 异步更新一个已存在的MQTT别名。
    /// </summary>
-    /// <param name="id">要删除MQTT别名的ID。</param>
-    /// <returns>表示异步操作的任务。</returns>
-    /// <exception cref="ApplicationException">如果删除MQTT别名时发生错误。</exception>
-    public async Task DeleteMqttAliasAsync(int id)
+    public async Task UpdateAliasAsync(int aliasId, string newAlias)
    {
        try
        {
            await _repoManager.BeginTranAsync();
-            await _repoManager.VariableMqttAliases.DeleteByIdAsync(id);
+            var aliasToUpdate = await _repoManager.VariableMqttAliases.GetByIdAsync(aliasId);
+            if (aliasToUpdate == null)
+            {
+                throw new KeyNotFoundException($"未找到ID为 {aliasId} 的MQTT别名关联。");
+            }
+            aliasToUpdate.Alias = newAlias;
+            await _repoManager.VariableMqttAliases.UpdateAsync(aliasToUpdate);
            await _repoManager.CommitAsync();
        }
        catch (Exception ex)
        {
            await _repoManager.RollbackAsync();
-            throw new ApplicationException("删除MQTT别名时发生错误，操作已回滚。", ex);
+            throw new ApplicationException("更新MQTT别名失败。", ex);
+        }
+    }
+
+    /// <summary>
+    /// 异步移除一个MQTT别名关联。
+    /// </summary>
+    public async Task RemoveAliasAsync(int aliasId)
+    {
+        try
+        {
+            await _repoManager.BeginTranAsync();
+            await _repoManager.VariableMqttAliases.DeleteByIdAsync(aliasId);
+            await _repoManager.CommitAsync();
+        }
+        catch (Exception ex)
+        {
+            await _repoManager.RollbackAsync();
+            throw new ApplicationException("移除MQTT别名失败。", ex);
        }
    }
 }
--- a/DMS.Core/Interfaces/Repositories/IVariableMqttAliasRepository.cs
+++ b/DMS.Core/Interfaces/Repositories/IVariableMqttAliasRepository.cs
@@ -1,9 +1,19 @@

+using System.Collections.Generic;
+using System.Threading.Tasks;
+
 namespace DMS.Core.Interfaces.Repositories
 {
-    public interface IVariableMqttAliasRepository:IBaseRepository<VariableMqttAlias>
+    public interface IVariableMqttAliasRepository : IBaseRepository<VariableMqttAlias>
    {
+        /// <summary>
+        /// 异步获取指定变量的所有MQTT别名关联。
+        /// </summary>
+        Task<List<VariableMqttAlias>> GetAliasesForVariableAsync(int variableId);

-
+        /// <summary>
+        /// 异步根据变量和服务器获取别名关联。
+        /// </summary>
+        Task<VariableMqttAlias> GetByVariableAndServerAsync(int variableId, int mqttServerId);
    }
 }
--- a/DMS.Infrastructure/Repositories/VariableMqttAliasRepository.cs
+++ b/DMS.Infrastructure/Repositories/VariableMqttAliasRepository.cs
@@ -2,11 +2,12 @@ using System.Diagnostics;
 using DMS.Core.Interfaces.Repositories;
 using DMS.Infrastructure.Data;
 using DMS.Infrastructure.Entities;
-
-
 using AutoMapper;
 using Microsoft.Extensions.Logging;
 using DMS.Core.Models;
+using System.Collections.Generic;
+using System.Threading.Tasks;
+using System.Linq;

 namespace DMS.Infrastructure.Repositories;

@@ -109,4 +110,26 @@ public class VariableMqttAliasRepository : BaseRepository<DbVariableMqttAlias>,
        var dbEntities = _mapper.Map<List<DbVariableMqttAlias>>(entities);
        return base.AddBatchAsync(dbEntities);
    }
+
+    /// <summary>
+    /// 异步获取指定变量的所有MQTT别名关联。
+    /// </summary>
+    public async Task<List<VariableMqttAlias>> GetAliasesForVariableAsync(int variableId)
+    {
+        var dbList = await Db.Queryable<DbVariableMqttAlias>()
+            .Where(x => x.VariableId == variableId)
+            .ToListAsync();
+        return _mapper.Map<List<VariableMqttAlias>>(dbList);
+    }
+
+    /// <summary>
+    /// 异步根据变量和服务器获取别名关联。
+    /// </summary>
+    public async Task<VariableMqttAlias> GetByVariableAndServerAsync(int variableId, int mqttServerId)
+    {
+        var dbAlias = await Db.Queryable<DbVariableMqttAlias>()
+            .Where(x => x.VariableId == variableId && x.MqttServerId == mqttServerId)
+            .FirstAsync();
+        return _mapper.Map<VariableMqttAlias>(dbAlias);
+    }
 }
--- a/DMS.WPF/App.xaml.cs
+++ b/DMS.WPF/App.xaml.cs
@@ -171,6 +171,7 @@ public partial class App : System.Windows.Application
        
        // 注册MQTT服务管理器
        services.AddSingleton<IMqttServiceManager, MqttServiceManager>();
+        services.AddSingleton<IMqttAliasAppService, MqttAliasAppService>();
        services.AddHostedService<MqttBackgroundService>();
        
        // 注册WPF中的服务