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/invoiceStatus.go | 16 ++++++++++++----
1 files changed, 12 insertions(+), 4 deletions(-)
diff --git a/model/invoiceStatus.go b/model/invoiceStatus.go
index a4f7dac..19a4cde 100644
--- a/model/invoiceStatus.go
+++ b/model/invoiceStatus.go
@@ -6,6 +6,7 @@
"errors"
"fmt"
"gorm.io/gorm"
+ "sync"
)
type (
@@ -124,21 +125,28 @@
}
// InitDefaultData 鍒濆鍖栨暟鎹�
-func (slf *InvoiceStatusSearch) InitDefaultData() error {
+func (slf *InvoiceStatusSearch) 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 := []*InvoiceStatus{
{1, "宸插垱寤�"},
{2, "宸插瘎閫�"},
{3, "鑷姩鍒涘缓"},
}
- return slf.CreateBatch(records)
+
+ err := slf.CreateBatch(records)
+ if err != nil {
+ errCh <- err
+ return
+ }
}
--
Gitblit v1.8.0