36 lines
1.1 KiB
Go
36 lines
1.1 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go-common/app/job/main/workflow/model"
|
|
)
|
|
|
|
// ChallByIDs get chall list by ids.
|
|
func (d *Dao) ChallByIDs(c context.Context, cids []int64) (res map[int64]*model.Chall, err error) {
|
|
if len(cids) <= 0 {
|
|
return
|
|
}
|
|
res = make(map[int64]*model.Chall)
|
|
cList := make([]*model.Chall, 0, len(cids))
|
|
if err = d.ReadORM.Table("workflow_chall").Select("id, business, dispatch_state, dispatch_time").Where("id IN (?)", cids).Find(&cList).Error; err != nil {
|
|
return
|
|
}
|
|
for _, c := range cList {
|
|
res[c.ID] = c
|
|
}
|
|
return
|
|
}
|
|
|
|
// UpDispatchStateByIDs update by ids.
|
|
func (d *Dao) UpDispatchStateByIDs(c context.Context, cids []int64, dispatchState int64) (err error) {
|
|
err = d.WriteORM.Table("workflow_chall").Where("id IN (?)", cids).Update("dispatch_state", dispatchState).Error
|
|
return
|
|
}
|
|
|
|
// UpDispatchStateAdminIDByIds .
|
|
func (d *Dao) UpDispatchStateAdminIDByIds(c context.Context, cids []int64, dispatchState, assignAdminid int64) (err error) {
|
|
err = d.WriteORM.Table("workflow_chall").Where("id IN (?)", cids).Update("dispatch_state", dispatchState).Update("assignee_adminid", assignAdminid).Error
|
|
return
|
|
}
|