Go边缘代理: 单文件可执行+嵌入式MediaMTX+监控+注册
- 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
This commit is contained in:
admin committed
1 parent
980b91f41f
commit
409e5d314e
14 files changed
+800
-11
No files matched your search
@@ -0,0 +1,126 @@
|
||||
package agent
|
||||
|
||||
import (
|
||||
"embed"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
NodeName string
|
||||
OriginURL string
|
||||
HlsPort int
|
||||
ApiPort int
|
||||
Subnet string
|
||||
DataDir string
|
||||
Weight int
|
||||
}
|
||||
|
||||
type Agent struct {
|
||||
cfg Config
|
||||
mediamtxFS embed.FS
|
||||
mtx *MediaMTXManager
|
||||
hb *Heartbeat
|
||||
monitor *Monitor
|
||||
nodeID string
|
||||
stopCh chan struct{}
|
||||
}
|
||||
|
||||
func New(cfg Config, mtxFS embed.FS) (*Agent, error) {
|
||||
return &Agent{
|
||||
cfg: cfg,
|
||||
mediamtxFS: mtxFS,
|
||||
nodeID: generateNodeID(cfg),
|
||||
stopCh: make(chan struct{}),
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (a *Agent) Start() error {
|
||||
if err := os.MkdirAll(a.cfg.DataDir, 0755); err != nil {
|
||||
return fmt.Errorf("create data dir: %w", err)
|
||||
}
|
||||
|
||||
a.mtx = NewMediaMTXManager(a.cfg, a.mediamtxFS)
|
||||
if err := a.mtx.Extract(); err != nil {
|
||||
return fmt.Errorf("extract mediamtx: %w", err)
|
||||
}
|
||||
|
||||
if err := a.mtx.WriteConfig(); err != nil {
|
||||
return fmt.Errorf("write config: %w", err)
|
||||
}
|
||||
|
||||
if err := a.mtx.Start(); err != nil {
|
||||
return fmt.Errorf("start mediamtx: %w", err)
|
||||
}
|
||||
|
||||
a.monitor = NewMonitor()
|
||||
a.hb = NewHeartbeat(a.cfg, a.nodeID)
|
||||
|
||||
time.Sleep(2 * time.Second)
|
||||
|
||||
if err := a.hb.Register(); err != nil {
|
||||
log.Printf("Warning: registration failed: %v (will retry)", err)
|
||||
}
|
||||
|
||||
go a.heartbeatLoop()
|
||||
go a.monitorLoop()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (a *Agent) Stop() {
|
||||
log.Println("Stopping agent...")
|
||||
close(a.stopCh)
|
||||
if a.mtx != nil {
|
||||
a.mtx.Stop()
|
||||
}
|
||||
log.Println("Agent stopped")
|
||||
}
|
||||
|
||||
func (a *Agent) heartbeatLoop() {
|
||||
ticker := time.NewTicker(10 * time.Second)
|
||||
defer ticker.Stop()
|
||||
for {
|
||||
select {
|
||||
case <-a.stopCh:
|
||||
return
|
||||
case <-ticker.C:
|
||||
metrics := a.monitor.Snapshot()
|
||||
if err := a.hb.Send(metrics); err != nil {
|
||||
log.Printf("Heartbeat error: %v", err)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (a *Agent) monitorLoop() {
|
||||
ticker := time.NewTicker(5 * time.Second)
|
||||
defer ticker.Stop()
|
||||
for {
|
||||
select {
|
||||
case <-a.stopCh:
|
||||
return
|
||||
case <-ticker.C:
|
||||
a.monitor.Tick()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (a *Agent) NodeID() string { return a.nodeID }
|
||||
func (a *Agent) Config() Config { return a.cfg }
|
||||
func (a *Agent) Metrics() Metrics {
|
||||
if a.monitor != nil {
|
||||
return a.monitor.Snapshot()
|
||||
}
|
||||
return Metrics{}
|
||||
}
|
||||
|
||||
func generateNodeID(cfg Config) string {
|
||||
host, _ := os.Hostname()
|
||||
if host == "" {
|
||||
host = "unknown"
|
||||
}
|
||||
return fmt.Sprintf("%s-%s", host, cfg.NodeName)
|
||||
}
|
||||
@@ -0,0 +1,84 @@
|
||||
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
|
||||
}
|
||||
@@ -0,0 +1,132 @@
|
||||
package agent
|
||||
|
||||
import (
|
||||
"embed"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type MediaMTXManager struct {
|
||||
cfg Config
|
||||
fs embed.FS
|
||||
binPath string
|
||||
configPath string
|
||||
cmd *exec.Cmd
|
||||
}
|
||||
|
||||
func NewMediaMTXManager(cfg Config, fs embed.FS) *MediaMTXManager {
|
||||
return &MediaMTXManager{
|
||||
cfg: cfg,
|
||||
fs: fs,
|
||||
binPath: filepath.Join(cfg.DataDir, "mediamtx"),
|
||||
configPath: filepath.Join(cfg.DataDir, "mediamtx.yml"),
|
||||
}
|
||||
}
|
||||
|
||||
func (m *MediaMTXManager) Extract() error {
|
||||
if _, err := os.Stat(m.binPath); err == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
srcPath := "mediamtx/"
|
||||
if runtime.GOOS == "windows" {
|
||||
srcPath += "mediamtx.exe"
|
||||
} else {
|
||||
srcPath += "mediamtx"
|
||||
}
|
||||
|
||||
entries, err := m.fs.ReadDir("mediamtx")
|
||||
if err != nil {
|
||||
return fmt.Errorf("read embedded dir: %w", err)
|
||||
}
|
||||
|
||||
for _, entry := range entries {
|
||||
if entry.IsDir() {
|
||||
continue
|
||||
}
|
||||
name := entry.Name()
|
||||
if runtime.GOOS == "windows" && !strings.Contains(name, ".exe") {
|
||||
continue
|
||||
}
|
||||
if runtime.GOOS != "windows" && strings.Contains(name, ".exe") {
|
||||
continue
|
||||
}
|
||||
|
||||
src, err := m.fs.Open("mediamtx/" + name)
|
||||
if err != nil {
|
||||
return fmt.Errorf("open embedded %s: %w", name, err)
|
||||
}
|
||||
defer src.Close()
|
||||
|
||||
destName := "mediamtx"
|
||||
if runtime.GOOS == "windows" {
|
||||
destName = "mediamtx.exe"
|
||||
}
|
||||
destPath := filepath.Join(m.cfg.DataDir, destName)
|
||||
dest, err := os.OpenFile(destPath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0755)
|
||||
if err != nil {
|
||||
src.Close()
|
||||
return fmt.Errorf("create %s: %w", destPath, err)
|
||||
}
|
||||
|
||||
if _, err := io.Copy(dest, src); err != nil {
|
||||
dest.Close()
|
||||
src.Close()
|
||||
return fmt.Errorf("copy: %w", err)
|
||||
}
|
||||
dest.Close()
|
||||
src.Close()
|
||||
m.binPath = destPath
|
||||
break
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *MediaMTXManager) WriteConfig() error {
|
||||
config := fmt.Sprintf(`logLevel: info
|
||||
logDestinations: [stdout]
|
||||
|
||||
rtmp: no
|
||||
rtsp: no
|
||||
|
||||
hls: yes
|
||||
hlsAddress: :%d
|
||||
|
||||
api: yes
|
||||
apiAddress: :%d
|
||||
|
||||
paths:
|
||||
all:
|
||||
source: rtmp://%s:1935/live/$MTX_PATH
|
||||
sourceOnDemand: yes
|
||||
`,
|
||||
m.cfg.HlsPort,
|
||||
m.cfg.ApiPort,
|
||||
strings.TrimPrefix(strings.TrimPrefix(m.cfg.OriginURL, "http://"), "https://"),
|
||||
)
|
||||
|
||||
return os.WriteFile(m.configPath, []byte(config), 0644)
|
||||
}
|
||||
|
||||
func (m *MediaMTXManager) Start() error {
|
||||
m.cmd = exec.Command(m.binPath, m.configPath)
|
||||
m.cmd.Stdout = os.Stdout
|
||||
m.cmd.Stderr = os.Stderr
|
||||
if err := m.cmd.Start(); err != nil {
|
||||
return fmt.Errorf("start mediamtx: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *MediaMTXManager) Stop() {
|
||||
if m.cmd != nil && m.cmd.Process != nil {
|
||||
m.cmd.Process.Kill()
|
||||
m.cmd.Wait()
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,83 @@
|
||||
package agent
|
||||
|
||||
import (
|
||||
"log"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/shirou/gopsutil/v3/cpu"
|
||||
"github.com/shirou/gopsutil/v3/mem"
|
||||
"github.com/shirou/gopsutil/v3/net"
|
||||
)
|
||||
|
||||
type Metrics struct {
|
||||
CPU float64
|
||||
Memory float64
|
||||
Bandwidth float64
|
||||
ActiveConns int
|
||||
HlsReaders int
|
||||
}
|
||||
|
||||
type Monitor struct {
|
||||
mu sync.RWMutex
|
||||
current Metrics
|
||||
prevNetIn uint64
|
||||
prevNetOut uint64
|
||||
prevNetTime time.Time
|
||||
}
|
||||
|
||||
func NewMonitor() *Monitor {
|
||||
return &Monitor{}
|
||||
}
|
||||
|
||||
func (m *Monitor) Tick() {
|
||||
metrics := Metrics{}
|
||||
|
||||
if pct, err := cpu.Percent(0, false); err == nil && len(pct) > 0 {
|
||||
metrics.CPU = pct[0]
|
||||
}
|
||||
|
||||
if v, err := mem.VirtualMemory(); err == nil {
|
||||
metrics.Memory = v.UsedPercent
|
||||
}
|
||||
|
||||
ioCounters, err := net.IOCounters(false)
|
||||
if err == nil && len(ioCounters) > 0 {
|
||||
now := time.Now()
|
||||
curIn := ioCounters[0].BytesRecv
|
||||
curOut := ioCounters[0].BytesSent
|
||||
|
||||
if m.prevNetTime.IsZero() {
|
||||
m.prevNetIn = curIn
|
||||
m.prevNetOut = curOut
|
||||
m.prevNetTime = now
|
||||
} else {
|
||||
elapsed := now.Sub(m.prevNetTime).Seconds()
|
||||
if elapsed > 0 {
|
||||
bytesTotal := (curIn - m.prevNetIn) + (curOut - m.prevNetOut)
|
||||
metrics.Bandwidth = float64(bytesTotal) * 8 / elapsed / 1e6
|
||||
}
|
||||
m.prevNetIn = curIn
|
||||
m.prevNetOut = curOut
|
||||
m.prevNetTime = now
|
||||
}
|
||||
}
|
||||
|
||||
conns, _ := net.Connections("tcp")
|
||||
metrics.ActiveConns = len(conns)
|
||||
|
||||
metrics.HlsReaders = 0
|
||||
|
||||
m.mu.Lock()
|
||||
m.current = metrics
|
||||
m.mu.Unlock()
|
||||
|
||||
log.Printf("Metrics: CPU=%.1f%% MEM=%.1f%% BW=%.1fMbps CONN=%d",
|
||||
metrics.CPU, metrics.Memory, metrics.Bandwidth, metrics.ActiveConns)
|
||||
}
|
||||
|
||||
func (m *Monitor) Snapshot() Metrics {
|
||||
m.mu.RLock()
|
||||
defer m.mu.RUnlock()
|
||||
return m.current
|
||||
}
|
||||
Reference in new issue
Block a user