mirror of
https://github.com/kyverno/kyverno.git
synced 2025-03-06 16:06:56 +00:00
2078f0dfd2
Signed-off-by: prateekpandey14 <prateek.pandey@nirmata.com> Signed-off-by: prateekpandey14 <prateek.pandey@nirmata.com>
67 lines
2.5 KiB
Go
67 lines
2.5 KiB
Go
package common
|
|
|
|
import (
|
|
"context"
|
|
|
|
kyvernov1 "github.com/kyverno/kyverno/api/kyverno/v1"
|
|
kyvernov1beta1 "github.com/kyverno/kyverno/api/kyverno/v1beta1"
|
|
"github.com/kyverno/kyverno/pkg/client/clientset/versioned"
|
|
kyvernov1beta1listers "github.com/kyverno/kyverno/pkg/client/listers/kyverno/v1beta1"
|
|
"github.com/kyverno/kyverno/pkg/config"
|
|
"github.com/kyverno/kyverno/pkg/logging"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/util/retry"
|
|
)
|
|
|
|
func Update(client versioned.Interface, urLister kyvernov1beta1listers.UpdateRequestNamespaceLister, name string, mutator func(*kyvernov1beta1.UpdateRequest)) (*kyvernov1beta1.UpdateRequest, error) {
|
|
var ur *kyvernov1beta1.UpdateRequest
|
|
err := retry.RetryOnConflict(retry.DefaultRetry, func() error {
|
|
ur, err := urLister.Get(name)
|
|
if err != nil {
|
|
logging.Error(err, "[ATTEMPT] failed to fetch update request", "name", name)
|
|
return err
|
|
}
|
|
ur = ur.DeepCopy()
|
|
mutator(ur)
|
|
_, err = client.KyvernoV1beta1().UpdateRequests(config.KyvernoNamespace()).Update(context.TODO(), ur, metav1.UpdateOptions{})
|
|
if err != nil {
|
|
logging.Error(err, "[ATTEMPT] failed to update update request", "name", name)
|
|
}
|
|
return err
|
|
})
|
|
if err != nil {
|
|
logging.Error(err, "failed to update update request", "name", name)
|
|
} else {
|
|
logging.V(3).Info("updated update request", "name", name)
|
|
}
|
|
return ur, err
|
|
}
|
|
|
|
func UpdateStatus(client versioned.Interface, urLister kyvernov1beta1listers.UpdateRequestNamespaceLister, name string, state kyvernov1beta1.UpdateRequestState, message string, genResources []kyvernov1.ResourceSpec) (*kyvernov1beta1.UpdateRequest, error) {
|
|
var ur *kyvernov1beta1.UpdateRequest
|
|
err := retry.RetryOnConflict(retry.DefaultRetry, func() error {
|
|
ur, err := urLister.Get(name)
|
|
if err != nil {
|
|
logging.Error(err, "[ATTEMPT] failed to fetch update request", "name", name)
|
|
return err
|
|
}
|
|
ur = ur.DeepCopy()
|
|
ur.Status.State = state
|
|
ur.Status.Message = message
|
|
if genResources != nil {
|
|
ur.Status.GeneratedResources = genResources
|
|
}
|
|
_, err = client.KyvernoV1beta1().UpdateRequests(config.KyvernoNamespace()).UpdateStatus(context.TODO(), ur, metav1.UpdateOptions{})
|
|
if err != nil {
|
|
logging.Error(err, "[ATTEMPT] failed to update update request status", "name", name)
|
|
return err
|
|
}
|
|
return err
|
|
})
|
|
if err != nil {
|
|
logging.Error(err, "failed to update update request status", "name", name)
|
|
} else {
|
|
logging.V(3).Info("updated update request status", "name", name, "status", string(state))
|
|
}
|
|
return ur, err
|
|
}
|