添加项目文件。

2025-09-29 13:48:36 +08:00
parent f61e922f0c
commit 5dc1780ef1
407 changed files with 63703 additions and 0 deletions
@@ -0,0 +1,76 @@
+namespace SimpleAdmin.System;
+
+/// <summary>
+/// <inheritdoc cref="IMqttService"/>
+/// </summary>
+public class MqttService : IMqttService
+{
+    private readonly ISimpleCacheService _simpleCacheService;
+    private readonly ISysUserService _sysUserService;
+    private readonly IConfigService _configService;
+
+    public MqttService(ISimpleCacheService simpleCacheService, ISysUserService sysUserService, IConfigService configService)
+    {
+        _simpleCacheService = simpleCacheService;
+        _sysUserService = sysUserService;
+        _configService = configService;
+    }
+
+    /// <inheritdoc/>
+    public async Task<MqttParameterOutput> GetWebLoginParameter()
+    {
+        var user = await _sysUserService.GetUserById(UserManager.UserId);//获取用户信息
+        var token = JWTEncryption.GetJwtBearerToken((DefaultHttpContext)App.HttpContext);// 获取当前token
+        //获取mqtt配置
+        var mqttconfig = await _configService.GetConfigsByCategory(CateGoryConst.CONFIG_MQTT_BASE);
+        //地址
+        var url = mqttconfig.Where(it => it.ConfigKey == SysConfigConst.MQTT_PARAM_URL).Select(it => it.ConfigValue).FirstOrDefault();
+        //用户名
+        var userName = mqttconfig.Where(it => it.ConfigKey == SysConfigConst.MQTT_PARAM_USERNAME).Select(it => it.ConfigValue).FirstOrDefault();
+        //密码
+        var password = mqttconfig.Where(it => it.ConfigKey == SysConfigConst.MQTT_PARAM_PASSWORD).Select(it => it.ConfigValue).FirstOrDefault();
+
+        #region 用户名特殊处理
+
+        if (userName.ToLower() == "$username")
+            userName = user.Account;
+        else if (userName.ToLower() == "$userid")
+            userName = user.Id.ToString();
+
+        #endregion 用户名特殊处理
+
+        #region 密码特殊处理
+
+        if (password.ToLower() == "$username")
+            password = token;// 当前token作为mqtt密码
+
+        #endregion 密码特殊处理
+
+        var clientId = $"{user.Id}_{RandomHelper.CreateLetterAndNumber(5)}";//客户端ID
+        _simpleCacheService.Set(MqttConst.CACHE_MQTT_CLIENT_USER + clientId, token,
+            TimeSpan.FromMinutes(1));//将该客户端ID对应的token插入redis后面可以根据这个判断是哪个token登录的
+        return new MqttParameterOutput
+        {
+            ClientId = clientId,
+            Password = password,
+            Url = url,
+            UserName = userName,
+            Topics = new List<string> { MqttConst.MQTT_TOPIC_PREFIX + user.Id }//默认监听自己
+        };
+    }
+
+    /// <inheritdoc/>
+    public async Task<MqttAuthOutput> Auth(MqttAuthInput input)
+    {
+        var user = await _sysUserService.GetUserByAccount(input.Username);
+        var mqttAuthOutput = new MqttAuthOutput { Is_superuser = false, Result = "deny" };
+        //获取用户token
+        var tokens = _simpleCacheService.HashGetOne<List<TokenInfo>>(CacheConst.CACHE_USER_TOKEN, user.Id.ToString());
+        if (tokens != null)
+        {
+            if (tokens.Any(it => it.Token == input.Password))//判断是否有token
+                mqttAuthOutput.Result = "allow";//允许登录
+        }
+        return mqttAuthOutput;
+    }
+}