projects
/
arvados.git
/ blobdiff
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
10703: Do not catch signals in crunch-dispatch-slurm. Simplify "stop dispatcher loop...
[arvados.git]
/
services
/
crunch-dispatch-local
/
crunch-dispatch-local.go
diff --git
a/services/crunch-dispatch-local/crunch-dispatch-local.go
b/services/crunch-dispatch-local/crunch-dispatch-local.go
index 0ca765185119c152dd11870641c15f905042311e..cfb0c7da947186d74b1a3526eb9a3b08d9d58bf5 100644
(file)
--- a/
services/crunch-dispatch-local/crunch-dispatch-local.go
+++ b/
services/crunch-dispatch-local/crunch-dispatch-local.go
@@
-10,7
+10,9
@@
import (
"log"
"os"
"os/exec"
"log"
"os"
"os/exec"
+ "os/signal"
"sync"
"sync"
+ "syscall"
"time"
)
"time"
)
@@
-54,16
+56,24
@@
func doMain() error {
arv.Retries = 25
dispatcher := dispatch.Dispatcher{
arv.Retries = 25
dispatcher := dispatch.Dispatcher{
- Arv:
arv,
- RunContainer:
run,
- PollInterval:
time.Duration(*pollInterval) * time.Second,
-
DoneProcessing: make(chan struct{})
}
+ Arv: arv,
+ RunContainer: run,
+ PollInterval: time.Duration(*pollInterval) * time.Second,
+ }
- err = dispatcher.Run
Dispatcher
()
+ err = dispatcher.Run()
if err != nil {
return err
}
if err != nil {
return err
}
+ c := make(chan os.Signal, 1)
+ signal.Notify(c, os.Interrupt, syscall.SIGTERM, syscall.SIGQUIT)
+ sig := <-c
+ log.Printf("Received %s, shutting down", sig)
+ signal.Stop(c)
+
+ dispatcher.Stop()
+
runningCmdsMutex.Lock()
// Finished dispatching; interrupt any crunch jobs that are still running
for _, cmd := range runningCmds {
runningCmdsMutex.Lock()
// Finished dispatching; interrupt any crunch jobs that are still running
for _, cmd := range runningCmds {