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/faultType.go | 16 ++++++++++++----
1 files changed, 12 insertions(+), 4 deletions(-)
diff --git a/model/faultType.go b/model/faultType.go
index 82e9a65..248233e 100644
--- a/model/faultType.go
+++ b/model/faultType.go
@@ -6,6 +6,7 @@
"errors"
"fmt"
"gorm.io/gorm"
+ "sync"
)
type (
@@ -119,16 +120,18 @@
}
// InitDefaultData 鍒濆鍖栨暟鎹�
-func (slf *FaultTypeSearch) InitDefaultData() error {
+func (slf *FaultTypeSearch) 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 := []*FaultType{
{1, "璁惧瀹夎"},
@@ -138,5 +141,10 @@
{5, "璁惧鏁呴殰"},
{6, "璁惧鑹搧鐜囦綆"},
}
- return slf.CreateBatch(records)
+
+ err := slf.CreateBatch(records)
+ if err != nil {
+ errCh <- err
+ return
+ }
}
--
Gitblit v1.8.0