zhangzengfei
2025-02-11 e4fc307d58856a0b319c41f0930cd3b731cb0b24
models/model.go
@@ -2,17 +2,23 @@
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{} },
   "night":            func() Model { return &nightModel{} },
   "accessRegularity": func() Model { return &RegularityModel{} },
   // 添加其他模型
}