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

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

diff --git a/model/severity.go b/model/severity.go
index 0fb1bc0..681c4f7 100644
--- a/model/severity.go
+++ b/model/severity.go
@@ -12,18 +12,18 @@
 	// Severity 涓ラ噸绋嬪害
 	Severity 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"`
 	}
 
 	// SeveritySearch 涓ラ噸绋嬪害鎼滅储鏉′欢
 	SeveritySearch struct {
 		Severity
-		Orm *gorm.DB
-        QueryClass  constvar.SeverityQueryClass
-        KeywordType constvar.SeverityKeywordType
-        Keyword     string
-        PageNum  int
-        PageSize int
+		Orm         *gorm.DB
+		QueryClass  constvar.SeverityQueryClass
+		KeywordType constvar.SeverityKeywordType
+		Keyword     string
+		PageNum     int
+		PageSize    int
 	}
 )
 
@@ -49,6 +49,11 @@
 func (slf *SeveritySearch) Create(record *Severity) error {
 	var db = slf.build()
 	return db.Create(record).Error
+}
+
+func (slf *SeveritySearch) CreateBatch(records []*Severity) error {
+	var db = slf.build()
+	return db.Create(records).Error
 }
 
 func (slf *SeveritySearch) Delete() error {
@@ -111,4 +116,25 @@
 
 	err := db.Find(&records).Error
 	return records, total, err
-}
\ No newline at end of file
+}
+
+// InitDefaultData 鍒濆鍖栨暟鎹�
+func (slf *SeveritySearch) 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 := []*Severity{
+		{Name: "杞�"},
+		{Name: "涓ラ噸"},
+		{Name: "杈冧弗閲�"},
+		{Name: "鐗逛弗閲�"},
+	}
+	return slf.CreateBatch(records)
+}

--
Gitblit v1.8.0