189 lines
7.8 KiB
C#
189 lines
7.8 KiB
C#
// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
|
||
//
|
||
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
|
||
//
|
||
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！
|
||
|
||
namespace Admin.NET.Core;
|
||
|
||
using Microsoft.Extensions.Hosting;
|
||
using MQTTnet.Protocol;
|
||
using MQTTnet.Server;
|
||
using System;
|
||
using System.Text;
|
||
using System.Threading;
|
||
using System.Threading.Tasks;
|
||
|
||
/// <summary>
|
||
/// MQTT 服务
|
||
/// </summary>
|
||
public class MqttHostedService(IOptions<MqttOptions> mqttOptions, ISqlSugarClient db) : IHostedService, IDisposable
|
||
{
|
||
private const string ServerClientId = "Admin.NET.MQTT";
|
||
public MqttServer MqttServer { get; set; }
|
||
private readonly MqttOptions _mqttOptions = mqttOptions.Value;
|
||
private readonly ISqlSugarClient _db = db;
|
||
|
||
public async Task StartAsync(CancellationToken cancellationToken)
|
||
{
|
||
if (!_mqttOptions.Enabled) return;
|
||
|
||
var options = new MqttServerOptionsBuilder()
|
||
.WithDefaultEndpoint() // 默认地址127.0.0.1
|
||
.WithDefaultEndpointPort(_mqttOptions.Port) // 端口号
|
||
//.WithDefaultEndpointBoundIPAddress(_mqttOptions.IPAddress) // IP地址
|
||
.WithConnectionBacklog(1000) // 最大连接数
|
||
.WithPersistentSessions()
|
||
.Build();
|
||
|
||
MqttServer = new MqttServerFactory().CreateMqttServer(options);
|
||
MqttServer.StartedAsync += MqttServer_StartedAsync; // 启动后事件
|
||
MqttServer.StoppedAsync += MqttServer_StoppedAsync; // 关闭后事件
|
||
|
||
MqttServer.ValidatingConnectionAsync += MqttServer_ValidatingConnectionAsync; // 客户端验证事件
|
||
MqttServer.ClientConnectedAsync += MqttServer_ClientConnectedAsync; // 客户端连接事件
|
||
MqttServer.ClientDisconnectedAsync += MqttServer_ClientDisconnectedAsync; // 客户端断开事件
|
||
|
||
MqttServer.ClientSubscribedTopicAsync += MqttServer_ClientSubscribedTopicAsync; // 订阅主题事件
|
||
MqttServer.ClientUnsubscribedTopicAsync += MqttServer_ClientUnsubscribedTopicAsync; // 取消订阅事件
|
||
MqttServer.InterceptingPublishAsync += MqttServer_InterceptingPublishAsync; // 拦截接收消息
|
||
MqttServer.ApplicationMessageNotConsumedAsync += MqttServer_ApplicationMessageNotConsumedAsync; // 消息未被消费
|
||
|
||
await MqttServer.StartAsync();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 启动后事件
|
||
/// </summary>
|
||
/// <param name="arg"></param>
|
||
/// <returns></returns>
|
||
private Task MqttServer_StartedAsync(EventArgs arg)
|
||
{
|
||
Console.WriteLine($"【MQTT】服务已启动，端口：{_mqttOptions.Port}...... {DateTime.Now}");
|
||
return Task.CompletedTask;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 关闭后事件
|
||
/// </summary>
|
||
/// <param name="arg"></param>
|
||
/// <returns></returns>
|
||
private Task MqttServer_StoppedAsync(EventArgs arg)
|
||
{
|
||
Console.WriteLine($"【MQTT】服务已关闭...... {DateTime.Now}");
|
||
return Task.CompletedTask;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 客户端验证事件
|
||
/// </summary>
|
||
/// <param name="arg"></param>
|
||
/// <returns></returns>
|
||
private Task MqttServer_ValidatingConnectionAsync(ValidatingConnectionEventArgs arg)
|
||
{
|
||
arg.ReasonCode = MqttConnectReasonCode.Success;
|
||
|
||
// 验证账号
|
||
var user = _db.Queryable<SysUser>().First(u => u.Account == arg.UserName);
|
||
if (user == null)
|
||
{
|
||
arg.ReasonCode = MqttConnectReasonCode.BadUserNameOrPassword;
|
||
Console.WriteLine($"客户端验证：客户端ID=【{arg.ClientId}】用户名或密码验证错误 {DateTime.Now} ");
|
||
throw Oops.Oh("MQTT客户端验证：账号不能为空或不存在！");
|
||
}
|
||
|
||
// 验证密码
|
||
var password = arg.Password;
|
||
if (CryptogramUtil.CryptoType == CryptogramEnum.MD5.ToString())
|
||
{
|
||
if (user.Password.Equals(MD5Encryption.Encrypt(password))) return Task.CompletedTask;
|
||
}
|
||
else
|
||
{
|
||
if (CryptogramUtil.Decrypt(user.Password).Equals(password)) return Task.CompletedTask;
|
||
}
|
||
arg.ReasonCode = MqttConnectReasonCode.BadUserNameOrPassword;
|
||
Console.WriteLine($"客户端验证：客户端ID=【{arg.ClientId}】用户名或密码验证错误 {DateTime.Now} ");
|
||
throw Oops.Oh("MQTT客户端验证：密码错误！");
|
||
}
|
||
|
||
/// <summary>
|
||
/// 客户端连接事件
|
||
/// </summary>
|
||
/// <param name="arg"></param>
|
||
/// <returns></returns>
|
||
private Task MqttServer_ClientConnectedAsync(ClientConnectedEventArgs arg)
|
||
{
|
||
Console.WriteLine($"客户端连接：客户端ID=【{arg.ClientId}】已连接：用户名=【{arg.UserName}】地址=【{arg.RemoteEndPoint}】 {DateTime.Now}");
|
||
return Task.CompletedTask;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 客户端断开事件
|
||
/// </summary>
|
||
/// <param name="arg"></param>
|
||
/// <returns></returns>
|
||
/// <exception cref="NotImplementedException"></exception>
|
||
private Task MqttServer_ClientDisconnectedAsync(ClientDisconnectedEventArgs arg)
|
||
{
|
||
Console.WriteLine($"客户端断开：客户端ID=【{arg.ClientId}】已断开：用户名=【{arg.UserName}】地址=【{arg.RemoteEndPoint}】 {DateTime.Now}");
|
||
return Task.CompletedTask;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 订阅主题事件
|
||
/// </summary>
|
||
/// <param name="arg"></param>
|
||
/// <returns></returns>
|
||
private Task MqttServer_ClientSubscribedTopicAsync(ClientSubscribedTopicEventArgs arg)
|
||
{
|
||
Console.WriteLine($"订阅主题：客户端ID=【{arg.ClientId}】订阅主题=【{arg.TopicFilter}】 {DateTime.Now}");
|
||
return Task.CompletedTask;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 取消订阅事件
|
||
/// </summary>
|
||
/// <param name="arg"></param>
|
||
/// <returns></returns>
|
||
private Task MqttServer_ClientUnsubscribedTopicAsync(ClientUnsubscribedTopicEventArgs arg)
|
||
{
|
||
Console.WriteLine($"取消订阅：客户端ID=【{arg.ClientId}】取消订阅主题=【{arg.TopicFilter}】 {DateTime.Now}");
|
||
return Task.CompletedTask;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 拦截接收消息
|
||
/// </summary>
|
||
/// <param name="arg"></param>
|
||
/// <returns></returns>
|
||
private Task MqttServer_InterceptingPublishAsync(InterceptingPublishEventArgs arg)
|
||
{
|
||
if (string.Equals(arg.ClientId, ServerClientId))
|
||
return Task.CompletedTask;
|
||
|
||
Console.WriteLine($"拦截消息：客户端ID=【{arg.ClientId}】 Topic主题=【{arg.ApplicationMessage.Topic}】 消息=【{Encoding.UTF8.GetString(arg.ApplicationMessage.Payload)}】 qos等级=【{arg.ApplicationMessage.QualityOfServiceLevel}】 {DateTime.Now}");
|
||
return Task.CompletedTask;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 消息未被消费
|
||
/// </summary>
|
||
/// <param name="arg"></param>
|
||
/// <returns></returns>
|
||
private Task MqttServer_ApplicationMessageNotConsumedAsync(ApplicationMessageNotConsumedEventArgs arg)
|
||
{
|
||
Console.WriteLine($"接收消息：发送端ID=【{arg.SenderId}】 Topic主题=【{arg.ApplicationMessage.Topic}】 消息=【{Encoding.UTF8.GetString(arg.ApplicationMessage.Payload)}】 qos等级=【{arg.ApplicationMessage.QualityOfServiceLevel}】 {DateTime.Now}");
|
||
return Task.CompletedTask;
|
||
}
|
||
|
||
public Task StopAsync(CancellationToken cancellationToken)
|
||
{
|
||
return Task.CompletedTask;
|
||
}
|
||
|
||
public void Dispose()
|
||
{
|
||
throw new NotImplementedException();
|
||
}
|
||
} |