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/serviceOrderStatus.go | 16 ++++++++++++---- 1 files changed, 12 insertions(+), 4 deletions(-) diff --git a/model/serviceOrderStatus.go b/model/serviceOrderStatus.go index 44ff749..fdb01e9 100644 --- a/model/serviceOrderStatus.go +++ b/model/serviceOrderStatus.go @@ -6,6 +6,7 @@ "errors" "fmt" "gorm.io/gorm" + "sync" ) type ( @@ -129,16 +130,19 @@ } // InitDefaultData 鍒濆鍖栨暟鎹� -func (slf *ServiceOrderStatusSearch) InitDefaultData() error { +func (slf *ServiceOrderStatusSearch) 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 := []*ServiceOrderStatus{ {1, "鏈鐞�"}, @@ -146,5 +150,9 @@ {3, "绛夊緟鍥炲簲"}, {4, "鎴愬姛鍏抽棴"}, } - return slf.CreateBatch(records) + err := slf.CreateBatch(records) + if err != nil { + errCh <- err + return + } } -- Gitblit v1.8.0