GetCluster should take a context

Signed-off-by: Vince Prignano <[email protected]>

Author: vincepri

examples/fleet/main.go

@@ -63,7 +63,7 @@ func main() {
 		func(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
 			log := log.FromContext(ctx)
 
-			cluster, err := mgr.GetCluster(req.Cluster)
+			cluster, err := mgr.GetCluster(ctx, req.Cluster)
 			if err != nil {
 				return reconcile.Result{}, err
 			}

pkg/builder/controller_test.go

@@ -623,7 +623,7 @@ var _ = Describe("application", func() {
 					},
 				},
 			}
-			cluster1, err := mgr.GetCluster("cluster1")
+			cluster1, err := mgr.GetCluster(ctx, "cluster1")
 			Expect(err).NotTo(HaveOccurred())
 			Expect(cluster1.GetClient().Create(ctx, dep)).To(Succeed())
 

pkg/cluster/cluster.go

@@ -39,7 +39,7 @@ import (
 )
 
 // LogicalGetterFunc is a function that returns a cluster for a given logical cluster name.
-type LogicalGetterFunc func(logical.Name) (Cluster, error)
+type LogicalGetterFunc func(context.Context, logical.Name) (Cluster, error)
 
 // Cluster provides various methods to interact with a cluster.
 type Cluster interface {

pkg/manager/internal.go

@@ -258,8 +258,8 @@ func (cm *controllerManager) AddReadyzCheck(name string, check healthz.Checker)
 	return nil
 }
 
-func (cm *controllerManager) GetCluster(name logical.Name) (cluster.Cluster, error) {
-	return cm.getLogicalCluster(context.TODO(), name)
+func (cm *controllerManager) GetCluster(ctx context.Context, name logical.Name) (cluster.Cluster, error) {
+	return cm.getLogicalCluster(ctx, name)
 }
 
 func (cm *controllerManager) GetHTTPClient() *http.Client {
@@ -365,13 +365,14 @@ func (cm *controllerManager) getLogicalCluster(ctx context.Context, name logical
 	// Create a new cluster.
 	var cfg *rest.Config
 	{
+		// TODO(vincepri): Make this timeout configurable.
 		ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
 		defer cancel()
 		var watchErr error
 		if err := wait.PollImmediateUntilWithContext(ctx, 1*time.Second, func(ctx context.Context) (done bool, err error) {
 			cfg, watchErr = cm.logicalAdapter.RESTConfig(name)
 			if watchErr != nil {
-				return false, nil // retry
+				return false, nil //nolint:nilerr // We want to keep trying.
 			}
 			return true, nil
 		}); err != nil {

pkg/manager/manager.go

@@ -90,7 +90,7 @@ type Manager interface {
 	Start(ctx context.Context) error
 
 	// GetCluster retrieves a Cluster from a given logical name.
-	GetCluster(logical.Name) (cluster.Cluster, error)
+	GetCluster(context.Context, logical.Name) (cluster.Cluster, error)
 
 	// GetWebhookServer returns a webhook.Server
 	GetWebhookServer() *webhook.Server