From 63b531279a8ab1d609910e7db53add83bbc0b97a Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 17 十月 2023 20:21:01 +0800
Subject: [PATCH] 添加默认数据
---
model/paymentType.go | 36 ++++++++++++++++++++++++------------
1 files changed, 24 insertions(+), 12 deletions(-)
diff --git a/model/paymentType.go b/model/paymentType.go
index e16a84a..cd1e6bc 100644
--- a/model/paymentType.go
+++ b/model/paymentType.go
@@ -6,24 +6,25 @@
"errors"
"fmt"
"gorm.io/gorm"
+ "sync"
)
type (
// PaymentType 鏀粯鏂瑰紡
PaymentType struct {
Id int `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
- Name string `json:"name" gorm:"column:name"`
+ Name string `json:"name" gorm:"column:name"`
}
// PaymentTypeSearch 鏀粯鏂瑰紡鎼滅储鏉′欢
PaymentTypeSearch struct {
PaymentType
- Orm *gorm.DB
- QueryClass constvar.PaymentTypeQueryClass
- KeywordType constvar.PaymentTypeKeywordType
- Keyword string
- PageNum int
- PageSize int
+ Orm *gorm.DB
+ QueryClass constvar.PaymentTypeQueryClass
+ KeywordType constvar.PaymentTypeKeywordType
+ Keyword string
+ PageNum int
+ PageSize int
}
)
@@ -124,17 +125,28 @@
}
// InitDefaultData 鍒濆鍖栨暟鎹�
-func (slf *PaymentTypeSearch) InitDefaultData() error {
+func (slf *PaymentTypeSearch) 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 := []*PaymentType{}
- return slf.CreateBatch(records)
+ records := []*PaymentType{
+ {1, "鍘熻矾杩斿洖"},
+ {2, "绾夸笅"},
+ {3, "鏀エ"},
+ }
+ err := slf.CreateBatch(records)
+ if err != nil {
+ errCh <- err
+ return
+ }
}
--
Gitblit v1.8.0