mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
27 lines
1.1 KiB
Go
27 lines
1.1 KiB
Go
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{})
|
|
}
|