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 | 18 +++++++++++++-----
1 files changed, 13 insertions(+), 5 deletions(-)
diff --git a/model/severity.go b/model/severity.go
index 51c7f1b..51284df 100644
--- a/model/severity.go
+++ b/model/severity.go
@@ -6,6 +6,7 @@
"errors"
"fmt"
"gorm.io/gorm"
+ "sync"
)
type (
@@ -28,7 +29,7 @@
)
func (Severity) TableName() string {
- return "severity"
+ return "severity_order"
}
func NewSeveritySearch() *SeveritySearch {
@@ -119,16 +120,19 @@
}
// InitDefaultData 鍒濆鍖栨暟鎹�
-func (slf *SeveritySearch) InitDefaultData() error {
+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 {
- return err
+ errCh <- err
+ return
}
if total != 0 {
- return nil
+ return
}
records := []*Severity{
{1, "杞�"},
@@ -136,5 +140,9 @@
{3, "杈冧弗閲�"},
{4, "鐗逛弗閲�"},
}
- return slf.CreateBatch(records)
+ err := slf.CreateBatch(records)
+ if err != nil {
+ errCh <- err
+ return
+ }
}
--
Gitblit v1.8.0