Files
pika/internal/handler/agent_query_handler.go
T
dushixiangandClaude Opus 4.7 c431f300bc 新增实时监控模式,agent 1s 并行采集 + 前端滑动窗口
agent
- 移除 collector.interval 配置项;按指标类型内置节奏:cpu/memory/network/
  disk_io/gpu/network_connection 1s,temperature 5s,disk 30s,host 60s
- doCollectAndSend 改为并行执行本 tick 到期的采集器,避免串行累计耗时
  超过 1s 节奏;每个 goroutine 加 recover 兜底,单点 panic 不拖垮整 tick

server
- LatestMetrics 加 sync.RWMutex + Update/Snapshot 方法,HandleMetricData
  所有写入走 Update;GetLatestMetrics 返回 Snapshot,handler 在副本上
  sanitize,修复 marshal 时与 ws 写入的 data race(go test -race 验证)
- HandleMetricData defer Set 缓存项,每次上报刷新 TTL;host case 也走
  Update,不再 1 小时后失效
- LatestMetrics.Timestamp 用服务端 time.Now(),agent 时钟倒拨不会冻结
  前端实时图;Timestamp/DiskIOSummary 字段补到 LatestMetrics
- AutoStep 新增 r ≤ 5m → 1s 档;timeRangeMilliseconds 加 2m

frontend
- 详情页加"实时"档位(默认),1s 拉 latestMetrics + 1m 历史播种 + 滑动
  2 分钟窗口;快图表(CPU/Memory/Network/DiskIO/NetworkConnection)走
  useLiveBuffer 追加,慢图表(GPU/Temperature/Monitor)保留 refetch
- useLiveBuffer:seed 不再覆盖已追加的 livePoint;resetKey 处理切网卡 /
  切 agent 时的语义复位
- formatChartTime live 档位精确到秒;修复 1d/24h 仅显示 HH:mm 跨夜歧义
- 慢图表用 React.memo,阻断 1Hz 节奏的级联重渲染
- GpuChart 修复 utilization/temperature 时间戳错位时温度点丢失

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-05-06 23:18:50 +08:00

108 lines
2.6 KiB
Go

package handler
import (
"github.com/dushixiang/pika/internal/models"
"github.com/dushixiang/pika/internal/utils"
"github.com/go-orz/orz"
"github.com/labstack/echo/v4"
"gorm.io/datatypes"
)
// Get 获取探针详情(公开接口,已登录返回全部,未登录返回公开可见)
func (h *AgentHandler) Get(c echo.Context) error {
id := c.Param("id")
ctx := c.Request().Context()
// 根据认证状态返回相应的探针
isAuthenticated := utils.IsAuthenticated(c)
agent, err := h.agentService.GetAgentByAuth(ctx, id, isAuthenticated)
if err != nil {
return err
}
// 隐藏敏感配置
agent.SSHLoginConfig = datatypes.JSONType[models.SSHLoginConfigData]{}
agent.TamperProtectConfig = datatypes.JSONType[models.TamperProtectConfigData]{}
// 未登录时隐藏敏感信息
if !isAuthenticated {
agent.IP = ""
agent.IPv4 = ""
agent.IPv6 = ""
agent.Hostname = ""
}
return orz.Ok(c, agent)
}
// GetAgents 获取探针列表(公开接口,已登录返回全部,未登录返回公开可见)
func (h *AgentHandler) GetAgents(c echo.Context) error {
ctx := c.Request().Context()
// 根据认证状态返回相应的探针列表
isAuthenticated := utils.IsAuthenticated(c)
agents, err := h.agentService.ListByAuth(ctx, isAuthenticated)
if err != nil {
return err
}
// 排序
SortAgents(agents)
result := make([]map[string]interface{}, 0, len(agents))
for _, agent := range agents {
result = append(result, h.buildAgentListItem(agent, isAuthenticated))
}
return orz.Ok(c, result)
}
func (h *AgentHandler) buildAgentListItem(agent models.Agent, isAuthenticated bool) map[string]interface{} {
item := map[string]any{
"id": agent.ID,
"name": agent.Name,
"os": agent.OS,
"arch": agent.Arch,
"version": agent.Version,
"tags": agent.Tags,
"expireTime": agent.ExpireTime,
"status": agent.Status,
"lastSeenAt": agent.LastSeenAt,
"visibility": agent.Visibility,
"weight": agent.Weight,
}
trafficStats := agent.TrafficStats.Data()
if trafficStats.Enabled {
item["trafficStats"] = trafficStats
} else {
item["trafficStats"] = map[string]any{
"enabled": false,
}
}
metrics, ok := h.metricService.GetLatestMetrics(agent.ID)
if ok && metrics != nil {
if !isAuthenticated {
metrics.NetworkInterfaces = nil
}
item["metrics"] = metrics
}
return item
}
// GetTags 获取所有探针的标签
func (h *AgentHandler) GetTags(c echo.Context) error {
ctx := c.Request().Context()
tags, err := h.agentService.GetAllTags(ctx)
if err != nil {
return err
}
return orz.Ok(c, orz.Map{
"tags": tags,
})
}