初始化仓库数据

zhangqian

2023-08-24 fd85a0bc722eb2d84cb751d6d521971f41684b7a

初始化仓库数据

 model/index.go

@@ -117,6 +117,7 @@
        NewIsVisitSearch(),
        NewTimelyRateSearch(),
        NewFaultTypeSearch(),
        NewRepositorySearch(),
    }

    for _, model := range models {

 model/repository.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:商机阶段名称"`
    }
   

    RepositorySearch struct {
        Repository
        Orm *gorm.DB
@@ -83,3 +84,35 @@
    var db = slf.build()
    return db.Updates(data).Error
}

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

// InitDefaultData 初始化数据
func (slf *RepositorySearch) 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 := []*Repository{
        {1, "总仓"},
        {2, "赠品仓"},
        {3, "其他仓"},
    }
    err := slf.CreateBatch(records)
    if err != nil {
        errCh <- err
        return
    }
}