zhangqian
2023-11-01 530fed8ec225453572d57b15c200ab062c335457
model/priorityLevel.go
@@ -6,24 +6,25 @@
   "errors"
   "fmt"
   "gorm.io/gorm"
   "sync"
)
type (
   // PriorityLevel 优先级别
   PriorityLevel 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"`
   }
   // PriorityLevelSearch 优先级别搜索条件
   PriorityLevelSearch struct {
      PriorityLevel
      Orm *gorm.DB
        QueryClass  constvar.PriorityLevelQueryClass
        KeywordType constvar.PriorityLevelKeywordType
        Keyword     string
        PageNum  int
        PageSize int
      Orm         *gorm.DB
      QueryClass  constvar.PriorityLevelQueryClass
      KeywordType constvar.PriorityLevelKeywordType
      Keyword     string
      PageNum     int
      PageSize    int
   }
)
@@ -49,6 +50,11 @@
func (slf *PriorityLevelSearch) Create(record *PriorityLevel) error {
   var db = slf.build()
   return db.Create(record).Error
}
func (slf *PriorityLevelSearch) CreateBatch(records []*PriorityLevel) error {
   var db = slf.build()
   return db.Create(records).Error
}
func (slf *PriorityLevelSearch) Delete() error {
@@ -111,4 +117,32 @@
   err := db.Find(&records).Error
   return records, total, err
}
}
// InitDefaultData 初始化数据
func (slf *PriorityLevelSearch) 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 := []*PriorityLevel{
      {1, "低"},
      {2, "一般"},
      {3, "高"},
      {4, "紧急"},
   }
   err := slf.CreateBatch(records)
   if err != nil {
      errCh <- err
      return
   }
}