controllerof_lookup.go

controllerof_lookup.go

package controllers

import (
	"k8s.io/apimachinery/pkg/runtime/schema"

	corev1alpha1 "package-operator.run/apis/core/v1alpha1"
)

type ControllerOfLookupKey struct {
	schema.GroupKind
	Namespace, Name string
}

type ControllerOfLookup map[ControllerOfLookupKey]struct{}

func NewControllerOfLookup(controlledObjectReferences []corev1alpha1.ControlledObjectReference) ControllerOfLookup {
	l := ControllerOfLookup{}
	for _, obj := range controlledObjectReferences {
		l[ControllerOfLookupKey{
			GroupKind: schema.GroupKind{
				Group: obj.Group,
				Kind:  obj.Kind,
			},
			Namespace: obj.Namespace,
			Name:      obj.Name,
		}] = struct{}{}
	}
	return l
}

func (l ControllerOfLookup) Has(gk schema.GroupKind, namespace, name string) bool {
	_, has := l[ControllerOfLookupKey{
		GroupKind: gk,
		Namespace: namespace,
		Name:      name,
	}]
	return has
}