From 63b531279a8ab1d609910e7db53add83bbc0b97a Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 17 十月 2023 20:21:01 +0800
Subject: [PATCH] 添加默认数据
---
model/enterpriseScale.go | 36 ++++++++++++++++++++++++++++++++++++
1 files changed, 36 insertions(+), 0 deletions(-)
diff --git a/model/enterpriseScale.go b/model/enterpriseScale.go
index e8cf21e..9943537 100644
--- a/model/enterpriseScale.go
+++ b/model/enterpriseScale.go
@@ -3,6 +3,7 @@
import (
"aps_crm/pkg/mysqlx"
"gorm.io/gorm"
+ "sync"
)
type (
@@ -86,3 +87,38 @@
slf.Id = id
return slf
}
+
+func (slf *EnterpriseScaleSearch) CreateBatch(records []*EnterpriseScale) error {
+ var db = slf.build()
+ return db.Create(records).Error
+}
+
+// InitDefaultData 鍒濆鍖栨暟鎹�
+func (slf *EnterpriseScaleSearch) 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 {
+ errCh <- err
+ return
+ }
+ if total != 0 {
+ return
+ }
+ records := []*EnterpriseScale{
+ {1, "20浜轰互涓�"},
+ {2, "20-50浜�"},
+ {3, "50-100浜�"},
+ {4, "100-500浜�"},
+ {5, "500-1000浜�"},
+ {6, "1000浜轰互涓�"},
+ }
+ err := slf.CreateBatch(records)
+ if err != nil {
+ errCh <- err
+ return
+ }
+}
--
Gitblit v1.8.0