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/serviceContractType.go | 16 ++++++++++++---- 1 files changed, 12 insertions(+), 4 deletions(-) diff --git a/model/serviceContractType.go b/model/serviceContractType.go index 3524737..4b64b32 100644 --- a/model/serviceContractType.go +++ b/model/serviceContractType.go @@ -3,6 +3,7 @@ import ( "aps_crm/pkg/mysqlx" "gorm.io/gorm" + "sync" ) type ( @@ -90,16 +91,18 @@ } // InitDefaultData 鍒濆鍖栨暟鎹� -func (slf *ServiceContractTypeSearch) InitDefaultData() error { +func (slf *ServiceContractTypeSearch) 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 := []*ServiceContractType{ {1, "鎸夊勾"}, @@ -107,5 +110,10 @@ {3, "鍏朵粬"}, {4, "铏氭嫙鍚堝悓"}, } - return slf.CreateBatch(records) + + err := slf.CreateBatch(records) + if err != nil { + errCh <- err + return + } } -- Gitblit v1.8.0