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

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

diff --git a/model/clientLevel.go b/model/clientLevel.go
index ee24d24..23ce754 100644
--- a/model/clientLevel.go
+++ b/model/clientLevel.go
@@ -3,6 +3,7 @@
 import (
 	"aps_crm/pkg/mysqlx"
 	"gorm.io/gorm"
+	"sync"
 )
 
 type (
@@ -82,3 +83,36 @@
 	var db = slf.build()
 	return db.Updates(data).Error
 }
+
+func (slf *ClientLevelSearch) CreateBatch(records []*ClientLevel) error {
+	var db = slf.build()
+	return db.Create(records).Error
+}
+
+// InitDefaultData 鍒濆鍖栨暟鎹�
+func (slf *ClientLevelSearch) 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 := []*ClientLevel{
+		{1, "閲嶇偣鍏虫敞瀹㈡埛"},
+		{2, "浼樿川瀹㈡埛"},
+		{3, "鏅�氬鎴�"},
+		{4, "鍏朵粬"},
+	}
+	err := slf.CreateBatch(records)
+	if err != nil {
+		errCh <- err
+		return
+	}
+}

--
Gitblit v1.8.0