diff options
Diffstat (limited to 'cmd')
| -rw-r--r-- | cmd/dtail/main.go | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/cmd/dtail/main.go b/cmd/dtail/main.go index b8bcd06..3cec667 100644 --- a/cmd/dtail/main.go +++ b/cmd/dtail/main.go @@ -71,9 +71,12 @@ func main() { version.PrintAndExit() } - ctx, _ := context.WithCancel(context.Background()) + ctx, cancel := context.WithCancel(context.Background()) + defer cancel() + if shutdownAfter > 0 { - ctx, _ = context.WithTimeout(ctx, time.Duration(shutdownAfter)*time.Second) + ctx, cancel = context.WithTimeout(ctx, time.Duration(shutdownAfter)*time.Second) + defer cancel() } if checkHealth { |