feat: mqtt v3

2026-07-08 15:31:58 +08:00
parent df27dfdb03
commit d0d22facf6
6 changed files with 3231 additions and 0 deletions
@@ -96,6 +96,66 @@ func (sc *SystemDebugController) StopMqtt(c *gin.Context) {
 	writeSuccess(c, http.StatusOK, "stop success", service.Status())
 }

+// @Summary 获取MQTT V3调试状态
+// @Description 获取MQTT V3调试服务的当前运行状态
+// @Tags 系统调试
+// @Produce json
+// @Security BearerAuth
+// @Success 200 {object} SwagAPIResponse "查询成功"
+// @Router /admin/system-debug/mqtt-v3/status [get]
+func (sc *SystemDebugController) MqttV3Status(c *gin.Context) {
+	service := mqtt.GetDebugV3Service()
+	if service == nil {
+		writeError(c, http.StatusServiceUnavailable, "mqtt v3 debug service unavailable")
+		return
+	}
+	writeSuccess(c, http.StatusOK, "query success", service.Status())
+}
+
+// @Summary 启动MQTT V3调试
+// @Description 启动MQTT V3调试服务
+// @Tags 系统调试
+// @Accept json
+// @Produce json
+// @Param persist body mqttDebugStartRequest false "是否持久化到数据库"
+// @Security BearerAuth
+// @Success 200 {object} SwagAPIResponse "启动成功"
+// @Router /admin/system-debug/mqtt-v3/start [post]
+func (sc *SystemDebugController) StartMqttV3(c *gin.Context) {
+	service := mqtt.GetDebugV3Service()
+	if service == nil {
+		writeError(c, http.StatusServiceUnavailable, "mqtt v3 debug service unavailable")
+		return
+	}
+	var payload mqttDebugStartRequest
+	if err := c.ShouldBindJSON(&payload); err != nil && !errors.Is(err, http.ErrBodyNotAllowed) {
+		writeError(c, http.StatusBadRequest, err.Error())
+		return
+	}
+	if err := service.Start(payload.PersistToDatabase); err != nil {
+		writeError(c, http.StatusInternalServerError, err.Error())
+		return
+	}
+	writeSuccess(c, http.StatusOK, "start success", service.Status())
+}
+
+// @Summary 停止MQTT V3调试
+// @Description 停止MQTT V3调试服务
+// @Tags 系统调试
+// @Produce json
+// @Security BearerAuth
+// @Success 200 {object} SwagAPIResponse "停止成功"
+// @Router /admin/system-debug/mqtt-v3/stop [post]
+func (sc *SystemDebugController) StopMqttV3(c *gin.Context) {
+	service := mqtt.GetDebugV3Service()
+	if service == nil {
+		writeError(c, http.StatusServiceUnavailable, "mqtt v3 debug service unavailable")
+		return
+	}
+	service.Stop()
+	writeSuccess(c, http.StatusOK, "stop success", service.Status())
+}
+
 // @Summary 获取MQTT重放状态
 // @Description 获取心率历史数据 MQTT 重放状态
 // @Tags 系统调试
@@ -215,3 +275,59 @@ func (sc *SystemDebugController) MqttWebSocket(c *gin.Context) {
 		}
 	}
 }
+
+// @Summary MQTT V3 WebSocket连接
+// @Description 通过WebSocket实时监听MQTT V3消息（需要SuperAdmin权限）
+// @Tags 系统调试
+// @Param token query string true "JWT Token"
+// @Success 101 "切换为WebSocket协议"
+// @Router /admin/system-debug/mqtt-v3/ws [get]
+func (sc *SystemDebugController) MqttV3WebSocket(c *gin.Context) {
+	service := mqtt.GetDebugV3Service()
+	if service == nil {
+		c.JSON(http.StatusServiceUnavailable, gin.H{"error": "mqtt v3 debug service unavailable"})
+		return
+	}
+
+	token := strings.TrimSpace(c.Query("token"))
+	if token == "" {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "missing token"})
+		return
+	}
+	claims, err := util.ParseToken(token)
+	if err != nil {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "invalid token"})
+		return
+	}
+
+	var user models.User
+	if err := config.DB.First(&user, claims.UserID).Error; err != nil {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "user not found"})
+		return
+	}
+	if !user.IsActive {
+		c.JSON(http.StatusForbidden, gin.H{"error": "user is disabled"})
+		return
+	}
+	if util.IsTokenRevoked(&user, claims) {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "token has been revoked"})
+		return
+	}
+	if user.Role != models.UserRoleSuperAdmin {
+		c.JSON(http.StatusForbidden, gin.H{"error": "super admin required"})
+		return
+	}
+
+	conn, err := debugUpgrader.Upgrade(c.Writer, c.Request, nil)
+	if err != nil {
+		return
+	}
+	service.AddSubscriber(conn)
+	defer service.RemoveSubscriber(conn)
+
+	for {
+		if _, _, err := conn.ReadMessage(); err != nil {
+			break
+		}
+	}
+}