83 lines
2.5 KiB
Go
83 lines
2.5 KiB
Go
package handler
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/go-orz/orz"
|
|
"github.com/gorilla/websocket"
|
|
"github.com/labstack/echo/v5"
|
|
"github.com/pika-monitor/pika/internal/service"
|
|
ws "github.com/pika-monitor/pika/internal/websocket"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// Enable 启用探针数据处理。
|
|
func (h *AgentHandler) Enable(c *echo.Context) error {
|
|
return h.updateEnabled(c, true)
|
|
}
|
|
|
|
// Disable 禁用探针数据处理和告警。
|
|
func (h *AgentHandler) Disable(c *echo.Context) error {
|
|
return h.updateEnabled(c, false)
|
|
}
|
|
|
|
func (h *AgentHandler) updateEnabled(c *echo.Context, enabled bool) error {
|
|
h.enabledMu.Lock()
|
|
defer h.enabledMu.Unlock()
|
|
|
|
agentID := c.Param("id")
|
|
if err := h.agentService.UpdateAgentEnabled(c.Request().Context(), agentID, enabled); err != nil {
|
|
return err
|
|
}
|
|
return orz.Ok(c, orz.Map{})
|
|
}
|
|
|
|
type AgentHandler struct {
|
|
enabledMu sync.RWMutex
|
|
logger *zap.Logger
|
|
agentService *service.AgentService
|
|
trafficService *service.TrafficService
|
|
metricService *service.MetricService
|
|
monitorSvc *service.MonitorService
|
|
tamperService *service.TamperService
|
|
ddnsService *service.DDNSService
|
|
sshLoginService *service.SSHLoginService
|
|
apiKeyService *service.ApiKeyService
|
|
propertyService *service.PropertyService
|
|
wsManager *ws.Manager
|
|
upgrader websocket.Upgrader
|
|
}
|
|
|
|
func NewAgentHandler(logger *zap.Logger, agentService *service.AgentService, trafficService *service.TrafficService,
|
|
metricService *service.MetricService, monitorService *service.MonitorService, tamperService *service.TamperService,
|
|
ddnsService *service.DDNSService, sshLoginService *service.SSHLoginService, apiKeyService *service.ApiKeyService,
|
|
propertyService *service.PropertyService, wsManager *ws.Manager) *AgentHandler {
|
|
|
|
h := &AgentHandler{
|
|
logger: logger,
|
|
agentService: agentService,
|
|
trafficService: trafficService,
|
|
metricService: metricService,
|
|
monitorSvc: monitorService,
|
|
tamperService: tamperService,
|
|
ddnsService: ddnsService,
|
|
sshLoginService: sshLoginService,
|
|
apiKeyService: apiKeyService,
|
|
propertyService: propertyService,
|
|
wsManager: wsManager,
|
|
}
|
|
|
|
// 初始化upgrader，需要在创建handler之后因为需要引用h.checkOrigin
|
|
h.upgrader = websocket.Upgrader{
|
|
ReadBufferSize: 1024 * 32,
|
|
WriteBufferSize: 1024 * 32,
|
|
EnableCompression: true,
|
|
}
|
|
|
|
// 设置WebSocket消息处理器
|
|
wsManager.SetMessageHandler(h.handleWebSocketMessage)
|
|
wsManager.SetPongHandler(h.handleWebSocketPong)
|
|
|
|
return h
|
|
}
|