go-judge/runner/runner.go
2019-09-29 15:19:36 -07:00

55 lines
913 B
Go

package runner
import (
"sync"
"github.com/criyle/go-judge/language"
"github.com/criyle/go-judge/taskqueue"
"github.com/criyle/go-sandbox/daemon"
)
// 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 *daemon.Builder
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
}