zhaoqingang
2025-02-10 bc37c44c45d08c2cb42822e08adf429f4bcffb51
models/model.go
@@ -9,12 +9,15 @@
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{} },
   "location":  func() Model { return &LocationModel{} },
   "night":     func() Model { return &nightModel{} },
   // 添加其他模型
}