140 lines
3.4 KiB
Go
140 lines
3.4 KiB
Go
package jobmgr
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"gitlink.org.cn/cloudream/common/pkgs/actor"
|
|
schsdk "gitlink.org.cn/cloudream/common/sdks/scheduler"
|
|
jobmod "gitlink.org.cn/cloudream/scheduler/common/models/job"
|
|
advtsk "gitlink.org.cn/cloudream/scheduler/common/pkgs/mq/advisor/task"
|
|
"gitlink.org.cn/cloudream/scheduler/manager/internal/jobmgr/event"
|
|
)
|
|
|
|
type makingAdjustSchemeJob struct {
|
|
job *jobmod.NormalJob
|
|
state *jobmod.StateMakingAdjustScheme
|
|
}
|
|
|
|
type MakingAdjustSchemeHandler struct {
|
|
mgr *Manager
|
|
|
|
jobs map[schsdk.JobID]*makingAdjustSchemeJob
|
|
|
|
cmdChan actor.CommandChannel
|
|
}
|
|
|
|
func NewMakingAdjustSchemeHandler(mgr *Manager) *MakingAdjustSchemeHandler {
|
|
return &MakingAdjustSchemeHandler{
|
|
mgr: mgr,
|
|
jobs: make(map[schsdk.JobID]*makingAdjustSchemeJob),
|
|
cmdChan: *actor.NewCommandChannel(),
|
|
}
|
|
}
|
|
|
|
func (h *MakingAdjustSchemeHandler) Handle(job jobmod.Job) {
|
|
h.cmdChan.Send(func() {
|
|
norJob, ok := job.(*jobmod.NormalJob)
|
|
if !ok {
|
|
h.changeJobState(job, jobmod.NewStateFailed(fmt.Sprintf("unknow job: %v", reflect.TypeOf(job)), job.GetState()))
|
|
return
|
|
}
|
|
|
|
state, ok := job.GetState().(*jobmod.StateMakingAdjustScheme)
|
|
if !ok {
|
|
h.changeJobState(job, jobmod.NewStateFailed(fmt.Sprintf("unknow state: %v", reflect.TypeOf(job.GetState())), job.GetState()))
|
|
return
|
|
}
|
|
|
|
rjob := &makingAdjustSchemeJob{
|
|
job: norJob,
|
|
state: state,
|
|
}
|
|
h.jobs[job.GetJobID()] = rjob
|
|
|
|
h.onJobEvent(nil, rjob)
|
|
})
|
|
}
|
|
|
|
func (h *MakingAdjustSchemeHandler) onJobEvent(evt event.Event, job *makingAdjustSchemeJob) {
|
|
if cloneEvt, ok := evt.(*event.CloneJob); ok {
|
|
cloneEvt.Callback.SetValue(job.job.Clone())
|
|
return
|
|
}
|
|
|
|
if job.state.FullTaskID == "" {
|
|
fullTaskID, err := h.mgr.advMgr.StartTask(job.job.GetJobID(), advtsk.NewMakeAdjustScheme(*job.job))
|
|
if err != nil {
|
|
h.changeJobState(job.job, jobmod.NewStateFailed(err.Error(), job.state))
|
|
return
|
|
}
|
|
|
|
job.state.FullTaskID = fullTaskID
|
|
}
|
|
|
|
if makingRet, err := event.AssertAdvisorTaskStatus[*advtsk.MakeAdjustSchemeStatus](evt, job.state.FullTaskID); err != event.ErrUnconcernedTask {
|
|
if err == event.ErrTaskTimeout {
|
|
h.changeJobState(job.job, jobmod.NewStateFailed("make adjust scheme timeout", job.state))
|
|
return
|
|
}
|
|
|
|
h.mgr.advMgr.ForgetTask(job.state.FullTaskID)
|
|
|
|
if makingRet.Error != "" {
|
|
h.changeJobState(job.job, jobmod.NewStateFailed(makingRet.Error, job.state))
|
|
return
|
|
}
|
|
|
|
h.changeJobState(job.job, jobmod.NewStateAdjusting(makingRet.Scheme))
|
|
}
|
|
}
|
|
|
|
func (h *MakingAdjustSchemeHandler) changeJobState(job jobmod.Job, state jobmod.JobState) {
|
|
job.SetState(state)
|
|
|
|
delete(h.jobs, job.GetJobID())
|
|
|
|
h.mgr.pubLock.Lock()
|
|
h.mgr.handleState(job)
|
|
h.mgr.pubLock.Unlock()
|
|
}
|
|
|
|
func (h *MakingAdjustSchemeHandler) OnEvent(broadcast event.Broadcast, evt event.Event) {
|
|
h.cmdChan.Send(func() {
|
|
if broadcast.ToAll() {
|
|
for _, job := range h.jobs {
|
|
h.onJobEvent(evt, job)
|
|
}
|
|
|
|
} else if broadcast.ToJobSet() {
|
|
for _, job := range h.jobs {
|
|
if job.job.GetJobSetID() != broadcast.JobSetID {
|
|
continue
|
|
}
|
|
|
|
h.onJobEvent(evt, job)
|
|
}
|
|
} else if broadcast.ToJob() {
|
|
if job, ok := h.jobs[broadcast.JobID]; ok {
|
|
h.onJobEvent(evt, job)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
func (h *MakingAdjustSchemeHandler) Serve() {
|
|
cmdChan := h.cmdChan.BeginChanReceive()
|
|
defer h.cmdChan.CloseChanReceive()
|
|
|
|
for {
|
|
select {
|
|
case cmd := <-cmdChan:
|
|
cmd()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (h *MakingAdjustSchemeHandler) Stop() {
|
|
// TODO 支持STOP
|
|
}
|