初始化字段补充

jiangshuai

2023-10-17 9e2b19b7336af1e58b7e3861f6cc3a28ad641326

 model/saleStage.go

@@ -3,6 +3,7 @@
import (
   "aps_crm/pkg/mysqlx"
   "gorm.io/gorm"
   "sync"
)

type (
@@ -84,3 +85,37 @@
   var db = slf.build()
   return db.Updates(data).Error
}

func (slf *SaleStageSearch) CreateBatch(records []*SaleStage) error {
   var db = slf.build()
   return db.Create(records).Error
}

// InitDefaultData 初始化数据
func (slf *SaleStageSearch) 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 := []*SaleStage{
      {1, "初期沟通"},
      {2, "需求分析"},
      {3, "商务谈判"},
      {4, "成功关闭"},
      {4, "成功结案"},
   }
   err := slf.CreateBatch(records)
   if err != nil {
      errCh <- err
      return
   }
}