From 63b531279a8ab1d609910e7db53add83bbc0b97a Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 17 十月 2023 20:21:01 +0800
Subject: [PATCH] 添加默认数据

---
 model/registeredCapital.go |   35 +++++++++++++++++++++++++++++++++++
 1 files changed, 35 insertions(+), 0 deletions(-)

diff --git a/model/registeredCapital.go b/model/registeredCapital.go
index fb682a0..7d1dd2e 100644
--- a/model/registeredCapital.go
+++ b/model/registeredCapital.go
@@ -3,6 +3,7 @@
 import (
 	"aps_crm/pkg/mysqlx"
 	"gorm.io/gorm"
+	"sync"
 )
 
 type (
@@ -82,3 +83,37 @@
 	slf.Name = name
 	return slf
 }
+
+func (slf *RegisteredCapitalSearch) CreateBatch(records []*RegisteredCapital) error {
+	var db = slf.build()
+	return db.Create(records).Error
+}
+
+// InitDefaultData 鍒濆鍖栨暟鎹�
+func (slf *RegisteredCapitalSearch) 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 := []*RegisteredCapital{
+		{1, "50涓囦互涓�"},
+		{2, "50-100涓�"},
+		{3, "100-500涓�"},
+		{4, "500-1000涓�"},
+		{5, "1000涓囦互涓�"},
+	}
+	err := slf.CreateBatch(records)
+	if err != nil {
+		errCh <- err
+		return
+	}
+}

--
Gitblit v1.8.0