zhangqian
2023-11-01 530fed8ec225453572d57b15c200ab062c335457
model/timelyRate.go
@@ -3,6 +3,7 @@
import (
   "aps_crm/pkg/mysqlx"
   "gorm.io/gorm"
   "sync"
)
type (
@@ -11,7 +12,7 @@
      Id   int    `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
      Name string `json:"name" gorm:"column:name;type:varchar(255);comment:商机阶段名称"`
   }
   TimelyRateSearch struct {
      TimelyRate
      Orm *gorm.DB
@@ -83,3 +84,42 @@
   var db = slf.build()
   return db.Updates(data).Error
}
func (slf *TimelyRateSearch) CreateBatch(records []*TimelyRate) error {
   var db = slf.build()
   return db.Create(records).Error
}
// InitDefaultData 初始化数据
func (slf *TimelyRateSearch) 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 := []*TimelyRate{
      {1, "100%"},
      {2, "90%"},
      {3, "80%"},
      {4, "70%"},
      {5, "60%"},
      {6, "50%"},
      {7, "40%"},
      {8, "30%"},
      {9, "20%"},
      {10, "10%"},
   }
   err := slf.CreateBatch(records)
   if err != nil {
      errCh <- err
      return
   }
}