norman/pkg/objectset/template.go

package objectset

import (
	"github.com/rancher/norman/controller"
	"github.com/rancher/norman/objectclient"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/apimachinery/pkg/runtime/schema"
)

type Client interface {
	Generic() controller.GenericController
	ObjectClient() *objectclient.ObjectClient
}

type Processor struct {
	setID       string
	codeVersion string
	clients     map[schema.GroupVersionKind]Client
}

func NewProcessor(setID string) Processor {
	return Processor{
		setID:   setID,
		clients: map[schema.GroupVersionKind]Client{},
	}
}

func (t Processor) CodeVersion(version string) Processor {
	t.codeVersion = version
	return t
}

func (t Processor) Client(clients ...Client) Processor {
	// ensure cache is enabled
	for _, client := range clients {
		client.Generic()
		t.clients[client.ObjectClient().GroupVersionKind()] = client
	}
	return t
}

func (t Processor) Remove(owner runtime.Object) error {
	return t.NewDesiredSet(owner, nil).Apply()
}

func (t Processor) NewDesiredSet(owner runtime.Object, objs *ObjectSet) *DesiredSet {
	if objs == nil {
		objs = &ObjectSet{}
	}
	return &DesiredSet{
		objs:        objs,
		setID:       t.setID,
		codeVersion: t.codeVersion,
		clients:     t.clients,
		owner:       owner,
	}
}