zhangqian
2023-11-01 530fed8ec225453572d57b15c200ab062c335457
model/severity.go
@@ -6,6 +6,7 @@
   "errors"
   "fmt"
   "gorm.io/gorm"
   "sync"
)
type (
@@ -28,7 +29,7 @@
)
func (Severity) TableName() string {
   return "severity"
   return "severity_order"
}
func NewSeveritySearch() *SeveritySearch {
@@ -119,22 +120,29 @@
}
// InitDefaultData 初始化数据
func (slf *SeveritySearch) InitDefaultData() error {
func (slf *SeveritySearch) 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 := []*Severity{
      {Name: "轻"},
      {Name: "严重"},
      {Name: "较严重"},
      {Name: "特严重"},
      {1, "轻"},
      {2, "严重"},
      {3, "较严重"},
      {4, "特严重"},
   }
   return slf.CreateBatch(records)
   err := slf.CreateBatch(records)
   if err != nil {
      errCh <- err
      return
   }
}