From 7438f17c0b3a9f79ee5890c554bedc64c362859b Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 13 十月 2023 14:04:59 +0800 Subject: [PATCH] 销售计划简单数据权限 --- model/serviceType.go | 17 ++++++++++++----- 1 files changed, 12 insertions(+), 5 deletions(-) diff --git a/model/serviceType.go b/model/serviceType.go index 4602418..b3e6348 100644 --- a/model/serviceType.go +++ b/model/serviceType.go @@ -6,13 +6,14 @@ "errors" "fmt" "gorm.io/gorm" + "sync" ) type ( // ServiceType 鏈嶅姟鏂瑰紡 ServiceType 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;type:varchar(255);not null;default:''"` } // ServiceTypeSearch 鏈嶅姟鏂瑰紡鎼滅储鏉′欢 @@ -119,16 +120,18 @@ } // InitDefaultData 鍒濆鍖栨暟鎹� -func (slf *ServiceTypeSearch) InitDefaultData() error { +func (slf *ServiceTypeSearch) 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 := []*ServiceType{ {1, "鐢佃瘽"}, @@ -137,5 +140,9 @@ {4, "涓婇棬"}, {5, "鍏朵粬"}, } - return slf.CreateBatch(records) + err := slf.CreateBatch(records) + if err != nil { + errCh <- err + return + } } -- Gitblit v1.8.0