初始化一些表shuju

zhangqian

2023-10-17 b45b8ce8051e2ad9c80aafa42d1a5892f6a36bce

 model/salesSources.go

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

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

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

// InitDefaultData 初始化数据
func (slf *SalesSourcesSearch) 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 := []*SalesSources{
      {1, "电话来访"},
      {2, "公司分配"},
      {3, "客户介绍"},
      {4, "独立开发"},
   }

   err := slf.CreateBatch(records)
   if err != nil {
      errCh <- err
      return
   }
}