Merge pull request #436 from getAlby/task-timeout

feat: introduce timeout in cluster status check
getAlby · Oct 3, 2023 · 5880dfe · 5880dfe
2 parents d2d9b03 + 0d11c9a
commit 5880dfe
Showing 1 changed file with 4 additions and 2 deletions.
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)
 		}
 	}
 }