forked from rook/rook
Joshua Hoblitt
9f1ed201db
All existing controller runtime watches are converted to use "typed" handlers and predicates instead of operating on `client.Object`. The intent is to be bug for bug equivalent with the existing logic while replacing run time type assertions and switch statements with compile time type constraints and type casts. In several cases, functions using assertions were split up such that each function only handles a single Kind at a time. It is hoped that this will improve readability and maintainability while facilitating future refactoring such as migrating some watches to using IndexFields. Of particular note is that the massive switch statement in `WatchControllerPredicate()` from `pkg/operator/ceph/controller/predicate.go` has been replaced with generics, reflection, and splitting the obc logic into its own predicate function. There are still many helper functions operating on `client.Object`. These were not updated unless required by the compiler in order to limit the size of this change. The type safety of these funcs should be improved as followup work. It is strongly suggested that going forward, handlers and predicates only handle a single Kind (generic or not) and that switches / type assertions are heavily discouraged or forbidden. This PR removed all but a single switch statement in a predicate, which should be addressed in future work. Signed-off-by: Joshua Hoblitt <josh@hoblitt.com>
104 lines
3.1 KiB
Go
104 lines
3.1 KiB
Go
/*
|
|
Copyright 2021 The Rook Authors. All rights reserved.
|
|
|
|
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 mon
|
|
|
|
import (
|
|
"encoding/json"
|
|
"reflect"
|
|
"sort"
|
|
|
|
opcontroller "github.com/rook/rook/pkg/operator/ceph/controller"
|
|
corev1 "k8s.io/api/core/v1"
|
|
"sigs.k8s.io/controller-runtime/pkg/event"
|
|
"sigs.k8s.io/controller-runtime/pkg/predicate"
|
|
)
|
|
|
|
func PredicateMonEndpointChanges[T *corev1.ConfigMap]() predicate.TypedFuncs[T] {
|
|
return predicate.TypedFuncs[T]{
|
|
CreateFunc: func(e event.TypedCreateEvent[T]) bool {
|
|
return false
|
|
},
|
|
DeleteFunc: func(e event.TypedDeleteEvent[T]) bool {
|
|
return false
|
|
},
|
|
GenericFunc: func(e event.TypedGenericEvent[T]) bool {
|
|
return false
|
|
},
|
|
UpdateFunc: func(e event.TypedUpdateEvent[T]) bool {
|
|
cmOld := (*corev1.ConfigMap)(e.ObjectOld)
|
|
cmNew := (*corev1.ConfigMap)(e.ObjectNew)
|
|
if cmNew.GetName() == EndpointConfigMapName {
|
|
if wereMonEndpointsUpdated(cmOld.Data, cmNew.Data) {
|
|
logger.Info("monitor endpoints changed, updating the bootstrap peer token")
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
},
|
|
}
|
|
}
|
|
|
|
func wereMonEndpointsUpdated(oldCMData, newCMData map[string]string) bool {
|
|
// Check the mapping key first
|
|
if oldMapping, ok := oldCMData["mapping"]; ok {
|
|
if newMapping, ok := newCMData["mapping"]; ok {
|
|
// Unmarshal both into a type
|
|
var oldMappingToGo opcontroller.Mapping
|
|
err := json.Unmarshal([]byte(oldMapping), &oldMappingToGo)
|
|
if err != nil {
|
|
logger.Debugf("failed to unmarshal new. %v", err)
|
|
return false
|
|
}
|
|
|
|
var newMappingToGo opcontroller.Mapping
|
|
err = json.Unmarshal([]byte(newMapping), &newMappingToGo)
|
|
if err != nil {
|
|
logger.Debugf("failed to unmarshal new. %v", err)
|
|
return false
|
|
}
|
|
|
|
// If length is different, monitors are different
|
|
if len(oldMappingToGo.Schedule) != len(newMappingToGo.Schedule) {
|
|
logger.Debugf("mons were added or removed from the endpoints cm")
|
|
return true
|
|
}
|
|
// Since Schedule is map, it's unordered, so let's order it
|
|
oldKeys := make([]string, 0, len(oldMappingToGo.Schedule))
|
|
for k := range oldMappingToGo.Schedule {
|
|
oldKeys = append(oldKeys, k)
|
|
}
|
|
sort.Strings(oldKeys)
|
|
|
|
newKeys := make([]string, 0, len(newMappingToGo.Schedule))
|
|
for k := range oldMappingToGo.Schedule {
|
|
newKeys = append(newKeys, k)
|
|
}
|
|
sort.Strings(newKeys)
|
|
|
|
// Iterate over the map and compare the values
|
|
for _, v := range oldKeys {
|
|
if !reflect.DeepEqual(oldMappingToGo.Schedule[v], newMappingToGo.Schedule[v]) {
|
|
logger.Debugf("oldMappingToGo.Schedule[v] AND newMappingToGo.Schedule[v]: %v | %v", oldMappingToGo.Schedule[v], newMappingToGo.Schedule[v])
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|