fyah/AI-tianyan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: implement token-based auto stream pulling

Browse Source
This commit is contained in:
fyah
2026-05-09 11:03:27 +08:00
parent 966ef512ab
commit 6ff71fded0
5 changed files with 270 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
cmd/edge-agent/main.go
View File

@@ -21,8 +21,8 @@ func main() {
flag.Parse()
cfg := config.Load(*cfgPath)
log.Printf("edge-agent start id=%s uuid=%s cloud=%s mqtt=%s",
cfg.EdgeID, cfg.GetDeviceIdentity(), cfg.CloudURL, cfg.MqttBroker)
log.Printf("edge-agent start id=%s uuid=%s cloud=%s mqtt=%s auto_pull=%v",
cfg.EdgeID, cfg.GetDeviceIdentity(), cfg.CloudURL, cfg.MqttBroker, cfg.AutoPull)
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
defer cancel()
@@ -43,8 +43,17 @@ func main() {
mqttMgr := control.NewMqttManager(cfg)
run(mqttMgr.Run)
// 1.5 Stream Puller (Optional: Dynamic URL fetching)
var puller *stream.Puller
if cfg.AutoPull {
puller = stream.NewPuller(cfg)
puller.StartHeartbeatLoop()
defer puller.Stop()
log.Println("puller: auto-pull mode enabled")
}
// 2. Stream Ingestion (Dynamic)
run(stream.NewStreamManager(cfg, frames, mqttMgr.GetUpdates()).Run)
run(stream.NewStreamManager(cfg, puller, frames, mqttMgr.GetUpdates()).Run)
// 3. Inference (Dynamic Workers)
run(infer.NewClient(cfg, frames, events, mqttMgr.GetUpdates()).Run)