From 8546acdc2a1f06253f30e0543fc202ebf5219bd0 Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期五, 18 八月 2023 16:31:40 +0800 Subject: [PATCH] fix --- model/priorityLevel.go | 16 ++++++++++++---- 1 files changed, 12 insertions(+), 4 deletions(-) diff --git a/model/priorityLevel.go b/model/priorityLevel.go index ee2f3fe..b1902f4 100644 --- a/model/priorityLevel.go +++ b/model/priorityLevel.go @@ -6,6 +6,7 @@ "errors" "fmt" "gorm.io/gorm" + "sync" ) type ( @@ -119,16 +120,19 @@ } // InitDefaultData 鍒濆鍖栨暟鎹� -func (slf *PriorityLevelSearch) InitDefaultData() error { +func (slf *PriorityLevelSearch) 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 := []*PriorityLevel{ {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