From da739a0c3d0328358dd17332920a1e1f1f9faa80 Mon Sep 17 00:00:00 2001 From: Quentin McGaw Date: Thu, 7 May 2020 12:58:37 +0000 Subject: [PATCH] Fix waitForAll context for graceful exits --- cmd/main.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/cmd/main.go b/cmd/main.go index 53675864..f62b3e62 100644 --- a/cmd/main.go +++ b/cmd/main.go @@ -275,8 +275,9 @@ func main() { logger.Error(err) } } - errors := waiter.WaitForAll(ctx) - for _, err := range errors { + timeoutCtx, cancel := context.WithTimeout(context.Background(), time.Second) + defer cancel() + for _, err := range waiter.WaitForAll(timeoutCtx) { logger.Error(err) } }