admin
409e5d314e
- edge-agent/: Go项目, 支持Windows托盘/Linux无头两种模式 - 使用go:embed将MediaMTX二进制嵌入单一可执行文件 - 首次运行自动释放MediaMTX并生成边缘节点配置 - 自动注册到origin (POST /api/nodes/register) - 10秒心跳上报CPU/RAM/带宽/连接数 (gopsutil采集) - Windows: getlantern/systray托盘菜单(状态/CPU/MEM/带宽/退出) - Linux: headless模式,适合systemd守护 - build.sh/build.ps1: 交叉编译脚本(Linux/Windows) - 更新start.sh/品牌名StreamCore
85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
package agent
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
type Heartbeat struct {
|
|
cfg Config
|
|
nodeID string
|
|
client *http.Client
|
|
}
|
|
|
|
type HeartbeatPayload struct {
|
|
NodeID string `json:"node_id"`
|
|
CPUPct float64 `json:"cpu_percent"`
|
|
MemPct float64 `json:"memory_percent"`
|
|
BandwidthMbps float64 `json:"bandwidth_mbps"`
|
|
ActiveConnections int `json:"active_connections"`
|
|
HlsReaders int `json:"hls_readers"`
|
|
}
|
|
|
|
type RegisterPayload struct {
|
|
NodeID string `json:"node_id"`
|
|
Name string `json:"name"`
|
|
Host string `json:"host"`
|
|
HlsPort int `json:"hls_port"`
|
|
ApiPort int `json:"api_port"`
|
|
Weight int `json:"weight"`
|
|
Subnet string `json:"subnet"`
|
|
}
|
|
|
|
func NewHeartbeat(cfg Config, nodeID string) *Heartbeat {
|
|
return &Heartbeat{
|
|
cfg: cfg,
|
|
nodeID: nodeID,
|
|
client: &http.Client{Timeout: 10 * time.Second},
|
|
}
|
|
}
|
|
|
|
func (h *Heartbeat) Register() error {
|
|
payload := RegisterPayload{
|
|
NodeID: h.nodeID,
|
|
Name: h.cfg.NodeName,
|
|
Host: h.cfg.OriginURL,
|
|
HlsPort: h.cfg.HlsPort,
|
|
ApiPort: h.cfg.ApiPort,
|
|
Weight: h.cfg.Weight,
|
|
Subnet: h.cfg.Subnet,
|
|
}
|
|
return h.post("/api/nodes/register", payload)
|
|
}
|
|
|
|
func (h *Heartbeat) Send(m Metrics) error {
|
|
payload := HeartbeatPayload{
|
|
NodeID: h.nodeID,
|
|
CPUPct: m.CPU,
|
|
MemPct: m.Memory,
|
|
BandwidthMbps: m.Bandwidth,
|
|
ActiveConnections: m.ActiveConns,
|
|
HlsReaders: m.HlsReaders,
|
|
}
|
|
return h.post("/api/nodes/heartbeat", payload)
|
|
}
|
|
|
|
func (h *Heartbeat) post(path string, payload interface{}) error {
|
|
body, err := json.Marshal(payload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
url := h.cfg.OriginURL + path
|
|
resp, err := h.client.Post(url, "application/json", bytes.NewReader(body))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
if resp.StatusCode >= 400 {
|
|
return fmt.Errorf("status %d", resp.StatusCode)
|
|
}
|
|
return nil
|
|
}
|