pkg/k8s/pod_wrapper.go

package k8s

import (
	"fmt"

	v1 "k8s.io/api/core/v1"
	apimeta "k8s.io/apimachinery/pkg/api/meta"
	"k8s.io/apimachinery/pkg/labels"
	"k8s.io/apimachinery/pkg/types"
	"sigs.k8s.io/controller-runtime/pkg/client"
)

// PodsRepository represents an interface with all the common operations on pod objects
type PodsRepository interface {
	GetPod(namespace string, name string) (*v1.Pod, error)
	ListPodsWithMatchingLabels(opts client.ListOptions) (*v1.PodList, error)
}

// podsRepository is the wrapper object with the client
type podsRepository struct {
	customController *CustomController
}

// NewPodsRepository returns a new PodsRepository
func NewPodsRepository(customController *CustomController) PodsRepository {
	return &podsRepository{
		customController: customController,
	}
}

// GetPod returns the pod object using NamespacedName
func (k *podsRepository) GetPod(namespace string, name string) (*v1.Pod, error) {
	nsName := types.NamespacedName{
		Namespace: namespace,
		Name:      name,
	}.String()
	obj, exists, err := k.customController.GetDataStore().GetByKey(nsName)
	if err != nil {
		return nil, err
	}
	if !exists {
		return nil, fmt.Errorf("failed to find pod %s", nsName)
	}
	return obj.(*v1.Pod), nil
}

// ListPods return list of pods within a Namespace having Matching Labels
// ListOptions.LabelSelector must be specified to return pods with matching labels
// ListOptions.Namespace will scope result list to a given namespace
func (k *podsRepository) ListPodsWithMatchingLabels(opts client.ListOptions) (*v1.PodList, error) {
	var items []interface{}
	var err error

	if opts.Namespace != "" {
		items, err = k.customController.GetDataStore().ByIndex(NamespaceIndexKey, opts.Namespace)
	} else {
		items = k.customController.GetDataStore().List()
	}
	if err != nil {
		return nil, err
	}

	podList := &v1.PodList{}

	var labelSel labels.Selector
	if opts.LabelSelector != nil {
		labelSel = opts.LabelSelector
	}

	for _, item := range items {
		pod, ok := item.(*v1.Pod)
		if !ok {
			return nil, fmt.Errorf("cache contained %T, which is not a Pod", item)
		}

		meta, err := apimeta.Accessor(pod)
		if err != nil {
			return nil, err
		}
		if labelSel != nil {
			lbls := labels.Set(meta.GetLabels())
			if !labelSel.Matches(lbls) {
				continue
			}
		}
		podList.Items = append(podList.Items, *pod)
	}
	return podList, nil
}