From 4e269daa939f3da316a0e079ca35b05e162adfcb Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期四, 19 十二月 2024 10:26:26 +0800
Subject: [PATCH] 调整初始化

---
 models/gather_model.go |    3 ++-
 db/db.go               |   18 +++++++++---------
 2 files changed, 11 insertions(+), 10 deletions(-)

diff --git a/db/db.go b/db/db.go
index 057191e..de59e6a 100644
--- a/db/db.go
+++ b/db/db.go
@@ -1,8 +1,6 @@
 package db
 
 import (
-	"time"
-
 	"github.com/elastic/go-elasticsearch/v6"
 	"gorm.io/gorm"
 
@@ -59,25 +57,27 @@
 
 // InitDefaultData 鍒濆鍖栨暟鎹�
 func InitDefaultData() error {
-	var models = []Model{
+	var models = []*Model{
 		{
 			BaseModel: BaseModel{
-				ID:        ModelIdGather,
-				CreatedAt: time.Time{},
-				UpdatedAt: time.Time{},
+				ID: ModelIdGather,
 			},
 			Name:        "鐤戜技鑱氶泦",
 			Description: "閫氱敤鑱氶泦妯″瀷",
-			Version:     "v1.0.0",
+			Version:     "v1.0.1",
 			Enabled:     false,
 		},
 	}
 
 	for i := range models {
-		GetDB().Save(&models[i])
+		if err := GetDB().Model(models[i]).First(models[i].ID).Error; err != nil {
+			GetDB().Model(models[i]).Create(models[i])
+		} else {
+			GetDB().Model(models[i]).Update("version", models[i].Version)
+		}
 	}
 
-	var rules = []ModelRule{
+	var rules = []*ModelRule{
 		{
 			Id:      "bfbdba7f-ee39-41fb-b188-b4c114a51eaa",
 			ModelId: ModelIdGather,
diff --git a/models/gather_model.go b/models/gather_model.go
index c576627..3724c0d 100644
--- a/models/gather_model.go
+++ b/models/gather_model.go
@@ -81,9 +81,10 @@
 		}
 	}
 
-	fmt.Printf("GatherModel init finish ...task id:%s, name:%s, rule:%+v\n", task.ID, task.Name, task.Rules)
+	fmt.Printf("GatherModel init finish ...task id:%s, name:%s, rule:%+v\n", task.ID, task.Name, m)
 
 	if m.GatherPersons == 0 || m.AppearInterval == 0 || m.DaysWindow == 0 || m.Threshold == 0 {
+		fmt.Printf("invalid parameters. task id:%s, name:%s\n", task.ID, task.Name)
 		return errors.New("invalid parameters")
 	}
 

--
Gitblit v1.8.0