kyverno/pkg/engine/globalcontext/externalapi/entry.go

package externalapi

import (
	"context"
	"sync"
	"time"

	"github.com/go-logr/logr"
	kyvernov1 "github.com/kyverno/kyverno/api/kyverno/v1"
	"github.com/kyverno/kyverno/pkg/engine/apicall"
	"k8s.io/apimachinery/pkg/util/wait"
)

type entry struct {
	sync.Mutex
	data any
	stop func()
}

func New(ctx context.Context, logger logr.Logger, client apicall.ClientInterface, call kyvernov1.APICall, period time.Duration) (*entry, error) {
	var group wait.Group
	ctx, cancel := context.WithCancel(ctx)
	stop := func() {
		// Send stop signal to informer's goroutine
		cancel()
		// Wait for the group to terminate
		group.Wait()
	}
	e := &entry{
		stop: stop,
	}
	group.StartWithContext(ctx, func(ctx context.Context) {
		// TODO: make sure we have called it at least once before returning
		// TODO: config
		config := apicall.NewAPICallConfiguration(10000)
		caller := apicall.NewCaller(logger, "TODO", client, config)
		wait.UntilWithContext(ctx, func(ctx context.Context) {
			// TODO
			if data, err := doCall(ctx, caller, call); err != nil {
				logger.Error(err, "failed to get data from api caller")
			} else {
				e.setData(data)
			}
		}, period)
	})
	return e, nil
}

func (e *entry) Get() (any, error) {
	e.Lock()
	defer e.Unlock()
	return e.data, nil
}

func (e *entry) Stop() {
	e.Lock()
	defer e.Unlock()
	e.stop()
}

func (e *entry) setData(data any) {
	e.Lock()
	defer e.Unlock()
	e.data = data
}

func doCall(ctx context.Context, caller apicall.Caller, call kyvernov1.APICall) (any, error) {
	// TODO: unmarshall json ?
	return caller.Execute(ctx, &call)
}