123 lines
2.8 KiB
Go
123 lines
2.8 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"
|
|
"gitlink.org.cn/cloudream/scheduler/manager/internal/jobmgr/event"
|
|
)
|
|
|
|
type readyToExecuteJob struct {
|
|
job jobmod.Job
|
|
state *jobmod.StateReadyToExecute
|
|
}
|
|
|
|
type ReadyToExecuteHandler struct {
|
|
mgr *Manager
|
|
|
|
jobs map[schsdk.JobID]*readyToExecuteJob
|
|
|
|
cmdChan actor.CommandChannel
|
|
}
|
|
|
|
func NewReadyToExecuteHandler(mgr *Manager) *ReadyToExecuteHandler {
|
|
return &ReadyToExecuteHandler{
|
|
mgr: mgr,
|
|
jobs: make(map[schsdk.JobID]*readyToExecuteJob),
|
|
cmdChan: *actor.NewCommandChannel(),
|
|
}
|
|
}
|
|
|
|
func (h *ReadyToExecuteHandler) Handle(job jobmod.Job) {
|
|
h.cmdChan.Send(func() {
|
|
state, ok := job.GetState().(*jobmod.StateReadyToExecute)
|
|
if !ok {
|
|
h.changeJobState(job, jobmod.NewStateFailed(fmt.Sprintf("unknow state: %v", reflect.TypeOf(job.GetState())), job.GetState()))
|
|
return
|
|
}
|
|
|
|
rjob := &readyToExecuteJob{
|
|
job: job,
|
|
state: state,
|
|
}
|
|
h.jobs[job.GetJobID()] = rjob
|
|
|
|
h.onJobEvent(nil, rjob)
|
|
})
|
|
}
|
|
|
|
func (h *ReadyToExecuteHandler) onJobEvent(evt event.Event, job *readyToExecuteJob) {
|
|
if cloneEvt, ok := evt.(*event.CloneJob); ok {
|
|
cloneEvt.Callback.SetValue(job.job.Clone())
|
|
return
|
|
}
|
|
|
|
if norJob, ok := job.job.(*jobmod.NormalJob); ok {
|
|
h.onNormalJobEvent(evt, job, norJob)
|
|
} else if resJob, ok := job.job.(*jobmod.ResourceJob); ok {
|
|
h.onResourceJobEvent(evt, job, resJob)
|
|
}
|
|
}
|
|
|
|
func (h *ReadyToExecuteHandler) onNormalJobEvent(evt event.Event, job *readyToExecuteJob, norJob *jobmod.NormalJob) {
|
|
// TODO 目前直接启动执行
|
|
h.changeJobState(job.job, jobmod.NewStateExecuting())
|
|
}
|
|
|
|
func (h *ReadyToExecuteHandler) onResourceJobEvent(evt event.Event, job *readyToExecuteJob, resJob *jobmod.ResourceJob) {
|
|
// TODO 目前直接启动执行
|
|
h.changeJobState(job.job, jobmod.NewStateExecuting())
|
|
}
|
|
|
|
func (h *ReadyToExecuteHandler) 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 *ReadyToExecuteHandler) 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 *ReadyToExecuteHandler) Serve() {
|
|
cmdChan := h.cmdChan.BeginChanReceive()
|
|
defer h.cmdChan.CloseChanReceive()
|
|
|
|
for {
|
|
select {
|
|
case cmd := <-cmdChan:
|
|
cmd()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (h *ReadyToExecuteHandler) Stop() {
|
|
// TODO 支持STOP
|
|
}
|