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
|
|
|
|
EnsureDeps(force bool) e.Status
|
|
|
|
// NewProblem = Download + Parse + Prebuild
|
2022-10-23 17:29:35 +08:00
|
|
|
NewProblem(version uint, url string, force bool) (Config, e.Status)
|
2022-10-22 17:38:39 +08:00
|
|
|
|
|
|
|
// Compile compile user submission
|
|
|
|
Compile(version uint, user string, lang string) (JudgeStatus, e.Status)
|
|
|
|
// RunAndJudge execute user program
|
|
|
|
RunAndJudge(version uint, user string, lang string, config *Config) (JudgeStatus, int32, e.Status)
|
|
|
|
|
|
|
|
// ParseConfig parse config file
|
|
|
|
ParseConfig(version uint, skipCheck bool) (Config, error)
|
2022-10-23 17:29:35 +08:00
|
|
|
// ProblemExists check if problem exists
|
|
|
|
ProblemExists(version uint) bool
|
2023-07-15 16:19:49 +08:00
|
|
|
|
|
|
|
HealthCheck() error
|
|
|
|
}
|
|
|
|
|
|
|
|
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)
|
|
|
|
|
2023-07-15 16:19:49 +08:00
|
|
|
return &service{
|
|
|
|
log: do.MustInvoke[log.Service](i).GetLogger("runner"),
|
2024-01-06 15:06:12 +08:00
|
|
|
pool: pool.NewTaskPool(concurrency, concurrency),
|
2023-07-15 16:19:49 +08:00
|
|
|
verbose: do.MustInvoke[config.Service](i).GetConfig().Development,
|
|
|
|
}, nil
|
2022-10-22 17:38:39 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
type service struct {
|
2022-10-28 21:57:28 +08:00
|
|
|
log *zap.Logger
|
2024-01-06 15:06:12 +08:00
|
|
|
pool *pool.TaskPool
|
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
|
|
|
}
|