go-judge/runner/runner.go
2019-10-14 00:34:48 -07:00

60 lines
1.0 KiB
Go

package runner
import (
"sync"
"github.com/criyle/go-judge/language"
"github.com/criyle/go-judge/taskqueue"
"github.com/criyle/go-sandbox/daemon"
)
// DaemonBuilder defines the abstract builder for daemon
type DaemonBuilder interface {
Build() (*daemon.Master, error)
}
// Runner is the task runner
type Runner struct {
// Queue is the message queue to receive run tasks
Queue taskqueue.Receiver
// Builder builds the sandbox runner
Builder DaemonBuilder
Language language.Language
// pool of sandbox to use
pool *pool
once sync.Once
}
func (r *Runner) init() {
r.pool = newPool(r.Builder)
}
// Loop status a runner in a forever loop, waiting for task and execute
// call it in new goroutine
func (r *Runner) Loop(done <-chan struct{}) error {
r.once.Do(r.init)
c := r.Queue.ReceiveC()
loop:
for {
select {
case <-done:
break loop
case task := <-c:
task.Done(r.run(done, task.Task()))
}
// check if cancel is signaled
select {
case <-done:
break loop
default:
}
}
return nil
}