添加模型注册

zhangzengfei

2024-12-20 dccccdb75b0f61fb2f42bbf81399fe45b54e2c99

添加模型注册

 db/db.go

@@ -63,7 +63,7 @@
            },
            Name:        "疑似聚集",
            Description: "通用聚集模型",
            Version:     "v1.0.1",
            Version:     "v1.0.2",
            Enabled:     false,
        },
        {
@@ -72,7 +72,7 @@
            },
            Name:        "疑似脱管",
            Description: "通用脱管模型",
            Version:     "v1.0.0",
            Version:     "v1.1.0",
            Enabled:     false,
        },
    }

 models/model.go

@@ -2,6 +2,7 @@

import (
    "fmt"

    "model-engine/db"
)

@@ -12,7 +13,8 @@
}

var modelRegistry = map[string]func() Model{
    "gather": func() Model { return &GatherModel{} },
    "gather":    func() Model { return &GatherModel{} },
    "disappear": func() Model { return &DisappearModel{} },
    // 添加其他模型
}