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 | 31 +++++++++++++++++++------------ 1 files changed, 19 insertions(+), 12 deletions(-) diff --git a/model/serviceType.go b/model/serviceType.go index 402a9a5..b3e6348 100644 --- a/model/serviceType.go +++ b/model/serviceType.go @@ -6,16 +6,17 @@ "errors" "fmt" "gorm.io/gorm" + "sync" ) type ( - // ServiceType 鏈嶅姟绫诲瀷 + // 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 鏈嶅姟绫诲瀷鎼滅储鏉′欢 + // ServiceTypeSearch 鏈嶅姟鏂瑰紡鎼滅储鏉′欢 ServiceTypeSearch struct { ServiceType Orm *gorm.DB @@ -119,23 +120,29 @@ } // 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{ - {Name: "鐢佃瘽"}, - {Name: "杩滅▼"}, - {Name: "閫佷慨"}, - {Name: "涓婇棬"}, - {Name: "鍏朵粬"}, + {1, "鐢佃瘽"}, + {2, "杩滅▼"}, + {3, "閫佷慨"}, + {4, "涓婇棬"}, + {5, "鍏朵粬"}, } - return slf.CreateBatch(records) + err := slf.CreateBatch(records) + if err != nil { + errCh <- err + return + } } -- Gitblit v1.8.0