Day 30 Pt.2 | 探针 Agent 开发

mkdir -p ~/final-project/agent && cd ~/final-project/agent

# 1. 初始化模块
go mod init github.com/akria/monitor-agent

# 2. 获取系统信息采集库
go get github.com/shirou/gopsutil/v3

# 3. 获取 HTTP 客户端库
go get github.com/go-resty/resty/v2

package main

import (
    "fmt"
    "os"
    "time"

    "github.com/go-resty/resty/v2"
    "github.com/shirou/gopsutil/v3/cpu"
    "github.com/shirou/gopsutil/v3/disk"
    "github.com/shirou/gopsutil/v3/mem"
)

// Data Model: 必须与 Python Master 的 Pydantic Schema 一致
// json tag 指定了序列化后的字段名
type MetricPayload struct {
    Hostname string  `json:"hostname"`
    IP       string  `json:"ip"`
    CPU      float64 `json:"cpu"`
    Mem      float64 `json:"mem"`
    Disk     float64 `json:"disk"`
}

// 配置常量
const MasterURL = "http://127.0.0.1:8000/api/v1/report"

// CollectAndReport 执行一次完整的 "采集 -> 上报" 流程
func CollectAndReport(client *resty.Client) {
    // --- 1. 采集阶段 ---
    host, _ := os.Hostname()
    
    // CPU: 采样 1秒。注意这会阻塞当前协程 1秒
    // 返回的是一个 slice，因为可能是多核，我们取第一个(总平均)
    cpuPercent, _ := cpu.Percent(1*time.Second, false)
    
    // Mem & Disk
    memStats, _ := mem.VirtualMemory()
    diskStats, _ := disk.Usage("/") // 采集根分区

    // --- 2. 组装阶段 ---
    payload := MetricPayload{
        Hostname: host,
        IP:       "192.168.1.10", // 暂写死，拓展任务里教怎么获取
        CPU:      cpuPercent[0],
        Mem:      memStats.UsedPercent,
        Disk:     diskStats.UsedPercent,
    }

    // --- 3. 上报阶段 ---
    fmt.Printf("📤 Reporting: %+v ... ", payload)
    
    // Chain API 调用风格
    resp, err := client.R().
        SetBody(payload).
        Post(MasterURL)

    if err != nil {
        fmt.Printf("❌ Failed: %v\n", err)
        return
    }
    
    // 检查业务状态码
    if resp.StatusCode() == 201 {
        fmt.Println("✅ Sent.")
    } else {
        fmt.Printf("⚠️ Server Error: %s\n", resp.Status())
    }
}

func main() {
    fmt.Println("🚀 Agent version 1.0.0 is running...")
    
    // 1. 初始化 HTTP Client (复用连接池)
    client := resty.New()
    client.SetTimeout(2 * time.Second) // 务必设置超时，防止卡死
    client.SetRetryCount(3)            // 自动重试 3 次

    // 2. 立即执行一次 (启动时自检)
    CollectAndReport(client)

    // 3. 启动定时器 (Ticker)
    // 类似于 Python 的 while True + sleep，但更精准
    ticker := time.NewTicker(5 * time.Second)
    
    // 4. 阻塞主线程，不断响应 Ticker 事件
    for range ticker.C {
        CollectAndReport(client)
    }
}

# 编译成 Linux 64位 可执行文件
# CGO_ENABLED=0: 禁用 CGO，确保生成纯静态二进制，不依赖系统 libc
CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o monitor_agent_linux main.go

# 检查产物
ls -lh monitor_agent_linux
# 你会发现它只有 7-10MB 左右