Files
aiops/agent/task/client.go
T
attacker 8240a88a84 feat(k8s): M14 Namespace 管理看板(资源配额 + 命名空间 CRUD)
- 后端: ListNamespacesDetailed/CreateNamespace/DeleteNamespace/ListResourceQuotas/ListLimitRanges
- 后端: GetNamespacesDetail handler (6 API calls aggregate, not N×3)
- 后端: CreateNamespaceHandler + DeleteNamespaceHandler (system NS protection)
- 前端: NamespaceManagementPage.tsx (~280行) — 表格+概览卡片+创建/删除弹框
- 路由: GET/POST/DELETE /:clusterCode/namespaces + GET /namespace/:ns/detail
- 验证: 9/9 通过 (urllib), TS 零错误, Vite 构建成功
2026-06-24 23:46:52 +08:00

93 lines
2.9 KiB
Go

// Package task 的 client.go 实现与后端任务/探针接口的 HTTP 交互。
// 涵盖拉取待执行任务(poll)、回传执行结果(submit result)、拉取探针配置(poll probes)。
// 通过共享 client 处理鉴权 + 重试,本文件仅负责业务路径和数据模型。
//
// 署名 Logan.Li
package task
import (
"context"
"fmt"
"log"
"net/url"
"ezops/agent/client"
"ezops/agent/model"
)
const (
pollPath = "/api/v1/agent/tasks/poll"
resultPathFmt = "/api/v1/agent/tasks/%s/result"
probePath = "/api/v1/agent/health-probes/config"
portPolicyPath = "/api/v1/agent/port-policy"
)
// Client 负责与后端任务/探针接口通信。
type Client struct {
client *client.Client
hostname string
logger *log.Logger
}
// NewClient 创建任务客户端实例。
func NewClient(cli *client.Client, hostname string, logger *log.Logger) *Client {
return &Client{client: cli, hostname: hostname, logger: logger}
}
// Poll 拉取当前主机待执行的任务列表。
func (c *Client) Poll(ctx context.Context) ([]Task, error) {
path := pollPath + "?hostname=" + url.QueryEscape(c.hostname)
var wrapper struct {
Code int `json:"code"`
Message string `json:"message"`
Data []Task `json:"data"`
}
if err := c.client.DoJSON(ctx, "GET", path, nil, &wrapper); err != nil {
return nil, fmt.Errorf("任务轮询失败: %w", err)
}
return wrapper.Data, nil
}
// Submit 将任务执行结果回传至后端。
func (c *Client) Submit(ctx context.Context, res Result) error {
path := fmt.Sprintf(resultPathFmt, url.PathEscape(res.TaskID))
if err := c.client.DoJSON(ctx, "POST", path, res, nil); err != nil {
return fmt.Errorf("任务结果回传失败: %w", err)
}
return nil
}
// PollProbes 拉取本机健康探针配置(v2.1)。
func (c *Client) PollProbes(ctx context.Context) ([]model.ProbeConfig, error) {
path := probePath + "?hostname=" + url.QueryEscape(c.hostname)
var wrapper struct {
Code int `json:"code"`
Message string `json:"message"`
Data []model.ProbeConfig `json:"data"`
}
if err := c.client.DoJSON(ctx, "GET", path, nil, &wrapper); err != nil {
return nil, fmt.Errorf("探针配置拉取失败: %w", err)
}
if wrapper.Data == nil {
return []model.ProbeConfig{}, nil
}
return wrapper.Data, nil
}
// PollPortPolicy 拉取本机端口过滤策略(可选配置,拉取失败静默忽略) // Logan.Li
func (c *Client) PollPortPolicy(ctx context.Context) (*model.PortPolicy, error) {
path := portPolicyPath + "?hostname=" + url.QueryEscape(c.hostname)
var wrapper struct {
Code int `json:"code"`
Message string `json:"message"`
Data model.PortPolicy `json:"data"`
}
if err := c.client.DoJSON(ctx, "GET", path, nil, &wrapper); err != nil {
return nil, fmt.Errorf("端口策略拉取失败: %w", err)
}
return &wrapper.Data, nil
}