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/serviceType.go | 15 +++++++++++----
1 files changed, 11 insertions(+), 4 deletions(-)
diff --git a/model/serviceType.go b/model/serviceType.go
index 9938c03..b3e6348 100644
--- a/model/serviceType.go
+++ b/model/serviceType.go
@@ -6,6 +6,7 @@
"errors"
"fmt"
"gorm.io/gorm"
+ "sync"
)
type (
@@ -119,16 +120,18 @@
}
// InitDefaultData 鍒濆鍖栨暟鎹�
-func (slf *ServiceTypeSearch) InitDefaultData() error {
+func (slf *ServiceTypeSearch) 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 := []*ServiceType{
{1, "鐢佃瘽"},
@@ -137,5 +140,9 @@
{4, "涓婇棬"},
{5, "鍏朵粬"},
}
- return slf.CreateBatch(records)
+ err := slf.CreateBatch(records)
+ if err != nil {
+ errCh <- err
+ return
+ }
}
--
Gitblit v1.8.0