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{} }, // 添加其他模型 }