fix

wangpengfei

2023-08-18 9a7e0a7da01a9f9625ceaca0c61a59c540c6438f

 model/severity.go

@@ -6,24 +6,25 @@
   "errors"
   "fmt"
   "gorm.io/gorm"
   "sync"
)

type (
   // Severity 严重程度
   Severity struct {
      Id   int    `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
      Name   string    `json:"name" gorm:"column:name"`
      Name string `json:"name" gorm:"column:name"`
   }

   // SeveritySearch 严重程度搜索条件
   SeveritySearch struct {
      Severity
      Orm *gorm.DB
        QueryClass  constvar.SeverityQueryClass
        KeywordType constvar.SeverityKeywordType
        Keyword     string
        PageNum  int
        PageSize int
      Orm         *gorm.DB
      QueryClass  constvar.SeverityQueryClass
      KeywordType constvar.SeverityKeywordType
      Keyword     string
      PageNum     int
      PageSize    int
   }
)

@@ -119,17 +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{}
   return slf.CreateBatch(records)
   records := []*Severity{
      {1, "轻"},
      {2, "严重"},
      {3, "较严重"},
      {4, "特严重"},
   }
   err := slf.CreateBatch(records)
   if err != nil {
      errCh <- err
      return
   }
}