dynamic_controller.go

package dynamic

import (
	"fmt"
	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
	"sigs.k8s.io/controller-runtime/pkg/client/config"
	"sigs.k8s.io/controller-runtime/pkg/controller"
	"sigs.k8s.io/controller-runtime/pkg/handler"
	"sigs.k8s.io/controller-runtime/pkg/manager"
	"sigs.k8s.io/controller-runtime/pkg/reconcile"
	"sigs.k8s.io/controller-runtime/pkg/source"
	"sync"
	"time"
)

type ControllerRegistry struct {
	controllers map[string](chan struct{})
	errChan     chan error
	sync.Mutex
	mgr manager.Manager
}

func (c *ControllerRegistry) Register(u unstructured.Unstructured) error {
	c.Lock()
	defer c.Unlock()
	if c.controllers == nil {
		c.controllers = map[string](chan struct{}){}
	}
	cfg, err := config.GetConfig()
	if err != nil {
		return err
	}

	stopMgr := make(chan struct{})
	if c.mgr == nil {
		mgr, err := manager.New(cfg, manager.Options{})
		c.mgr = mgr
		if err != nil {
			return err
		}
		go c.mgr.Start(stopMgr)
	}

	ctrlName := fmt.Sprintf("%s-%s-%d", u.GroupVersionKind().String(), "controller", time.Now().Unix())
	ctrl, err := controller.New(ctrlName, c.mgr, controller.Options{
		Reconciler: reconcile.Func(func(o reconcile.Request) (reconcile.Result, error) {
			fmt.Println("RECONCILING DYNAMIC THING")
			return reconcile.Result{}, nil
		}),
	})
	if err := ctrl.Watch(&source.Kind{Type: &u}, &handler.EnqueueRequestForObject{}); err != nil {
		fmt.Printf("%+v\n", err)
	}
	ctrlCh := make(chan struct{})
	c.controllers[u.GroupVersionKind().String()] = ctrlCh
	return nil
}

func (c *ControllerRegistry) Stop(u unstructured.Unstructured) error {
	c.Lock()
	gvk := u.GroupVersionKind().String()
	close(c.controllers[gvk])
	delete(c.controllers, gvk)
	c.Unlock()
	return nil
}