add global timeout

2 files changed, 6 insertions, 4 deletions

diff --git a/internal/execute.go b/internal/execute.go
index d8f426f..1029ff7 100644
--- a/internal/execute.go
+++ b/internal/execute.go
@@ -7,7 +7,7 @@ import (
 	"time"
 )
 
-func execute(state state, config config) state {
+func execute(globalCtx context.Context, state state, config config) state {
 	limiterCh := make(chan struct{}, config.CheckConcurrency)
 	inputCh := make(chan namedCheck)
 	outputCh := make(chan checkResult)
@@ -40,7 +40,7 @@ func execute(state state, config config) state {
 				inputWg.Done()
 			}()
 
-			ctx, cancel := context.WithTimeout(context.Background(),
+			ctx, cancel := context.WithTimeout(globalCtx,
 				time.Duration(config.CheckTimeoutS)*time.Second)
 			defer cancel()
 
diff --git a/internal/run.go b/internal/run.go
index 417b81b..ef9919d 100644
--- a/internal/run.go
+++ b/internal/run.go
@@ -1,6 +1,8 @@
 package internal
 
-func Run(configFile string) {
+import "context"
+
+func Run(cts context.Context, configFile string) {
 	config, err := newConfig(configFile)
 	if err != nil {
 		panic(err)
@@ -11,7 +13,7 @@ func Run(configFile string) {
 		notifyError(config, err)
 	}
 
-	state = execute(state, config)
+	state = execute(ctx, state, config)
 
 	if err := state.persist(); err != nil {
 		notifyError(config, err)