105 lines
1.8 KiB
Go
105 lines
1.8 KiB
Go
package service
|
|
|
|
import (
|
|
"blazing/cool"
|
|
"blazing/modules/blazing/model"
|
|
"encoding/json"
|
|
)
|
|
|
|
type TaskService struct {
|
|
*UserService
|
|
}
|
|
|
|
func NewTaskService(id uint32) *TaskService {
|
|
return &TaskService{
|
|
&UserService{
|
|
userid: id,
|
|
Service: &cool.Service{
|
|
Model: model.NewTask(),
|
|
},
|
|
},
|
|
}
|
|
}
|
|
func (s *TaskService) Exec(t func(map[uint32]model.TaskInfo) bool) (ret bool) {
|
|
var tt model.Task
|
|
s.GetModel().Scan(&tt)
|
|
var ttt map[uint32]model.TaskInfo
|
|
json.Unmarshal([]byte(tt.Data), &ttt)
|
|
ret = t(ttt)
|
|
t1, _ := json.Marshal(&ttt)
|
|
tt.Data = string(t1)
|
|
s.GetModel().Save(&tt) //退出时保存
|
|
return
|
|
}
|
|
func (s *TaskService) AcceptTask(task uint32) (ret bool) {
|
|
|
|
s.Exec(func(ttt map[uint32]model.TaskInfo) bool {
|
|
ft, ok := ttt[task]
|
|
if ok { //如果找到任务
|
|
if ft.Status == 0 { //可以接受
|
|
ft.Status = 1
|
|
return true
|
|
} else {
|
|
return false
|
|
}
|
|
} else {
|
|
ttt[task] = model.TaskInfo{
|
|
Status: 1,
|
|
}
|
|
|
|
}
|
|
|
|
return false
|
|
})
|
|
|
|
return ret
|
|
}
|
|
|
|
/**
|
|
* 更新任务步骤
|
|
*/
|
|
func (s *TaskService) AddTaskBuf(task uint32, info model.TaskInfo) bool {
|
|
|
|
return s.Exec(func(ttt map[uint32]model.TaskInfo) bool {
|
|
if conditions, ok := ttt[task]; ok {
|
|
conditions.TaskInfo = info.TaskInfo
|
|
ttt[task] = conditions
|
|
return true
|
|
}
|
|
|
|
return false
|
|
})
|
|
|
|
}
|
|
|
|
/**
|
|
* 完成任务
|
|
*/
|
|
func (s *TaskService) CompleteTask(task uint32) bool {
|
|
|
|
return s.Exec(func(ttt map[uint32]model.TaskInfo) bool {
|
|
if conditions, ok := ttt[task]; ok {
|
|
conditions.Status = 3
|
|
ttt[task] = conditions
|
|
|
|
}
|
|
|
|
return false
|
|
})
|
|
}
|
|
|
|
/**
|
|
* 校验任务是否已经完成
|
|
*/
|
|
func (s *TaskService) CheckTaskCompleted(task uint32) bool {
|
|
|
|
return s.Exec(func(ttt map[uint32]model.TaskInfo) bool {
|
|
if conditions, ok := ttt[task]; ok {
|
|
|
|
return conditions.Status == 3
|
|
}
|
|
|
|
return false
|
|
})
|
|
}
|