From 39d53cf7a2080b49224dec6c395fc4ed2463a424 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期三, 18 十二月 2024 18:23:20 +0800
Subject: [PATCH] 修改参数的获取逻辑
---
db/db.go | 104 ++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 104 insertions(+), 0 deletions(-)
diff --git a/db/db.go b/db/db.go
index 54de63e..057191e 100644
--- a/db/db.go
+++ b/db/db.go
@@ -1,8 +1,11 @@
package db
import (
+ "time"
+
"github.com/elastic/go-elasticsearch/v6"
"gorm.io/gorm"
+
"model-engine/config"
"model-engine/db/es"
"model-engine/pkg/logger"
@@ -17,6 +20,9 @@
if err := es.InitClient([]string{"http://" + config.EsInfo.Ip + ":" + config.EsInfo.Port}); err != nil {
return err
}
+
+ InitDefaultData()
+
return nil
}
@@ -50,3 +56,101 @@
return nil
}
+
+// InitDefaultData 鍒濆鍖栨暟鎹�
+func InitDefaultData() error {
+ var models = []Model{
+ {
+ BaseModel: BaseModel{
+ ID: ModelIdGather,
+ CreatedAt: time.Time{},
+ UpdatedAt: time.Time{},
+ },
+ Name: "鐤戜技鑱氶泦",
+ Description: "閫氱敤鑱氶泦妯″瀷",
+ Version: "v1.0.0",
+ Enabled: false,
+ },
+ }
+
+ for i := range models {
+ GetDB().Save(&models[i])
+ }
+
+ var rules = []ModelRule{
+ {
+ Id: "bfbdba7f-ee39-41fb-b188-b4c114a51eaa",
+ ModelId: ModelIdGather,
+ Scope: "",
+ RuleArg: RuleArg{
+ Alias: "gatherPersons",
+ Name: "鑱氶泦浜烘暟",
+ Type: "input",
+ Must: true,
+ Unit: "浜�",
+ Range: "1,100",
+ Value: "2",
+ ValType: "int",
+ Operator: ">=",
+ Sort: 0,
+ },
+ },
+ {
+ Id: "941bef84-ff7f-4460-b5dc-2ac6060304a4",
+ ModelId: ModelIdGather,
+ Scope: "",
+ RuleArg: RuleArg{
+ Alias: "appearInterval",
+ Name: "鍑虹幇闂撮殧",
+ Type: "input",
+ Must: true,
+ Unit: "绉�",
+ Range: "1,7200",
+ Value: "60",
+ ValType: "int",
+ Operator: ">=",
+ Sort: 1,
+ },
+ },
+ {
+ Id: "a9b50bae-2c40-40a1-9ebc-ac34850db964",
+ ModelId: ModelIdGather,
+ Scope: "",
+ RuleArg: RuleArg{
+ Alias: "threshold",
+ Name: "鍑虹幇娆℃暟",
+ Type: "input",
+ Must: true,
+ Unit: "娆�",
+ Range: "1,60",
+ Value: "1",
+ ValType: "int",
+ Operator: ">=",
+ Sort: 2,
+ },
+ },
+ {
+ Id: "aed7f95e-1ce6-4fa2-b1b3-aaf59ed86c50",
+ ModelId: ModelIdGather,
+ Scope: "",
+ RuleArg: RuleArg{
+ Alias: "daysWindow",
+ Name: "鐩戞帶鏃堕棿",
+ Type: "input",
+ Must: true,
+ Unit: "澶╁唴",
+ Range: "1,7",
+ Value: "1",
+ ValType: "int",
+ Operator: "==",
+ Sort: 2,
+ },
+ },
+ }
+
+ for i := range rules {
+ GetDB().Save(&rules[i])
+ }
+
+ return nil
+}
--
Gitblit v1.8.0