Skip to content

Commit a33241c

Browse files
vinceprivincepri
committed
⚠️ Propagate context.Context throughout the codebase
Signed-off-by: Vince Prignano <[email protected]>
1 parent 5757a38 commit a33241c

37 files changed

+433
-429
lines changed

‎designs/move-cluster-specific-code-out-of-manager.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ func NewSecretMirrorReconciler(mgr manager.Manager, mirrorCluster cluster.Cluste
203203

204204
func main(){
205205

206-
mgr, err := manager.New(cfg1, manager.Options{})
206+
mgr, err := manager.New(context.Background(), cfg1, manager.Options{})
207207
if err != nil {
208208
panic(err)
209209
}

‎example_test.go

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ import (
3838
func Example() {
3939
var log = controllers.Log.WithName("builder-examples")
4040

41-
manager, err := controllers.NewManager(controllers.GetConfigOrDie(), controllers.Options{})
41+
manager, err := controllers.NewManager(context.Background(), controllers.GetConfigOrDie(), controllers.Options{})
4242
if err != nil {
4343
log.Error(err, "could not create manager")
4444
os.Exit(1)
@@ -78,11 +78,14 @@ func Example_updateLeaderElectionDurations() {
7878
leaseDuration := 100 * time.Second
7979
renewDeadline := 80 * time.Second
8080
retryPeriod := 20 * time.Second
81-
manager, err := controllers.NewManager(controllers.GetConfigOrDie(), controllers.Options{
82-
LeaseDuration: &leaseDuration,
83-
RenewDeadline: &renewDeadline,
84-
RetryPeriod: &retryPeriod,
85-
})
81+
manager, err := controllers.NewManager(
82+
context.Background(),
83+
controllers.GetConfigOrDie(),
84+
controllers.Options{
85+
LeaseDuration: &leaseDuration,
86+
RenewDeadline: &renewDeadline,
87+
RetryPeriod: &retryPeriod,
88+
})
8689
if err != nil {
8790
log.Error(err, "could not create manager")
8891
os.Exit(1)

‎examples/builtins/main.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ limitations under the License.
1717
package main
1818

1919
import (
20+
"context"
2021
"os"
2122

2223
appsv1 "k8s.io/api/apps/v1"
@@ -42,7 +43,7 @@ func main() {
4243

4344
// Setup a Manager
4445
entryLog.Info("setting up manager")
45-
mgr, err := manager.New(config.GetConfigOrDie(), manager.Options{})
46+
mgr, err := manager.New(context.Background(), config.GetConfigOrDie(), manager.Options{})
4647
if err != nil {
4748
entryLog.Error(err, "unable to set up overall controller manager")
4849
os.Exit(1)

‎examples/crd/main.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ func (r *reconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Resu
104104
func main() {
105105
ctrl.SetLogger(zap.New())
106106

107-
mgr, err := ctrl.NewManager(ctrl.GetConfigOrDie(), ctrl.Options{})
107+
mgr, err := ctrl.NewManager(context.Background(), ctrl.GetConfigOrDie(), ctrl.Options{})
108108
if err != nil {
109109
setupLog.Error(err, "unable to start manager")
110110
os.Exit(1)

‎pkg/builder/controller_test.go

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ var _ = Describe("application", func() {
7979
Describe("New", func() {
8080
It("should return success if given valid objects", func() {
8181
By("creating a controller manager")
82-
m, err := manager.New(cfg, manager.Options{})
82+
m, err := manager.New(context.Background(), cfg, manager.Options{})
8383
Expect(err).NotTo(HaveOccurred())
8484

8585
instance, err := ControllerManagedBy(m).
@@ -92,7 +92,7 @@ var _ = Describe("application", func() {
9292

9393
It("should return an error if there is no GVK for an object, and thus we can't default the controller name", func() {
9494
By("creating a controller manager")
95-
m, err := manager.New(cfg, manager.Options{})
95+
m, err := manager.New(context.Background(), cfg, manager.Options{})
9696
Expect(err).NotTo(HaveOccurred())
9797

9898
By("creating a controller with a bad For type")
@@ -115,7 +115,7 @@ var _ = Describe("application", func() {
115115
}
116116

117117
By("creating a controller manager")
118-
m, err := manager.New(cfg, manager.Options{})
118+
m, err := manager.New(context.Background(), cfg, manager.Options{})
119119
Expect(err).NotTo(HaveOccurred())
120120

121121
instance, err := ControllerManagedBy(m).
@@ -138,7 +138,7 @@ var _ = Describe("application", func() {
138138
}
139139

140140
By("creating a controller manager")
141-
m, err := manager.New(cfg, manager.Options{})
141+
m, err := manager.New(context.Background(), cfg, manager.Options{})
142142
Expect(err).NotTo(HaveOccurred())
143143

144144
instance, err := ControllerManagedBy(m).
@@ -160,7 +160,7 @@ var _ = Describe("application", func() {
160160
}
161161

162162
By("creating a controller manager")
163-
m, err := manager.New(cfg, manager.Options{})
163+
m, err := manager.New(context.Background(), cfg, manager.Options{})
164164
Expect(err).NotTo(HaveOccurred())
165165

166166
instance, err := ControllerManagedBy(m).
@@ -183,7 +183,7 @@ var _ = Describe("application", func() {
183183
}
184184

185185
By("creating a controller manager")
186-
m, err := manager.New(cfg, manager.Options{})
186+
m, err := manager.New(context.Background(), cfg, manager.Options{})
187187
Expect(err).NotTo(HaveOccurred())
188188

189189
instance, err := ControllerManagedBy(m).
@@ -204,7 +204,7 @@ var _ = Describe("application", func() {
204204
}
205205

206206
By("creating a controller manager")
207-
m, err := manager.New(cfg, manager.Options{})
207+
m, err := manager.New(context.Background(), cfg, manager.Options{})
208208
Expect(err).NotTo(HaveOccurred())
209209

210210
instance, err := ControllerManagedBy(m).
@@ -218,7 +218,7 @@ var _ = Describe("application", func() {
218218

219219
It("should allow multiple controllers for the same kind", func() {
220220
By("creating a controller manager")
221-
m, err := manager.New(cfg, manager.Options{})
221+
m, err := manager.New(context.Background(), cfg, manager.Options{})
222222
Expect(err).NotTo(HaveOccurred())
223223

224224
By("registering the type in the Scheme")
@@ -247,7 +247,7 @@ var _ = Describe("application", func() {
247247

248248
Describe("Start with ControllerManagedBy", func() {
249249
It("should Reconcile Owns objects", func(done Done) {
250-
m, err := manager.New(cfg, manager.Options{})
250+
m, err := manager.New(context.Background(), cfg, manager.Options{})
251251
Expect(err).NotTo(HaveOccurred())
252252

253253
bldr := ControllerManagedBy(m).
@@ -258,7 +258,7 @@ var _ = Describe("application", func() {
258258
}, 10)
259259

260260
It("should Reconcile Watches objects", func(done Done) {
261-
m, err := manager.New(cfg, manager.Options{})
261+
m, err := manager.New(context.Background(), cfg, manager.Options{})
262262
Expect(err).NotTo(HaveOccurred())
263263

264264
bldr := ControllerManagedBy(m).
@@ -273,7 +273,7 @@ var _ = Describe("application", func() {
273273

274274
Describe("Set custom predicates", func() {
275275
It("should execute registered predicates only for assigned kind", func(done Done) {
276-
m, err := manager.New(cfg, manager.Options{})
276+
m, err := manager.New(context.Background(), cfg, manager.Options{})
277277
Expect(err).NotTo(HaveOccurred())
278278

279279
var (

‎pkg/builder/example_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ func ExampleBuilder() {
4545

4646
var log = logf.Log.WithName("builder-examples")
4747

48-
mgr, err := manager.New(config.GetConfigOrDie(), manager.Options{})
48+
mgr, err := manager.New(context.Background(), config.GetConfigOrDie(), manager.Options{})
4949
if err != nil {
5050
log.Error(err, "could not create manager")
5151
os.Exit(1)

‎pkg/builder/example_webhook_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ limitations under the License.
1717
package builder_test
1818

1919
import (
20+
"context"
2021
"os"
2122

2223
"sigs.k8s.io/controller-runtime/pkg/builder"
@@ -39,7 +40,7 @@ var _ admission.Validator = &examplegroup.ChaosPod{}
3940
func ExampleWebhookBuilder() {
4041
var log = logf.Log.WithName("webhookbuilder-example")
4142

42-
mgr, err := manager.New(config.GetConfigOrDie(), manager.Options{})
43+
mgr, err := manager.New(context.Background(), config.GetConfigOrDie(), manager.Options{})
4344
if err != nil {
4445
log.Error(err, "could not create manager")
4546
os.Exit(1)

‎pkg/builder/webhook_test.go

Lines changed: 19 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ limitations under the License.
1717
package builder
1818

1919
import (
20+
"context"
2021
"errors"
2122
"fmt"
2223
"net/http"
@@ -49,7 +50,7 @@ var _ = Describe("webhook", func() {
4950
Describe("New", func() {
5051
It("should scaffold a defaulting webhook if the type implements the Defaulter interface", func() {
5152
By("creating a controller manager")
52-
m, err := manager.New(cfg, manager.Options{})
53+
m, err := manager.New(context.Background(), cfg, manager.Options{})
5354
Expect(err).NotTo(HaveOccurred())
5455

5556
By("registering the type in the Scheme")
@@ -89,11 +90,11 @@ var _ = Describe("webhook", func() {
8990
}
9091
}`)
9192

92-
stopCh := make(chan struct{})
93-
close(stopCh)
93+
ctx, cancel := context.WithCancel(context.Background())
94+
cancel()
9495
// TODO: we may want to improve it to make it be able to inject dependencies,
9596
// but not always try to load certs and return not found error.
96-
err = svr.Start(stopCh)
97+
err = svr.Start(ctx)
9798
if err != nil && !os.IsNotExist(err) {
9899
Expect(err).NotTo(HaveOccurred())
99100
}
@@ -121,7 +122,7 @@ var _ = Describe("webhook", func() {
121122

122123
It("should scaffold a validating webhook if the type implements the Validator interface", func() {
123124
By("creating a controller manager")
124-
m, err := manager.New(cfg, manager.Options{})
125+
m, err := manager.New(context.Background(), cfg, manager.Options{})
125126
Expect(err).NotTo(HaveOccurred())
126127

127128
By("registering the type in the Scheme")
@@ -163,11 +164,11 @@ var _ = Describe("webhook", func() {
163164
}
164165
}`)
165166

166-
stopCh := make(chan struct{})
167-
close(stopCh)
167+
ctx, cancel := context.WithCancel(context.Background())
168+
cancel()
168169
// TODO: we may want to improve it to make it be able to inject dependencies,
169170
// but not always try to load certs and return not found error.
170-
err = svr.Start(stopCh)
171+
err = svr.Start(ctx)
171172
if err != nil && !os.IsNotExist(err) {
172173
Expect(err).NotTo(HaveOccurred())
173174
}
@@ -194,7 +195,7 @@ var _ = Describe("webhook", func() {
194195

195196
It("should scaffold defaulting and validating webhooks if the type implements both Defaulter and Validator interfaces", func() {
196197
By("creating a controller manager")
197-
m, err := manager.New(cfg, manager.Options{})
198+
m, err := manager.New(context.Background(), cfg, manager.Options{})
198199
Expect(err).NotTo(HaveOccurred())
199200

200201
By("registering the type in the Scheme")
@@ -234,11 +235,11 @@ var _ = Describe("webhook", func() {
234235
}
235236
}`)
236237

237-
stopCh := make(chan struct{})
238-
close(stopCh)
238+
ctx, cancel := context.WithCancel(context.Background())
239+
cancel()
239240
// TODO: we may want to improve it to make it be able to inject dependencies,
240241
// but not always try to load certs and return not found error.
241-
err = svr.Start(stopCh)
242+
err = svr.Start(ctx)
242243
if err != nil && !os.IsNotExist(err) {
243244
Expect(err).NotTo(HaveOccurred())
244245
}
@@ -269,7 +270,9 @@ var _ = Describe("webhook", func() {
269270

270271
It("should scaffold a validating webhook if the type implements the Validator interface to validate deletes", func() {
271272
By("creating a controller manager")
272-
m, err := manager.New(cfg, manager.Options{})
273+
ctx, cancel := context.WithCancel(context.Background())
274+
275+
m, err := manager.New(ctx, cfg, manager.Options{})
273276
Expect(err).NotTo(HaveOccurred())
274277

275278
By("registering the type in the Scheme")
@@ -308,11 +311,11 @@ var _ = Describe("webhook", func() {
308311
}
309312
}
310313
}`)
311-
stopCh := make(chan struct{})
312-
close(stopCh)
314+
315+
cancel()
313316
// TODO: we may want to improve it to make it be able to inject dependencies,
314317
// but not always try to load certs and return not found error.
315-
err = svr.Start(stopCh)
318+
err = svr.Start(ctx)
316319
if err != nil && !os.IsNotExist(err) {
317320
Expect(err).NotTo(HaveOccurred())
318321
}

‎pkg/cache/cache.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -58,12 +58,12 @@ type Informers interface {
5858
// of the underlying object.
5959
GetInformerForKind(ctx context.Context, gvk schema.GroupVersionKind) (Informer, error)
6060

61-
// Start runs all the informers known to this cache until the given channel is closed.
61+
// Start runs all the informers known to this cache until the context is closed.
6262
// It blocks.
63-
Start(stopCh <-chan struct{}) error
63+
Start(ctx context.Context) error
6464

6565
// WaitForCacheSync waits for all the caches to sync. Returns false if it could not sync a cache.
66-
WaitForCacheSync(stop <-chan struct{}) bool
66+
WaitForCacheSync(ctx context.Context) bool
6767

6868
// Informers knows how to add indices to the caches (informers) that it manages.
6969
client.FieldIndexer

0 commit comments

Comments
 (0)