models/model.go
@@ -2,17 +2,21 @@ import ( "fmt" "model-engine/db" ) 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{} }, "gather": func() Model { return &GatherModel{} }, "disappear": func() Model { return &DisappearModel{} }, "location": func() Model { return &LocationModel{} }, // 添加其他模型 }