Git-Mirrors/constellation
1
0
Fork 0
mirror of https://github.com/edgelesssys/constellation.git synced 2025-05-02 14:26:23 -04:00
Code Issues Projects Releases Packages Wiki Activity

[node operator] patching util functions

Browse source 
Signed-off-by: Malte Poll <mp@edgeless.systems>
This commit is contained in:
Malte Poll 2022-06-28 09:55:05 +02:00 • committed by Malte Poll
parent 614447495d
commit 3932581f2a
6 changed files with 212 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

26
operators/constellation-node-operator/internal/patch/annotations.go Normal file
View file

@ -0,0 +1,26 @@
package patch
import "sigs.k8s.io/controller-runtime/pkg/client"
// SetAnnotations creates a patch for a client.Object by merging annotations with existing annotations.
func SetAnnotations(original, patched client.Object, annotations map[string]string) client.Patch {
mergedAnnotations := patched.GetAnnotations()
if mergedAnnotations == nil {
mergedAnnotations = make(map[string]string, len(annotations))
}
for annotationKey, annotationValue := range annotations {
mergedAnnotations[annotationKey] = annotationValue
}
patched.SetAnnotations(mergedAnnotations)
return client.StrategicMergeFrom(original, client.MergeFromWithOptimisticLock{})
}
// UnsetAnnotations creates a patch for a client.Object by deleting annotations from the object.
func UnsetAnnotations(original, patched client.Object, annotationKeys []string) client.Patch {
mergedAnnotations := patched.GetAnnotations()
for _, annotationKey := range annotationKeys {
delete(mergedAnnotations, annotationKey)
}
patched.SetAnnotations(mergedAnnotations)
return client.StrategicMergeFrom(original, client.MergeFromWithOptimisticLock{})
}