-
Notifications
You must be signed in to change notification settings - Fork 40
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
handler,predicate: pause object reconciliation by annotation #60
Merged
openshift-merge-robot
merged 2 commits into
operator-framework:main
from
estroz:feature/pause
Jul 23, 2021
Merged
Changes from all commits
Commits
Show all changes
2 commits
Select commit
Hold shift + click to select a range
File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,66 @@ | ||
// Copyright 2021 The Operator-SDK Authors | ||
// | ||
// Licensed under the Apache License, Version 2.0 (the "License"); | ||
// you may not use this file except in compliance with the License. | ||
// You may obtain a copy of the License at | ||
// | ||
// http://www.apache.org/licenses/LICENSE-2.0 | ||
// | ||
// Unless required by applicable law or agreed to in writing, software | ||
// distributed under the License is distributed on an "AS IS" BASIS, | ||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
// See the License for the specific language governing permissions and | ||
// limitations under the License. | ||
|
||
package handler_test | ||
|
||
import ( | ||
"context" | ||
"os" | ||
|
||
"github.com/operator-framework/operator-lib/handler" | ||
v1 "k8s.io/api/core/v1" | ||
"sigs.k8s.io/controller-runtime/pkg/client/config" | ||
"sigs.k8s.io/controller-runtime/pkg/controller" | ||
"sigs.k8s.io/controller-runtime/pkg/manager" | ||
"sigs.k8s.io/controller-runtime/pkg/manager/signals" | ||
"sigs.k8s.io/controller-runtime/pkg/reconcile" | ||
"sigs.k8s.io/controller-runtime/pkg/source" | ||
) | ||
|
||
// This example applies the Pause handler to all incoming Pod events on a Pod controller. | ||
func ExampleNewPause() { | ||
cfg, err := config.GetConfig() | ||
if err != nil { | ||
os.Exit(1) | ||
} | ||
|
||
mgr, err := manager.New(cfg, manager.Options{}) | ||
if err != nil { | ||
os.Exit(1) | ||
} | ||
|
||
c, err := controller.NewUnmanaged("pod", mgr, controller.Options{ | ||
Reconciler: reconcile.Func(func(context.Context, reconcile.Request) (reconcile.Result, error) { | ||
return reconcile.Result{}, nil | ||
}), | ||
}) | ||
if err != nil { | ||
os.Exit(1) | ||
} | ||
|
||
// Filter out Pods with the "my.app/paused: true" annotation. | ||
pause, err := handler.NewPause("my.app/paused") | ||
if err != nil { | ||
os.Exit(1) | ||
} | ||
if err := c.Watch(&source.Kind{Type: &v1.Pod{}}, pause); err != nil { | ||
os.Exit(1) | ||
} | ||
|
||
<-mgr.Elected() | ||
|
||
if err := c.Start(signals.SetupSignalHandler()); err != nil { | ||
os.Exit(1) | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,36 @@ | ||
// Copyright 2021 The Operator-SDK Authors | ||
// | ||
// Licensed under the Apache License, Version 2.0 (the "License"); | ||
// you may not use this file except in compliance with the License. | ||
// You may obtain a copy of the License at | ||
// | ||
// http://www.apache.org/licenses/LICENSE-2.0 | ||
// | ||
// Unless required by applicable law or agreed to in writing, software | ||
// distributed under the License is distributed on an "AS IS" BASIS, | ||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
// See the License for the specific language governing permissions and | ||
// limitations under the License. | ||
|
||
package handler | ||
|
||
import ( | ||
"github.com/operator-framework/operator-lib/internal/annotation" | ||
|
||
"sigs.k8s.io/controller-runtime/pkg/handler" | ||
) | ||
|
||
// NewPause returns an event handler that filters out objects with a truthy "paused" annotation. | ||
// When an annotation with key string key is present on an object and has a truthy value, ex. "true", | ||
// the watch constructed with this event handler will not add events for that object to the queue. | ||
// Key string key must be a valid annotation key. | ||
// | ||
// A note on security: since users that can CRUD a particular API can apply or remove annotations with | ||
// default cluster admission controllers, this same set of users can therefore start or stop reconciliation | ||
// of objects via this pause mechanism. If this is a concern, configure an admission webhook to enforce | ||
// a stricter annotation modification policy. See AdmissionReview configuration for user info available | ||
// to a webhook: | ||
// https://kubernetes.io/docs/reference/access-authn-authz/extensible-admission-controllers/#request | ||
func NewPause(key string) (handler.EventHandler, error) { | ||
return annotation.NewFalsyEventHandler(key, annotation.Options{Log: log}) | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,210 @@ | ||
// Copyright 2021 The Operator-SDK Authors | ||
// | ||
// Licensed under the Apache License, Version 2.0 (the "License"); | ||
// you may not use this file except in compliance with the License. | ||
// You may obtain a copy of the License at | ||
// | ||
// http://www.apache.org/licenses/LICENSE-2.0 | ||
// | ||
// Unless required by applicable law or agreed to in writing, software | ||
// distributed under the License is distributed on an "AS IS" BASIS, | ||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
// See the License for the specific language governing permissions and | ||
// limitations under the License. | ||
|
||
// Package annotation contains event handler and predicate builders for annotations. | ||
// There are two types of builders: | ||
// | ||
// - Falsy builders result in objects being queued if the annotation is not present OR contains a falsy value. | ||
// - Truthy builders are the falsy complement: objects will be enqueued if the annotation is present AND contains a truthy value. | ||
// | ||
// Truthiness/falsiness is determined by Go's strconv.ParseBool(). | ||
package annotation | ||
|
||
import ( | ||
"fmt" | ||
"strconv" | ||
|
||
"github.com/go-logr/logr" | ||
"k8s.io/apimachinery/pkg/api/validation" | ||
"k8s.io/apimachinery/pkg/util/validation/field" | ||
"k8s.io/client-go/util/workqueue" | ||
"sigs.k8s.io/controller-runtime/pkg/client" | ||
"sigs.k8s.io/controller-runtime/pkg/event" | ||
"sigs.k8s.io/controller-runtime/pkg/handler" | ||
logf "sigs.k8s.io/controller-runtime/pkg/log" | ||
"sigs.k8s.io/controller-runtime/pkg/predicate" | ||
) | ||
|
||
// Options configures a filter. | ||
type Options struct { | ||
Log logr.Logger | ||
|
||
// Internally set. | ||
truthy bool | ||
} | ||
|
||
// NewFalsyPredicate returns a predicate that passes objects | ||
// that do not have annotation with key string key or whose value is falsy. | ||
func NewFalsyPredicate(key string, opts Options) (predicate.Predicate, error) { | ||
opts.truthy = false | ||
return newFilter(key, opts) | ||
} | ||
|
||
// NewFalsyEventHandler returns an event handler that enqueues objects | ||
// that do not have annotation with key string key or whose value is falsy. | ||
func NewFalsyEventHandler(key string, opts Options) (handler.EventHandler, error) { | ||
opts.truthy = false | ||
return newEventHandler(key, opts) | ||
} | ||
|
||
// NewTruthyPredicate returns a predicate that passes objects | ||
// that do have annotation with key string key and whose value is truthy. | ||
func NewTruthyPredicate(key string, opts Options) (predicate.Predicate, error) { | ||
opts.truthy = true | ||
return newFilter(key, opts) | ||
} | ||
|
||
// NewTruthyEventHandler returns an event handler that enqueues objects | ||
// that do have annotation with key string key and whose value is truthy. | ||
func NewTruthyEventHandler(key string, opts Options) (handler.EventHandler, error) { | ||
opts.truthy = true | ||
return newEventHandler(key, opts) | ||
} | ||
|
||
func defaultOptions(opts *Options) { | ||
if opts.Log == nil { | ||
opts.Log = logf.Log | ||
} | ||
} | ||
|
||
// newEventHandler returns a filter for use as an event handler. | ||
func newEventHandler(key string, opts Options) (handler.EventHandler, error) { | ||
f, err := newFilter(key, opts) | ||
if err != nil { | ||
return nil, err | ||
} | ||
|
||
f.hdlr = &handler.EnqueueRequestForObject{} | ||
return handler.Funcs{ | ||
CreateFunc: func(evt event.CreateEvent, q workqueue.RateLimitingInterface) { | ||
if f.Create(evt) { | ||
f.hdlr.Create(evt, q) | ||
} | ||
}, | ||
UpdateFunc: func(evt event.UpdateEvent, q workqueue.RateLimitingInterface) { | ||
if f.Update(evt) { | ||
f.hdlr.Update(evt, q) | ||
} | ||
}, | ||
DeleteFunc: func(evt event.DeleteEvent, q workqueue.RateLimitingInterface) { | ||
if f.Delete(evt) { | ||
f.hdlr.Delete(evt, q) | ||
} | ||
}, | ||
GenericFunc: func(evt event.GenericEvent, q workqueue.RateLimitingInterface) { | ||
if f.Generic(evt) { | ||
f.hdlr.Generic(evt, q) | ||
} | ||
}, | ||
}, nil | ||
} | ||
|
||
// newFilter returns a filter for use as a predicate. | ||
func newFilter(key string, opts Options) (*filter, error) { | ||
defaultOptions(&opts) | ||
|
||
// Make sure the annotation key and eventual value are valid together. | ||
if err := validateAnnotation(key, opts.truthy); err != nil { | ||
return nil, err | ||
} | ||
|
||
f := filter{} | ||
f.key = key | ||
// Falsy filters return true in all cases except when the annotation is present and true. | ||
// Truthy filters only return true when the annotation is present and true. | ||
f.ret = !opts.truthy | ||
f.log = opts.Log.WithName("pause") | ||
return &f, nil | ||
} | ||
|
||
func validateAnnotation(key string, truthy bool) error { | ||
fldPath := field.NewPath("metadata", "annotations") | ||
annotation := map[string]string{key: fmt.Sprintf("%v", truthy)} | ||
return validation.ValidateAnnotations(annotation, fldPath).ToAggregate() | ||
} | ||
|
||
// filter implements a filter for objects with a truthy "paused" annotation (see Key). | ||
// When this annotation is removed or value does not evaluate to "true", | ||
// the controller will see events from these objects again. | ||
type filter struct { | ||
key string | ||
ret bool | ||
log logr.Logger | ||
hdlr *handler.EnqueueRequestForObject | ||
} | ||
|
||
// Create implements predicate.Predicate.Create(). | ||
func (f *filter) Create(evt event.CreateEvent) bool { | ||
if evt.Object == nil { | ||
if f.hdlr == nil { | ||
f.log.Error(nil, "CreateEvent received with no metadata", "event", evt) | ||
} | ||
return f.ret | ||
} | ||
return f.run(evt.Object) | ||
} | ||
|
||
// Update implements predicate.Predicate.Update(). | ||
func (f *filter) Update(evt event.UpdateEvent) bool { | ||
if evt.ObjectNew != nil { | ||
return f.run(evt.ObjectNew) | ||
} else if evt.ObjectOld != nil { | ||
return f.run(evt.ObjectOld) | ||
} | ||
if f.hdlr == nil { | ||
f.log.Error(nil, "UpdateEvent received with no metadata", "event", evt) | ||
} | ||
return f.ret | ||
} | ||
|
||
// Delete implements predicate.Predicate.Delete(). | ||
func (f *filter) Delete(evt event.DeleteEvent) bool { | ||
if evt.Object == nil { | ||
if f.hdlr == nil { | ||
f.log.Error(nil, "DeleteEvent received with no metadata", "event", evt) | ||
} | ||
return f.ret | ||
} | ||
return f.run(evt.Object) | ||
} | ||
|
||
// Generic implements predicate.Predicate.Generic(). | ||
func (f *filter) Generic(evt event.GenericEvent) bool { | ||
if evt.Object == nil { | ||
if f.hdlr == nil { | ||
f.log.Error(nil, "GenericEvent received with no metadata", "event", evt) | ||
} | ||
return f.ret | ||
} | ||
return f.run(evt.Object) | ||
} | ||
|
||
func (f *filter) run(obj client.Object) bool { | ||
annotations := obj.GetAnnotations() | ||
if len(annotations) == 0 { | ||
return f.ret | ||
} | ||
annoStr, hasAnno := annotations[f.key] | ||
if !hasAnno { | ||
return f.ret | ||
} | ||
annoBool, err := strconv.ParseBool(annoStr) | ||
if err != nil { | ||
f.log.Error(err, "Bad annotation value", "key", f.key, "value", annoStr) | ||
return f.ret | ||
} | ||
// If the filter is falsy (f.ret == true) and value is false, then the object passes the filter. | ||
// If the filter is truthy (f.ret == false) and value is true, then the object passes the filter. | ||
return !annoBool == f.ret | ||
} |
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Non-blocking thought:
It might be generally useful to add a
Not(in predicate.Predicate) predicate.Predicate
wrapper in C-R that inverts the logic of the input predicate.If that existed, would you need the
ret
field?I alluded to this in the C-R PR where I added the
Or
andAnd
wrappers here, but I never added it because there hasn't been a use case for it.There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I don't think you'd need
ret
here in that case. Given that operator-lib is pre v1.0, are you ok with adding this as-is now then removing ifNot
is added to c-r?