zhangzengfei
2025-02-11 e4fc307d58856a0b319c41f0930cd3b731cb0b24
models/model.go
@@ -9,12 +9,16 @@
type Model interface {
   Init(task *db.ModelTask) error
   Run() error
   KeepAlive() error
   Shutdown() error
}
var modelRegistry = map[string]func() Model{
   "gather":    func() Model { return &GatherModel{} },
   "disappear": func() Model { return &DisappearModel{} },
   "gather":           func() Model { return &GatherModel{} },
   "disappear":        func() Model { return &DisappearModel{} },
   "location":         func() Model { return &LocationModel{} },
   "night":            func() Model { return &nightModel{} },
   "accessRegularity": func() Model { return &RegularityModel{} },
   // 添加其他模型
}