mirror of
https://github.com/criyle/go-judge.git
synced 2025-11-04 14:50:02 +08:00
71 lines
1.3 KiB
Go
71 lines
1.3 KiB
Go
package runner
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/criyle/go-judge/language"
|
|
"github.com/criyle/go-judge/taskqueue"
|
|
"github.com/criyle/go-sandbox/container"
|
|
"github.com/criyle/go-sandbox/pkg/cgroup"
|
|
)
|
|
|
|
// EnvironmentBuilder defines the abstract builder for container environment
|
|
type EnvironmentBuilder interface {
|
|
Build() (container.Environment, error)
|
|
}
|
|
|
|
// CgroupBuilder builds cgroup for runner
|
|
type CgroupBuilder interface {
|
|
Build() (cg *cgroup.CGroup, err 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 EnvironmentBuilder
|
|
|
|
// Builder for cgroups
|
|
CgroupBuilder CgroupBuilder
|
|
|
|
Language language.Language
|
|
|
|
// pool of sandbox to use
|
|
pool *pool
|
|
|
|
// ensure init / shutdown only once
|
|
onceInit, onceShutdown 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{}) {
|
|
r.onceInit.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:
|
|
}
|
|
}
|
|
r.onceShutdown.Do(r.pool.Shutdown)
|
|
}
|