Merge pull request #436 from getAlby/task-timeout

kiwiidb · web-flow · commit 5880dfe27f1d · 2023-10-03T12:00:33.000+02:00
feat: introduce timeout in cluster status check
diff --git a/lnd/lnd_cluster.go b/lnd/lnd_cluster.go
@@ -27,8 +27,10 @@ func (cluster *LNDCluster) StartLivenessLoop(ctx context.Context) {
 		case <-ctx.Done():
 			return
 		case <-ticker.C:
-			cluster.Logger.Info("Checking cluster status")
-			cluster.checkClusterStatus(ctx)
+			cluster.Logger.Info("Checking cluster status")	
+			checkCtx, cancel := context.WithTimeout(ctx, 30*time.Second)
+			defer cancel()
+			cluster.checkClusterStatus(checkCtx)
 		}
 	}
 }

Original file line number	Diff line number	Diff line change
`@@ -27,8 +27,10 @@ func (cluster *LNDCluster) StartLivenessLoop(ctx context.Context) {`
`27`	`27`	`case <-ctx.Done():`
`28`	`28`	`return`
`29`	`29`	`case <-ticker.C:`
`30`		`- cluster.Logger.Info("Checking cluster status")`
`31`		`- cluster.checkClusterStatus(ctx)`
	`30`	`+ cluster.Logger.Info("Checking cluster status")`
	`31`	`+ checkCtx, cancel := context.WithTimeout(ctx, 30*time.Second)`
	`32`	`+ defer cancel()`
	`33`	`+ cluster.checkClusterStatus(checkCtx)`
`32`	`34`	`}`
`33`	`35`	`}`
`34`	`36`	`}`