Files
geo/server/internal/tenant/transport/desktop_monitoring_handler.go
T
root 4142c53fa6 feat(monitoring): dispatch monitoring tasks to desktop via AMQP outbox
Replace SSE /desktop/events with priority AMQP dispatch for monitoring
runs, and add phase1/phase2 infrastructure so desktop workers can lease,
resume, report, skip, and cancel monitoring tasks over the existing
dispatch WebSocket.

- Add monitoring collect outbox worker and phase2 desktop task fields
- Add /desktop/monitoring/tasks/{lease,resume,result,skip,cancel} routes
- Introduce execution-devtools and network-observer on desktop runtime
- Refactor runtime-controller, LoginView, and doubao adapter for the new flow
- Add channelName column to tracking views

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-04-22 00:24:21 +08:00

117 lines
3.5 KiB
Go

package transport
import (
"errors"
"io"
"strconv"
"github.com/gin-gonic/gin"
"github.com/geo-platform/tenant-api/internal/bootstrap"
"github.com/geo-platform/tenant-api/internal/shared/response"
"github.com/geo-platform/tenant-api/internal/tenant/app"
)
type DesktopMonitoringHandler struct {
svc *app.MonitoringCallbackService
}
func NewDesktopMonitoringHandler(a *bootstrap.App) *DesktopMonitoringHandler {
return &DesktopMonitoringHandler{
svc: app.NewMonitoringCallbackService(a.DB, a.MonitoringDB, a.RabbitMQ, a.LLM, a.Logger),
}
}
func (h *DesktopMonitoringHandler) Lease(c *gin.Context) {
var req app.MonitoringLeaseTasksRequest
if err := c.ShouldBindJSON(&req); err != nil && !errors.Is(err, io.EOF) {
response.Error(c, response.ErrBadRequest(40041, "invalid_monitoring_lease_request", err.Error()))
return
}
data, err := h.svc.LeaseTasksForDesktopClient(c.Request.Context(), MustDesktopClient(c.Request.Context()), req)
if err != nil {
response.Error(c, err)
return
}
response.Success(c, data)
}
func (h *DesktopMonitoringHandler) Resume(c *gin.Context) {
var req app.MonitoringResumeTasksRequest
if err := c.ShouldBindJSON(&req); err != nil && !errors.Is(err, io.EOF) {
response.Error(c, response.ErrBadRequest(40041, "invalid_monitoring_resume_request", err.Error()))
return
}
data, err := h.svc.ResumeTasksForDesktopClient(c.Request.Context(), MustDesktopClient(c.Request.Context()), req)
if err != nil {
response.Error(c, err)
return
}
response.Success(c, data)
}
func (h *DesktopMonitoringHandler) Result(c *gin.Context) {
taskID, err := strconv.ParseInt(c.Param("id"), 10, 64)
if err != nil {
response.Error(c, response.ErrBadRequest(40041, "invalid_monitoring_task_id", "monitoring task id must be an integer"))
return
}
var req app.MonitoringTaskResultRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.Error(c, response.ErrBadRequest(40041, "invalid_monitoring_result_request", err.Error()))
return
}
data, svcErr := h.svc.HandleTaskResultForDesktopClient(c.Request.Context(), MustDesktopClient(c.Request.Context()), taskID, req)
if svcErr != nil {
response.Error(c, svcErr)
return
}
response.Success(c, data)
}
func (h *DesktopMonitoringHandler) Skip(c *gin.Context) {
taskID, err := strconv.ParseInt(c.Param("id"), 10, 64)
if err != nil {
response.Error(c, response.ErrBadRequest(40041, "invalid_monitoring_task_id", "monitoring task id must be an integer"))
return
}
var req app.MonitoringSkipTaskRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.Error(c, response.ErrBadRequest(40041, "invalid_monitoring_skip_request", err.Error()))
return
}
data, svcErr := h.svc.SkipTaskForDesktopClient(c.Request.Context(), MustDesktopClient(c.Request.Context()), taskID, req)
if svcErr != nil {
response.Error(c, svcErr)
return
}
response.Success(c, data)
}
func (h *DesktopMonitoringHandler) Cancel(c *gin.Context) {
taskID, err := strconv.ParseInt(c.Param("id"), 10, 64)
if err != nil {
response.Error(c, response.ErrBadRequest(40041, "invalid_monitoring_task_id", "monitoring task id must be an integer"))
return
}
var req app.MonitoringCancelTaskRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.Error(c, response.ErrBadRequest(40041, "invalid_monitoring_cancel_request", err.Error()))
return
}
data, svcErr := h.svc.CancelTaskForDesktopClient(c.Request.Context(), MustDesktopClient(c.Request.Context()), taskID, req)
if svcErr != nil {
response.Error(c, svcErr)
return
}
response.Success(c, data)
}