2022-10-22 17:38:39 +08:00
|
|
|
package runner
|
|
|
|
|
|
|
|
import (
|
2023-07-14 21:47:11 +08:00
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/e"
|
2023-07-15 16:19:49 +08:00
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/misc/config"
|
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/misc/log"
|
2024-01-06 15:06:12 +08:00
|
|
|
"git.0x7f.app/WOJ/woj-server/pkg/pool"
|
|
|
|
"git.0x7f.app/WOJ/woj-server/pkg/utils"
|
2023-07-15 16:19:49 +08:00
|
|
|
"github.com/samber/do"
|
2022-10-22 17:38:39 +08:00
|
|
|
"go.uber.org/zap"
|
2024-01-06 15:06:12 +08:00
|
|
|
"runtime"
|
2022-10-22 17:38:39 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
var _ Service = (*service)(nil)
|
|
|
|
|
|
|
|
type Service interface {
|
|
|
|
// EnsureDeps build docker images
|
2024-01-28 21:51:16 +08:00
|
|
|
EnsureDeps() e.Status
|
2022-10-22 17:38:39 +08:00
|
|
|
// NewProblem = Download + Parse + Prebuild
|
2024-01-27 23:07:14 +08:00
|
|
|
NewProblem(meta *JudgeMeta, url string, force bool) (*Config, e.Status)
|
2022-10-22 17:38:39 +08:00
|
|
|
|
|
|
|
// Compile compile user submission
|
2024-01-27 23:07:14 +08:00
|
|
|
Compile(meta *JudgeMeta) (*JudgeStatus, e.Status)
|
2022-10-22 17:38:39 +08:00
|
|
|
// RunAndJudge execute user program
|
2024-01-27 23:07:14 +08:00
|
|
|
RunAndJudge(meta *JudgeMeta) (*JudgeStatus, int32, e.Status)
|
2022-10-22 17:38:39 +08:00
|
|
|
|
|
|
|
// ParseConfig parse config file
|
2024-01-27 23:07:14 +08:00
|
|
|
ParseConfig(meta *JudgeMeta, skipCheck bool) (*Config, error)
|
2022-10-23 17:29:35 +08:00
|
|
|
// ProblemExists check if problem exists
|
2024-01-27 23:07:14 +08:00
|
|
|
ProblemExists(meta *JudgeMeta) bool
|
2024-01-29 21:15:39 +08:00
|
|
|
// ValidatePath check if problem/user exists
|
|
|
|
ValidatePath(meta *JudgeMeta) (*JudgeStatus, e.Status)
|
|
|
|
// GetConfig return config and filter language config, will fill result to meta.Config
|
|
|
|
GetConfig(meta *JudgeMeta, skipCheck bool) e.Status
|
2023-07-15 16:19:49 +08:00
|
|
|
|
|
|
|
HealthCheck() error
|
2024-01-06 19:21:37 +08:00
|
|
|
Shutdown() error
|
2023-07-15 16:19:49 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func NewService(i *do.Injector) (Service, error) {
|
2024-01-06 15:06:12 +08:00
|
|
|
concurrency := utils.If(runtime.NumCPU() > 1, runtime.NumCPU()-1, 1)
|
2024-01-27 17:37:27 +08:00
|
|
|
cfg := do.MustInvoke[config.Service](i).GetConfig()
|
2024-01-06 15:06:12 +08:00
|
|
|
|
2024-01-06 19:21:37 +08:00
|
|
|
srv := &service{
|
2023-07-15 16:19:49 +08:00
|
|
|
log: do.MustInvoke[log.Service](i).GetLogger("runner"),
|
2024-01-06 15:06:12 +08:00
|
|
|
pool: pool.NewTaskPool(concurrency, concurrency),
|
2024-03-13 20:03:12 +08:00
|
|
|
cgroup: cfg.Runner.CGroup,
|
2024-01-27 17:37:27 +08:00
|
|
|
verbose: cfg.Development,
|
2024-01-06 19:21:37 +08:00
|
|
|
}
|
|
|
|
|
2024-01-27 17:37:27 +08:00
|
|
|
srv.pool.Start()
|
2024-01-06 19:21:37 +08:00
|
|
|
return srv, nil
|
2022-10-22 17:38:39 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
type service struct {
|
2024-03-13 20:03:12 +08:00
|
|
|
log *zap.Logger
|
|
|
|
pool *pool.TaskPool
|
|
|
|
cgroup string
|
2022-10-28 21:57:28 +08:00
|
|
|
verbose bool
|
2022-10-22 17:38:39 +08:00
|
|
|
}
|
|
|
|
|
2023-07-15 16:19:49 +08:00
|
|
|
func (s *service) HealthCheck() error {
|
|
|
|
return nil
|
2022-10-22 17:38:39 +08:00
|
|
|
}
|
2024-01-06 19:21:37 +08:00
|
|
|
|
|
|
|
func (s *service) Shutdown() error {
|
|
|
|
s.pool.Stop()
|
|
|
|
return nil
|
|
|
|
}
|