model/severity.go
@@ -51,6 +51,11 @@
   return db.Create(record).Error
}
func (slf *SeveritySearch) CreateBatch(records []*Severity) error {
   var db = slf.build()
   return db.Create(records).Error
}
func (slf *SeveritySearch) Delete() error {
   var db = slf.build()
   return db.Delete(&Severity{}).Error
@@ -112,3 +117,24 @@
   err := db.Find(&records).Error
   return records, total, err
}
// InitDefaultData 初始化数据
func (slf *SeveritySearch) InitDefaultData() error {
   var (
      db          = slf.Orm.Table(slf.TableName())
      total int64 = 0
   )
   if err := db.Count(&total).Error; err != nil {
      return err
   }
   if total != 0 {
      return nil
   }
   records := []*Severity{
      {Name: "轻"},
      {Name: "严重"},
      {Name: "较严重"},
      {Name: "特严重"},
   }
   return slf.CreateBatch(records)
}