From 4e269daa939f3da316a0e079ca35b05e162adfcb Mon Sep 17 00:00:00 2001 From: zhangzengfei <zhangzengfei@smartai.com> Date: 星期四, 19 十二月 2024 10:26:26 +0800 Subject: [PATCH] 调整初始化 --- db/db.go | 18 +++++++++--------- 1 files changed, 9 insertions(+), 9 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, -- Gitblit v1.8.0