mirror of
https://github.com/dreamstarsky/runbin.git
synced 2026-05-15 22:33:09 +00:00
first demo
This commit is contained in:
xkm
152
internal/worker/RunCppTask.go
Normal file
152
internal/worker/RunCppTask.go
Normal file
@@ -0,0 +1,152 @@
|
||||
package worker
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"time"
|
||||
|
||||
"runbin/internal/model"
|
||||
|
||||
"github.com/docker/docker/api/types/container"
|
||||
"github.com/docker/docker/client"
|
||||
)
|
||||
|
||||
func (w *Worker) RunCppTask(ctx context.Context, task *model.Paste) error {
|
||||
// Create temporary workspace
|
||||
tmpDir, err := os.MkdirTemp("", "cpp_compile_")
|
||||
if err != nil {
|
||||
return fmt.Errorf("create temp dir error: %v", err)
|
||||
}
|
||||
defer os.RemoveAll(tmpDir)
|
||||
|
||||
// Write user code to main.cpp
|
||||
codePath := filepath.Join(tmpDir, "main.cpp")
|
||||
if err := os.WriteFile(codePath, []byte(task.Code), 0644); err != nil {
|
||||
return fmt.Errorf("write code file error: %v", err)
|
||||
}
|
||||
|
||||
// Get execution limits from config
|
||||
timeout := time.Duration(w.cfg.Limit.Time) * time.Second
|
||||
ctx, cancel := context.WithTimeout(ctx, timeout)
|
||||
defer cancel()
|
||||
|
||||
// Initialize Docker client
|
||||
cli, err := client.NewClientWithOpts(client.FromEnv)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to create docker client: %v", err)
|
||||
}
|
||||
|
||||
hostConfig := &container.HostConfig{
|
||||
Binds: []string{tmpDir + ":/app"},
|
||||
Resources: container.Resources{
|
||||
Memory: int64(w.cfg.Limit.Memory * 1024 * 1024),
|
||||
CPUQuota: int64(w.cfg.Limit.Cpu * 100000),
|
||||
},
|
||||
AutoRemove: true,
|
||||
NetworkMode: "none",
|
||||
}
|
||||
|
||||
// Create compile container configuration
|
||||
resp, err := cli.ContainerCreate(ctx, &container.Config{
|
||||
Image: "gcc:14",
|
||||
Cmd: []string{"sh", "-c", "g++ -std=c++20 /app/main.cpp -o /app/output > /app/compile.txt 2>&1"},
|
||||
}, hostConfig, nil, nil, filepath.Base(tmpDir)+"_builder")
|
||||
if err != nil {
|
||||
return fmt.Errorf("create compile container error: %v", err)
|
||||
}
|
||||
|
||||
// Start compile container execution
|
||||
if err := cli.ContainerStart(ctx, resp.ID, container.StartOptions{}); err != nil {
|
||||
return fmt.Errorf("failed to start compile container: %v", err)
|
||||
}
|
||||
|
||||
// Wait for compile container completion
|
||||
statusCh, errCh := cli.ContainerWait(ctx, resp.ID, container.WaitConditionNotRunning)
|
||||
|
||||
// Handle compile container execution results
|
||||
select {
|
||||
case <-statusCh:
|
||||
// Get compile container detailed state
|
||||
containerState, err := cli.ContainerInspect(ctx, resp.ID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("container state inspection error: %v", err)
|
||||
}
|
||||
|
||||
// Read compilation log
|
||||
if logData, err := os.ReadFile(filepath.Join(tmpDir, "compile.txt")); err == nil {
|
||||
task.CompileLog = string(logData)
|
||||
}
|
||||
|
||||
// Non-zero exit code indicates compilation failure
|
||||
if containerState.State.ExitCode != 0 {
|
||||
task.Status = model.StatusCompileError
|
||||
return nil
|
||||
}
|
||||
case err := <-errCh:
|
||||
return err
|
||||
case <-ctx.Done():
|
||||
task.Status = model.StatusCompileError
|
||||
task.CompileLog = "Compile process exceeded time limit"
|
||||
return nil
|
||||
}
|
||||
|
||||
// Write input to input.txt
|
||||
inputPath := filepath.Join(tmpDir, "input.txt")
|
||||
if err := os.WriteFile(inputPath, []byte(task.Stdin), 0644); err != nil {
|
||||
return fmt.Errorf("write input file error: %v", err)
|
||||
}
|
||||
|
||||
// Create runner container configuration
|
||||
resp, err = cli.ContainerCreate(ctx, &container.Config{
|
||||
Image: "gcc:14",
|
||||
Cmd: []string{"sh", "-c", "sh -c \"/app/output < /app/input.txt > /app/stdout.txt\" > /app/stderr.txt 2>&1"},
|
||||
}, hostConfig, nil, nil, filepath.Base(tmpDir)+"_runner")
|
||||
if err != nil {
|
||||
return fmt.Errorf("create runner container error: %v", err)
|
||||
}
|
||||
|
||||
// Start runner container execution
|
||||
if err := cli.ContainerStart(ctx, resp.ID, container.StartOptions{}); err != nil {
|
||||
return fmt.Errorf("failed to start runner container: %v", err)
|
||||
}
|
||||
|
||||
// Wait for container completion
|
||||
statusCh, errCh = cli.ContainerWait(ctx, resp.ID, container.WaitConditionNotRunning)
|
||||
|
||||
// Handle container execution results
|
||||
select {
|
||||
case <-statusCh:
|
||||
// Get container detailed state
|
||||
containerState, err := cli.ContainerInspect(ctx, resp.ID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("container state inspection error: %v", err)
|
||||
}
|
||||
|
||||
// Non-zero exit code indicates compilation failure
|
||||
if containerState.State.ExitCode != 0 {
|
||||
task.Status = model.StatusRuntimeError
|
||||
} else {
|
||||
task.Status = model.StatusCompleted
|
||||
}
|
||||
case err := <-errCh:
|
||||
return err
|
||||
case <-ctx.Done():
|
||||
task.Status = model.StatusTimeLimitExceed
|
||||
}
|
||||
|
||||
// Process execution results by reading output files
|
||||
stdoutPath := filepath.Join(tmpDir, "stdout.txt")
|
||||
stderrPath := filepath.Join(tmpDir, "stderr.txt")
|
||||
|
||||
// Read program output
|
||||
if outData, err := os.ReadFile(stdoutPath); err == nil {
|
||||
task.Stdout = string(outData)
|
||||
}
|
||||
if outData, err := os.ReadFile(stderrPath); err == nil {
|
||||
task.Stderr = string(outData)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
89
internal/worker/worker.go
Normal file
89
internal/worker/worker.go
Normal file
@@ -0,0 +1,89 @@
|
||||
package worker
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"time"
|
||||
|
||||
"runbin/internal/config"
|
||||
"runbin/internal/model"
|
||||
"runbin/internal/repository"
|
||||
)
|
||||
|
||||
type Worker struct {
|
||||
repo repository.PasteRepository
|
||||
cfg *config.WorkerConfig
|
||||
}
|
||||
|
||||
func NewWorker(repo repository.PasteRepository, cfg *config.WorkerConfig) *Worker {
|
||||
return &Worker{
|
||||
repo: repo,
|
||||
cfg: cfg,
|
||||
}
|
||||
}
|
||||
|
||||
func (w *Worker) Run() {
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
// run n process
|
||||
for range w.cfg.Process {
|
||||
go w.processTasks(ctx)
|
||||
}
|
||||
|
||||
<-ctx.Done()
|
||||
}
|
||||
|
||||
func (w *Worker) processTasks(ctx context.Context) {
|
||||
ticker := time.NewTicker(1 * time.Second)
|
||||
defer ticker.Stop()
|
||||
|
||||
log.Println("Thread start!")
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-ticker.C:
|
||||
if task, err := w.repo.GetTask(ctx); err == nil {
|
||||
if task == nil {
|
||||
continue
|
||||
}
|
||||
if err := w.handleTask(ctx, task); err != nil {
|
||||
log.Printf("Worker error at PasteID: %s, error: %v\n", task.ID, err)
|
||||
}
|
||||
if err := w.repo.Update(task); err != nil {
|
||||
log.Printf("Update error at PasteID: %s, error: %v\n", task.ID, err)
|
||||
}
|
||||
log.Printf("Judged task %s", task.ID)
|
||||
} else {
|
||||
log.Printf("Worker get task error: %v\n", err)
|
||||
}
|
||||
case <-ctx.Done():
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (w *Worker) handleTask(ctx context.Context, task *model.Paste) error {
|
||||
log.Printf("Hangling task %s for language %s", task.ID, task.Language)
|
||||
|
||||
task.Status = model.StatusRunning
|
||||
task.BackEnd = w.cfg.Name
|
||||
w.repo.Update(task)
|
||||
|
||||
|
||||
var err error
|
||||
|
||||
switch task.Language {
|
||||
case "c++20":
|
||||
err = w.RunCppTask(ctx, task)
|
||||
default:
|
||||
err = fmt.Errorf("Unsupported language '%s'", task.Language)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
task.Status = model.StatusUnknownError
|
||||
task.CompileLog = err.Error()
|
||||
}
|
||||
return err
|
||||
}
|
||||
Reference in New Issue
Block a user