From 7438f17c0b3a9f79ee5890c554bedc64c362859b Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 13 十月 2023 14:04:59 +0800 Subject: [PATCH] 销售计划简单数据权限 --- model/timeSpent.go | 30 +++++++++++++++++++----------- 1 files changed, 19 insertions(+), 11 deletions(-) diff --git a/model/timeSpent.go b/model/timeSpent.go index 1a4a5b0..926d4de 100644 --- a/model/timeSpent.go +++ b/model/timeSpent.go @@ -6,13 +6,14 @@ "errors" "fmt" "gorm.io/gorm" + "sync" ) type ( // TimeSpent 鑺辫垂鏃堕棿 TimeSpent 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:'';comment:鍚嶇О"` } // TimeSpentSearch 鑺辫垂鏃堕棿鎼滅储鏉′欢 @@ -119,24 +120,31 @@ } // InitDefaultData 鍒濆鍖栨暟鎹� -func (slf *TimeSpentSearch) InitDefaultData() error { +func (slf *TimeSpentSearch) 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 := []*TimeSpent{ - {Name: "1灏忔椂"}, - {Name: "2灏忔椂"}, - {Name: "0.5涓伐浣滄棩"}, - {Name: "1涓伐浣滄棩"}, - {Name: "2涓伐浣滄棩"}, - {Name: "2涓伐浣滄棩浠ヤ笂"}, + {1, "1灏忔椂"}, + {2, "2灏忔椂"}, + {3, "0.5涓伐浣滄棩"}, + {4, "1涓伐浣滄棩"}, + {5, "2涓伐浣滄棩"}, + {6, "2涓伐浣滄棩浠ヤ笂"}, } - return slf.CreateBatch(records) + err := slf.CreateBatch(records) + if err != nil { + errCh <- err + return + } } -- Gitblit v1.8.0