David P.G
a6fc543e4f
2
3 1. 在MqttServer和MqttServerDto模型中添加IsConnect属性，用于跟踪连接状态
4 2. 重构MqttManagementService服务，使用事件驱动方式管理服务器状态变化
5 3. 实现MqttServerChangedEventArgs事件参数类，支持区分不同变更类型
6 4. 在IEventService中添加OnMqttServerChanged事件，实现事件通知机制
7 5. 优化数据存储结构，将MqttServers从ObservableCollection改为ObservableDictionary
8 6. 更新MqttServiceManager以正确处理连接状态和事件触发
9 7. 在WPF层更新UI以响应服务器状态变化
10 8. 删除不再需要的Helper类(DataServicesHelper, MessageHelper, SiemensHelper)
11 9. 在NLog配置中添加调试器输出目标以便调试
12 10. 完善VariableHistoryViewModel防止空引用异常
368 lines
16 KiB
C#
368 lines
16 KiB
C#
using System.Collections.Concurrent;
|
||
using System.Diagnostics;
|
||
using System.Text;
|
||
using AutoMapper;
|
||
using DMS.Application.DTOs;
|
||
using DMS.Application.Events;
|
||
using DMS.Application.Interfaces;
|
||
using DMS.Core.Enums;
|
||
using DMS.Core.Interfaces.Services;
|
||
using DMS.Core.Models;
|
||
using DMS.Infrastructure.Interfaces.Services;
|
||
using Microsoft.Extensions.Logging;
|
||
|
||
namespace DMS.Infrastructure.Services.Mqtt
|
||
{
|
||
/// <summary>
|
||
/// MQTT服务管理器，负责管理MQTT连接和变量监控
|
||
/// </summary>
|
||
public class MqttServiceManager : IMqttServiceManager
|
||
{
|
||
private readonly ILogger<MqttServiceManager> _logger;
|
||
private readonly IDataProcessingService _dataProcessingService;
|
||
private readonly IAppDataCenterService _appDataCenterService;
|
||
private readonly IMqttServiceFactory _mqttServiceFactory;
|
||
private readonly IEventService _eventService;
|
||
private readonly IMapper _mapper;
|
||
private readonly ConcurrentDictionary<int, MqttDeviceContext> _mqttContexts;
|
||
private readonly SemaphoreSlim _semaphore;
|
||
private bool _disposed = false;
|
||
|
||
public MqttServiceManager(
|
||
ILogger<MqttServiceManager> logger,
|
||
IDataProcessingService dataProcessingService,
|
||
IAppDataCenterService appDataCenterService,
|
||
IMqttServiceFactory mqttServiceFactory,
|
||
IEventService eventService,
|
||
IMapper mapper)
|
||
{
|
||
_logger = logger ?? throw new ArgumentNullException(nameof(logger));
|
||
_dataProcessingService = dataProcessingService ?? throw new ArgumentNullException(nameof(dataProcessingService));
|
||
_appDataCenterService = appDataCenterService ?? throw new ArgumentNullException(nameof(appDataCenterService));
|
||
_mqttServiceFactory = mqttServiceFactory ?? throw new ArgumentNullException(nameof(mqttServiceFactory));
|
||
_eventService = eventService ?? throw new ArgumentNullException(nameof(eventService));
|
||
_mapper = mapper ?? throw new ArgumentNullException(nameof(mapper));
|
||
_mqttContexts = new ConcurrentDictionary<int, MqttDeviceContext>();
|
||
_semaphore = new SemaphoreSlim(10, 10); // 默认最大并发连接数为10
|
||
}
|
||
|
||
/// <summary>
|
||
/// 初始化服务管理器
|
||
/// </summary>
|
||
public async Task InitializeAsync(CancellationToken cancellationToken = default)
|
||
{
|
||
_logger.LogInformation("MQTT服务管理器正在初始化...");
|
||
// 初始化逻辑可以在需要时添加
|
||
_logger.LogInformation("MQTT服务管理器初始化完成");
|
||
}
|
||
|
||
/// <summary>
|
||
/// 添加MQTT服务器到监控列表
|
||
/// </summary>
|
||
public void AddMqttServer(MqttServer mqttServer)
|
||
{
|
||
if (mqttServer == null)
|
||
throw new ArgumentNullException(nameof(mqttServer));
|
||
|
||
var context = new MqttDeviceContext
|
||
{
|
||
MqttServerConfig = mqttServer,
|
||
MqttService = _mqttServiceFactory.CreateService(),
|
||
};
|
||
|
||
_mqttContexts.AddOrUpdate(mqttServer.Id, context, (key, oldValue) => context);
|
||
_logger.LogInformation("已添加MQTT服务器 {MqttServerId} 到监控列表", mqttServer.Id);
|
||
|
||
// 使用AutoMapper触发MQTT服务器改变事件
|
||
var mqttServerDto = _mapper.Map<MqttServerDto>(mqttServer);
|
||
|
||
_eventService.RaiseMqttServerChanged(this, new MqttServerChangedEventArgs(Core.Enums.ActionChangeType.Added, mqttServerDto));
|
||
}
|
||
|
||
/// <summary>
|
||
/// 移除MQTT服务器监控
|
||
/// </summary>
|
||
public async Task RemoveMqttServerAsync(int mqttServerId, CancellationToken cancellationToken = default)
|
||
{
|
||
if (_mqttContexts.TryRemove(mqttServerId, out var context))
|
||
{
|
||
await DisconnectMqttServerAsync(mqttServerId, cancellationToken);
|
||
_logger.LogInformation("已移除MQTT服务器 {MqttServerId} 的监控", mqttServerId);
|
||
|
||
// 使用AutoMapper触发MQTT服务器删除事件
|
||
var mqttServerDto = _mapper.Map<MqttServerDto>(context.MqttServerConfig);
|
||
|
||
_eventService.RaiseMqttServerChanged(this, new MqttServerChangedEventArgs(Core.Enums.ActionChangeType.Deleted, mqttServerDto));
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 更新MQTT服务器变量别名
|
||
/// </summary>
|
||
public void UpdateVariableMqttAliases(int mqttServerId, List<VariableMqttAlias> variableMqttAliases)
|
||
{
|
||
if (_mqttContexts.TryGetValue(mqttServerId, out var context))
|
||
{
|
||
context.VariableMqttAliases.Clear();
|
||
foreach (var alias in variableMqttAliases)
|
||
{
|
||
context.VariableMqttAliases.AddOrUpdate(alias.Id, alias, (key, oldValue) => alias);
|
||
}
|
||
_logger.LogInformation("已更新MQTT服务器 {MqttServerId} 的变量别名列表，共 {Count} 个别名", mqttServerId, variableMqttAliases.Count);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取MQTT服务器连接状态
|
||
/// </summary>
|
||
public bool IsMqttServerConnected(int mqttServerId)
|
||
{
|
||
return _mqttContexts.TryGetValue(mqttServerId, out var context) && context.MqttService.IsConnected;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 重新连接MQTT服务器
|
||
/// </summary>
|
||
public async Task ReconnectMqttServerAsync(int mqttServerId, CancellationToken cancellationToken = default)
|
||
{
|
||
if (_mqttContexts.TryGetValue(mqttServerId, out var context))
|
||
{
|
||
await DisconnectMqttServerAsync(mqttServerId, cancellationToken);
|
||
await ConnectMqttServerAsync(mqttServerId, cancellationToken);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取所有监控的MQTT服务器ID
|
||
/// </summary>
|
||
public IEnumerable<int> GetMonitoredMqttServerIds()
|
||
{
|
||
return _mqttContexts.Keys.ToList();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 连接MQTT服务器
|
||
/// </summary>
|
||
public async Task ConnectMqttServerAsync(int mqttServerId, CancellationToken cancellationToken = default)
|
||
{
|
||
if (!_mqttContexts.TryGetValue(mqttServerId, out var context))
|
||
{
|
||
_logger.LogWarning("未找到MQTT服务器 {MqttServerId}", mqttServerId);
|
||
return;
|
||
}
|
||
|
||
await _semaphore.WaitAsync(cancellationToken);
|
||
try
|
||
{
|
||
_logger.LogInformation("正在连接MQTT服务器 {ServerName} ({ServerUrl}:{Port})",
|
||
context.MqttServerConfig.ServerName, context.MqttServerConfig.ServerUrl, context.MqttServerConfig.Port);
|
||
|
||
var stopwatch = Stopwatch.StartNew();
|
||
|
||
// 设置连接超时
|
||
using var timeoutToken = new CancellationTokenSource(5000); // 5秒超时
|
||
using var linkedToken = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, timeoutToken.Token);
|
||
|
||
await context.MqttService.ConnectAsync(
|
||
context.MqttServerConfig.ServerUrl,
|
||
context.MqttServerConfig.Port,
|
||
context.MqttServerConfig.ClientId,
|
||
context.MqttServerConfig.Username,
|
||
context.MqttServerConfig.Password);
|
||
|
||
stopwatch.Stop();
|
||
_logger.LogInformation("MQTT服务器 {ServerName} 连接耗时 {ElapsedMs} ms",
|
||
context.MqttServerConfig.ServerName, stopwatch.ElapsedMilliseconds);
|
||
if (context.MqttService.IsConnected)
|
||
{
|
||
context.ReconnectAttempts = 0; // 重置重连次数
|
||
context.MqttServerConfig.IsConnect=true;
|
||
// 订阅主题
|
||
if (!string.IsNullOrEmpty(context.MqttServerConfig.SubscribeTopic))
|
||
{
|
||
await context.MqttService.SubscribeAsync(context.MqttServerConfig.SubscribeTopic);
|
||
}
|
||
|
||
_logger.LogInformation("MQTT服务器 {ServerName} 连接成功", context.MqttServerConfig.ServerName);
|
||
|
||
//
|
||
}
|
||
else
|
||
{
|
||
context.MqttServerConfig.IsConnect = false;
|
||
_logger.LogWarning("MQTT服务器 {ServerName} 连接失败", context.MqttServerConfig.ServerName);
|
||
}
|
||
//触发MQTT连接状态改变事件
|
||
_eventService.RaiseMqttServerChanged(this, new MqttServerChangedEventArgs(ActionChangeType.Updated, _mapper.Map<MqttServerDto>(context.MqttServerConfig), MqttServerPropertyType.IsConnect));
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
_logger.LogError(ex, "连接MQTT服务器 {ServerName} 时发生错误: {ErrorMessage}",
|
||
context.MqttServerConfig.ServerName, ex.Message);
|
||
context.ReconnectAttempts++;
|
||
context.MqttServerConfig.IsConnect = false;
|
||
_eventService.RaiseMqttServerChanged(this, new MqttServerChangedEventArgs(ActionChangeType.Updated, _mapper.Map<MqttServerDto>(context.MqttServerConfig), MqttServerPropertyType.IsConnect));
|
||
}
|
||
finally
|
||
{
|
||
_semaphore.Release();
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 断开MQTT服务器连接
|
||
/// </summary>
|
||
public async Task DisconnectMqttServerAsync(int mqttServerId, CancellationToken cancellationToken = default)
|
||
{
|
||
if (!_mqttContexts.TryGetValue(mqttServerId, out var context))
|
||
return;
|
||
|
||
try
|
||
{
|
||
_logger.LogInformation("正在断开MQTT服务器 {ServerName} 的连接", context.MqttServerConfig.ServerName);
|
||
await context.MqttService.DisconnectAsync();
|
||
_logger.LogInformation("MQTT服务器 {ServerName} 连接已断开", context.MqttServerConfig.ServerName);
|
||
|
||
// 如果连接状态从连接变为断开，触发事件
|
||
context.MqttServerConfig.IsConnect = false;
|
||
_eventService.RaiseMqttServerChanged(this, new MqttServerChangedEventArgs(ActionChangeType.Updated, _mapper.Map<MqttServerDto>(context.MqttServerConfig), MqttServerPropertyType.IsConnect));
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
_logger.LogError(ex, "断开MQTT服务器 {ServerName} 连接时发生错误: {ErrorMessage}",
|
||
context.MqttServerConfig.ServerName, ex.Message);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 发布变量数据到MQTT服务器
|
||
/// </summary>
|
||
public async Task PublishVariableDataAsync(VariableMqttAlias variableMqtt, CancellationToken cancellationToken = default)
|
||
{
|
||
if (variableMqtt?.MqttServer == null || variableMqtt.Variable == null)
|
||
{
|
||
_logger.LogWarning("无效的VariableMqtt对象，跳过发布");
|
||
return;
|
||
}
|
||
|
||
if (!_mqttContexts.TryGetValue(variableMqtt.MqttServer.Id, out var context))
|
||
{
|
||
_logger.LogWarning("未找到MQTT服务器 {MqttServerId}", variableMqtt.MqttServer.Id);
|
||
return;
|
||
}
|
||
|
||
if (!context.MqttService.IsConnected)
|
||
{
|
||
_logger.LogWarning("MQTT服务器 {ServerName} 未连接，跳过发布", context.MqttServerConfig.ServerName);
|
||
return;
|
||
}
|
||
|
||
try
|
||
{
|
||
var topic = context.MqttServerConfig.PublishTopic;
|
||
|
||
var sendMsg = BuildSendMessage(variableMqtt);
|
||
|
||
await context.MqttService.PublishAsync(topic, sendMsg);
|
||
_logger.LogDebug("成功向MQTT服务器 {ServerName} 发布变量 {VariableName} 的数据：{sendMsg}",
|
||
context.MqttServerConfig.ServerName, variableMqtt.Variable.Name, sendMsg);
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
_logger.LogError(ex, "向MQTT服务器 {ServerName} 发布变量 {VariableName} 数据时发生错误: {ErrorMessage}",
|
||
context.MqttServerConfig.ServerName, variableMqtt.Variable.Name, ex.Message);
|
||
}
|
||
}
|
||
|
||
private string BuildSendMessage(VariableMqttAlias variableMqtt)
|
||
{
|
||
StringBuilder sb = new StringBuilder();
|
||
var now = DateTime.Now;
|
||
var timestamp = ((DateTimeOffset)now).ToUnixTimeMilliseconds();
|
||
sb.Append(variableMqtt.MqttServer.MessageHeader.Replace("{timestamp}", timestamp.ToString()));
|
||
sb.Append(variableMqtt.MqttServer.MessageContent.Replace("{name}", variableMqtt.Alias).Replace("{value}", variableMqtt.Variable.DataValue));
|
||
sb.Append(variableMqtt.MqttServer.MessageFooter);
|
||
|
||
return sb.ToString();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 发布批量变量数据到MQTT服务器
|
||
/// </summary>
|
||
public async Task PublishVariablesDataAsync(List<VariableMqtt> variableMqtts, CancellationToken cancellationToken = default)
|
||
{
|
||
if (variableMqtts == null || !variableMqtts.Any())
|
||
{
|
||
_logger.LogWarning("变量MQTT列表为空，跳过批量发布");
|
||
return;
|
||
}
|
||
|
||
// 按MQTT服务器ID进行分组
|
||
var groupedByMqtt = variableMqtts.GroupBy(vm => vm.Mqtt.Id);
|
||
|
||
foreach (var group in groupedByMqtt)
|
||
{
|
||
var mqttId = group.Key;
|
||
if (!_mqttContexts.TryGetValue(mqttId, out var context))
|
||
{
|
||
_logger.LogWarning("未找到MQTT服务器 {MqttServerId}，跳过 {Count} 条消息", mqttId, group.Count());
|
||
continue;
|
||
}
|
||
|
||
if (!context.MqttService.IsConnected)
|
||
{
|
||
_logger.LogWarning("MQTT服务器 {ServerName} 未连接，跳过 {Count} 条消息",
|
||
context.MqttServerConfig.ServerName, group.Count());
|
||
continue;
|
||
}
|
||
|
||
try
|
||
{
|
||
foreach (var variableMqtt in group)
|
||
{
|
||
var topic = context.MqttServerConfig.PublishTopic;
|
||
var payload = variableMqtt.Variable?.DataValue?.ToString() ?? string.Empty;
|
||
|
||
await context.MqttService.PublishAsync(topic, payload);
|
||
}
|
||
|
||
_logger.LogInformation("成功向MQTT服务器 {ServerName} 发布 {Count} 条变量数据",
|
||
context.MqttServerConfig.ServerName, group.Count());
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
_logger.LogError(ex, "向MQTT服务器 {ServerName} 批量发布变量数据时发生错误: {ErrorMessage}",
|
||
context.MqttServerConfig.ServerName, ex.Message);
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 释放资源
|
||
/// </summary>
|
||
public void Dispose()
|
||
{
|
||
if (!_disposed)
|
||
{
|
||
_disposed = true;
|
||
_semaphore?.Dispose();
|
||
|
||
// 断开所有MQTT连接
|
||
foreach (var context in _mqttContexts.Values)
|
||
{
|
||
try
|
||
{
|
||
context.MqttService?.DisconnectAsync().Wait(TimeSpan.FromSeconds(5));
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
_logger.LogError(ex, "断开MQTT服务器 {ServerName} 连接时发生错误",
|
||
context.MqttServerConfig?.ServerName ?? "Unknown");
|
||
}
|
||
}
|
||
|
||
_logger.LogInformation("MQTT服务管理器已释放资源");
|
||
}
|
||
}
|
||
}
|
||
} |