From 63b531279a8ab1d609910e7db53add83bbc0b97a Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期二, 17 十月 2023 20:21:01 +0800 Subject: [PATCH] 添加默认数据 --- model/enterpriseNature.go | 37 +++++++++++++++++++++++++++++++++++++ 1 files changed, 37 insertions(+), 0 deletions(-) diff --git a/model/enterpriseNature.go b/model/enterpriseNature.go index f210218..44379b8 100644 --- a/model/enterpriseNature.go +++ b/model/enterpriseNature.go @@ -3,6 +3,7 @@ import ( "aps_crm/pkg/mysqlx" "gorm.io/gorm" + "sync" ) type ( @@ -84,3 +85,39 @@ var db = slf.build() return db.Updates(data).Error } + +func (slf *EnterpriseNatureSearch) CreateBatch(records []*EnterpriseNature) error { + var db = slf.build() + return db.Create(records).Error +} + +// InitDefaultData 鍒濆鍖栨暟鎹� +func (slf *EnterpriseNatureSearch) 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 := []*EnterpriseNature{ + {1, "鍥芥湁浼佷笟"}, + {2, "闆嗕綋浼佷笟"}, + {3, "绉佽惀浼佷笟"}, + {4, "涓綋宸ュ晢鎴�"}, + {5, "鍚堜紮浼佷笟"}, + {6, "鑱旇惀浼佷笟"}, + {7, "鑲′唤鍚堜綔鍒朵紒涓�"}, + } + err := slf.CreateBatch(records) + if err != nil { + errCh <- err + return + } +} -- Gitblit v1.8.0