kyverno/pkg/utils/generator/updaterequests.go

package generator

import (
	"context"
	"errors"

	"github.com/go-logr/logr"
	"github.com/kyverno/kyverno/api/kyverno/v1beta1"
	"github.com/kyverno/kyverno/pkg/client/clientset/versioned"
	configutils "github.com/kyverno/kyverno/pkg/config"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/runtime/schema"
	"k8s.io/client-go/metadata"
)

type UpdateRequestGenerator = Generator[*v1beta1.UpdateRequest]

type updaterequestsgenerator struct {
	config     configutils.Configuration
	metaClient metadata.Interface
}

func NewUpdateRequestGenerator(config configutils.Configuration, metaClient metadata.Interface) UpdateRequestGenerator {
	return &updaterequestsgenerator{
		config:     config,
		metaClient: metaClient,
	}
}

func (g *updaterequestsgenerator) Generate(ctx context.Context, client versioned.Interface, resource *v1beta1.UpdateRequest, log logr.Logger) (*v1beta1.UpdateRequest, error) {
	objects, err := g.metaClient.Resource(
		schema.GroupVersionResource{
			Group:    "kyverno.io",
			Version:  "v1beta1",
			Resource: "updaterequests",
		},
	).List(ctx, metav1.ListOptions{})
	if err != nil {
		return nil, err
	}

	count := len(objects.Items)
	threshold := g.config.GetUpdateRequestThreshold()
	if int64(count) >= threshold {
		log.Error(errors.New("UpdateRequest creation skipped"),
			"the number of updaterequests exceeds the threshold, please adjust updateRequestThreshold in the Kyverno configmap",
			"current count", count, "threshold", threshold)
		return nil, nil
	}

	created, err := client.KyvernoV1beta1().UpdateRequests(configutils.KyvernoNamespace()).Create(ctx, resource, metav1.CreateOptions{})
	return created, err
}