diff options
| -rw-r--r-- | cmd/bookpipeline/main.go | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/cmd/bookpipeline/main.go b/cmd/bookpipeline/main.go index f4b1ad4..032f614 100644 --- a/cmd/bookpipeline/main.go +++ b/cmd/bookpipeline/main.go @@ -42,6 +42,7 @@ one is found this general process is followed: ` const PauseBetweenChecks = 3 * time.Minute +const TimeBeforeShutdown = 5 * time.Minute const HeartbeatTime = 60 // null writer to enable non-verbose logging to be discarded @@ -671,6 +672,7 @@ func main() { var checkOCRQueue <-chan time.Time var checkOCRPageQueue <-chan time.Time var checkAnalyseQueue <-chan time.Time + var shutdownIfQuiet <-chan time.Time if !*nopreproc { checkPreQueue = time.After(0) } @@ -705,6 +707,7 @@ func main() { if err != nil { log.Println("Error during preprocess", err) } + shutdownIfQuiet = time.After(TimeBeforeShutdown) case <-checkWipeQueue: msg, err := conn.CheckQueue(conn.WipeQueueId(), HeartbeatTime*2) checkWipeQueue = time.After(PauseBetweenChecks) @@ -721,6 +724,7 @@ func main() { if err != nil { log.Println("Error during wipe", err) } + shutdownIfQuiet = time.After(TimeBeforeShutdown) case <-checkOCRPageQueue: msg, err := conn.CheckQueue(conn.OCRPageQueueId(), HeartbeatTime*2) checkOCRPageQueue = time.After(PauseBetweenChecks) @@ -739,6 +743,7 @@ func main() { if err != nil { log.Println("Error during OCR Page process", err) } + shutdownIfQuiet = time.After(TimeBeforeShutdown) case <-checkOCRQueue: msg, err := conn.CheckQueue(conn.OCRQueueId(), HeartbeatTime*2) checkOCRQueue = time.After(PauseBetweenChecks) @@ -755,6 +760,7 @@ func main() { if err != nil { log.Println("Error during OCR process", err) } + shutdownIfQuiet = time.After(TimeBeforeShutdown) case <-checkAnalyseQueue: msg, err := conn.CheckQueue(conn.AnalyseQueueId(), HeartbeatTime*2) checkAnalyseQueue = time.After(PauseBetweenChecks) @@ -771,6 +777,9 @@ func main() { if err != nil { log.Println("Error during analysis", err) } + shutdownIfQuiet = time.After(TimeBeforeShutdown) + case <-shutdownIfQuiet: + log.Println("If I was sufficiently brave, now would be the time I would shut down") } } } |