X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/df591042778408d03d410d5c22a669d85652d1ea..79bce4a71a58118a9003882e0ca9bbfb9d2957a9:/services/crunch-dispatch-slurm/slurm.go diff --git a/services/crunch-dispatch-slurm/slurm.go b/services/crunch-dispatch-slurm/slurm.go index 9e9f45270f..791f294df1 100644 --- a/services/crunch-dispatch-slurm/slurm.go +++ b/services/crunch-dispatch-slurm/slurm.go @@ -20,7 +20,15 @@ type Slurm interface { Renice(name string, nice int64) error } -type slurmCLI struct{} +type slurmCLI struct { + runSemaphore chan bool +} + +func NewSlurmCLI() *slurmCLI { + return &slurmCLI{ + runSemaphore: make(chan bool, 3), + } +} func (scli *slurmCLI) Batch(script io.Reader, args []string) error { return scli.run(script, "sbatch", args) @@ -64,6 +72,8 @@ func (scli *slurmCLI) Renice(name string, nice int64) error { } func (scli *slurmCLI) run(stdin io.Reader, prog string, args []string) error { + scli.runSemaphore <- true + defer func() { <-scli.runSemaphore }() cmd := exec.Command(prog, args...) cmd.Stdin = stdin out, err := cmd.CombinedOutput()