From 663db1dd160e759c9401289554543b1d1ea1e5cb Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期六, 05 八月 2023 12:17:17 +0800
Subject: [PATCH] 服务单管理故障类别增删改查,服务单管理花费时间增删改查,增加服务方式、花费时间、严重程度、优先级别初始化数据

---
 model/priorityLevel.go |   42 ++++++++++++++++++++++++++++++++++--------
 1 files changed, 34 insertions(+), 8 deletions(-)

diff --git a/model/priorityLevel.go b/model/priorityLevel.go
index be5b783..b2240f8 100644
--- a/model/priorityLevel.go
+++ b/model/priorityLevel.go
@@ -12,18 +12,18 @@
 	// PriorityLevel 浼樺厛绾у埆
 	PriorityLevel struct {
 		Id   int    `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
-		Name   string    `json:"name" gorm:"column:name"`
+		Name string `json:"name" gorm:"column:name"`
 	}
 
 	// PriorityLevelSearch 浼樺厛绾у埆鎼滅储鏉′欢
 	PriorityLevelSearch struct {
 		PriorityLevel
-		Orm *gorm.DB
-        QueryClass  constvar.PriorityLevelQueryClass
-        KeywordType constvar.PriorityLevelKeywordType
-        Keyword     string
-        PageNum  int
-        PageSize int
+		Orm         *gorm.DB
+		QueryClass  constvar.PriorityLevelQueryClass
+		KeywordType constvar.PriorityLevelKeywordType
+		Keyword     string
+		PageNum     int
+		PageSize    int
 	}
 )
 
@@ -49,6 +49,11 @@
 func (slf *PriorityLevelSearch) Create(record *PriorityLevel) error {
 	var db = slf.build()
 	return db.Create(record).Error
+}
+
+func (slf *PriorityLevelSearch) CreateBatch(records []*PriorityLevel) error {
+	var db = slf.build()
+	return db.Create(records).Error
 }
 
 func (slf *PriorityLevelSearch) Delete() error {
@@ -111,4 +116,25 @@
 
 	err := db.Find(&records).Error
 	return records, total, err
-}
\ No newline at end of file
+}
+
+// InitDefaultData 鍒濆鍖栨暟鎹�
+func (slf *PriorityLevelSearch) InitDefaultData() error {
+	var (
+		db          = slf.Orm.Table(slf.TableName())
+		total int64 = 0
+	)
+	if err := db.Count(&total).Error; err != nil {
+		return err
+	}
+	if total != 0 {
+		return nil
+	}
+	records := []*PriorityLevel{
+		{Name: "浣�"},
+		{Name: "涓�鑸�"},
+		{Name: "楂�"},
+		{Name: "绱ф��"},
+	}
+	return slf.CreateBatch(records)
+}

--
Gitblit v1.8.0