| | |
| | | "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 |
| | | } |
| | | ) |
| | | |
| | |
| | | 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 { |
| | |
| | | |
| | | 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 |
| | | } |
| | | } |