From 5f0170df9787c6f3ff17cc168b3f2e3f511453f6 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 31 十月 2023 20:04:16 +0800
Subject: [PATCH] 支持职级降级的情况
---
model/severity.go | 52 +++++++++++++++++++++++++++++++++++++++++++---------
1 files changed, 43 insertions(+), 9 deletions(-)
diff --git a/model/severity.go b/model/severity.go
index 0fb1bc0..51284df 100644
--- a/model/severity.go
+++ b/model/severity.go
@@ -6,29 +6,30 @@
"errors"
"fmt"
"gorm.io/gorm"
+ "sync"
)
type (
// 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
}
)
func (Severity) TableName() string {
- return "severity"
+ return "severity_order"
}
func NewSeveritySearch() *SeveritySearch {
@@ -49,6 +50,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 +117,32 @@
err := db.Find(&records).Error
return records, total, err
-}
\ No newline at end of file
+}
+
+// InitDefaultData 鍒濆鍖栨暟鎹�
+func (slf *SeveritySearch) InitDefaultData(errCh chan<- error, wg *sync.WaitGroup) {
+ var (
+ db = slf.Orm.Table(slf.TableName())
+ total int64 = 0
+ )
+ defer wg.Done()
+
+ if err := db.Count(&total).Error; err != nil {
+ errCh <- err
+ return
+ }
+ if total != 0 {
+ return
+ }
+ records := []*Severity{
+ {1, "杞�"},
+ {2, "涓ラ噸"},
+ {3, "杈冧弗閲�"},
+ {4, "鐗逛弗閲�"},
+ }
+ err := slf.CreateBatch(records)
+ if err != nil {
+ errCh <- err
+ return
+ }
+}
--
Gitblit v1.8.0