From d9067344a935d509e25242f3a66611ce55a119f1 Mon Sep 17 00:00:00 2001
From: Gibson Smith <123333011+gdsmith1@users.noreply.github.com>
Date: Wed, 20 Nov 2024 15:27:09 -0800
Subject: [PATCH] chore: added skeleton controller and deployment to examples
 (#609)

---
 examples/ch8/controllers/deployment.yaml      |  19 ++
 .../ch8/controllers/skeleton-controller.go    | 163 ++++++++++++++++++
 2 files changed, 182 insertions(+)
 create mode 100644 examples/ch8/controllers/deployment.yaml
 create mode 100644 examples/ch8/controllers/skeleton-controller.go

diff --git a/examples/ch8/controllers/deployment.yaml b/examples/ch8/controllers/deployment.yaml
new file mode 100644
index 00000000..8d32af01
--- /dev/null
+++ b/examples/ch8/controllers/deployment.yaml
@@ -0,0 +1,19 @@
+# This is going to need some RBAC permissions...
+apiVersion: apps/v1
+kind: Deployment
+metadata:
+  name: my-controller
+  namespace: #YOUR-NAMESPACE
+spec:
+  replicas: 1
+  selector:
+    matchLabels:
+      app: #YOUR-APP
+  template:
+    metadata:
+      labels:
+        app: #YOUR-APP
+    spec:
+      containers:
+      - name: #YOUR-CONTAINER
+        image: #YOUR-IMAGE
\ No newline at end of file
diff --git a/examples/ch8/controllers/skeleton-controller.go b/examples/ch8/controllers/skeleton-controller.go
new file mode 100644
index 00000000..9d42e0d3
--- /dev/null
+++ b/examples/ch8/controllers/skeleton-controller.go
@@ -0,0 +1,163 @@
+package main
+
+import (
+	"fmt"
+	"time"
+
+	_ "github.com/google/go-cmp/cmp" // Add this line
+	"k8s.io/apimachinery/pkg/util/runtime"
+	"k8s.io/apimachinery/pkg/util/wait"
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/client-go/tools/cache"
+	"k8s.io/client-go/tools/clientcmd"
+	"k8s.io/client-go/util/workqueue"
+	"k8s.io/klog/v2"
+)
+
+type Controller struct {
+	kubeclientset kubernetes.Interface
+	workqueue     workqueue.RateLimitingInterface
+	//Needs a resource specific lister
+	//Needs a resource specific informer
+	//Pod Example:
+	// podsLister    v1.PodLister
+	// podsInformer  cache.SharedIndexInformer
+}
+
+func NewController(
+	kubeclientset kubernetes.Interface,
+	podsInformer cache.SharedIndexInformer) *Controller {
+
+	workqueue := workqueue.NewNamedRateLimitingQueue(workqueue.DefaultControllerRateLimiter(), "YourResourceHere")
+
+	controller := &Controller{
+		kubeclientset: kubeclientset,
+		workqueue:     workqueue,
+		//Needs a resource specific lister
+		//Needs a resource specific informer
+		//Pod Example:
+		// podsLister:    v1.NewPodLister(podsInformer.GetIndexer()),
+		// podsInformer:  podsInformer,
+	}
+
+	podsInformer.AddEventHandler(cache.ResourceEventHandlerFuncs{
+		AddFunc: controller.enqueue,
+		UpdateFunc: func(old, new interface{}) {
+			controller.enqueue(new)
+		},
+		DeleteFunc: controller.enqueue,
+	})
+
+	return controller
+}
+
+func (c *Controller) enqueue(obj interface{}) {
+	key, err := cache.MetaNamespaceKeyFunc(obj)
+	if err != nil {
+		runtime.HandleError(err)
+		return
+	}
+	c.workqueue.Add(key)
+}
+
+func (c *Controller) runWorker() {
+	for c.processNextWorkItem() {
+	}
+}
+
+func (c *Controller) processNextWorkItem() bool {
+	obj, shutdown := c.workqueue.Get()
+
+	if shutdown {
+		return false
+	}
+
+	err := func(obj interface{}) error {
+		defer c.workqueue.Done(obj)
+		var key string
+		var ok bool
+		if key, ok = obj.(string); !ok {
+			c.workqueue.Forget(obj)
+			klog.Errorf("expected string in workqueue but got %#v", obj)
+			return nil
+		}
+		if err := c.syncHandler(key); err != nil {
+			c.workqueue.AddRateLimited(key)
+			return fmt.Errorf("error syncing '%s': %s, requeuing", key, err.Error())
+		}
+		c.workqueue.Forget(key)
+		klog.Infof("Successfully synced '%s'", key)
+		return nil
+	}(obj)
+
+	if err != nil {
+		klog.Errorf("error processing item: %v", err)
+		return true
+	}
+
+	return true
+}
+
+func (c *Controller) syncHandler(key string) error {
+	//resource-specific logic for object processing
+	return nil
+}
+
+func (c *Controller) Run(threadiness int, stopCh <-chan struct{}) error {
+	defer runtime.HandleCrash()
+	defer c.workqueue.ShutDown()
+
+	klog.Info("Starting Pod controller")
+
+	//Need to run informer as a goroutine here
+	//Example for pods:
+	//go c.podsInformer.Run(stopCh)
+
+	// if !cache.WaitForCacheSync(stopCh, c.podsInformer.HasSynced) {
+	// 	return fmt.Errorf("failed to wait for caches to sync")
+	// }
+
+	for i := 0; i < threadiness; i++ {
+		go wait.Until(c.runWorker, time.Second, stopCh)
+	}
+
+	<-stopCh
+	klog.Info("Shutting down workers")
+
+	return nil
+}
+
+func main() {
+	kubeconfig := clientcmd.NewNonInteractiveDeferredLoadingClientConfig(
+		clientcmd.NewDefaultClientConfigLoadingRules(),
+		&clientcmd.ConfigOverrides{},
+	)
+	config, err := kubeconfig.ClientConfig()
+	if err != nil {
+		klog.Fatalf("Error building kubeconfig: %s", err.Error())
+	}
+
+	kubeclientset, err := kubernetes.NewForConfig(config)
+	if err != nil {
+		klog.Fatalf("Error building kubernetes clientset: %s", err.Error())
+	}
+
+	//Set up informer factory and informer instance
+	//Make controller instance
+	//Pod Example:
+
+	// informerFactory := informers.NewSharedInformerFactory(kubeclientset, time.Second*30)
+	// podsInformer := informerFactory.Core().V1().Pods().Informer()
+
+	// controller := NewController(kubeclientset, podsInformer)
+
+	// stopCh := make(chan struct{})
+	// defer close(stopCh)
+
+	// go informerFactory.Start(stopCh)
+
+	// if err = controller.Run(2, stopCh); err != nil {
+	// 	klog.Fatalf("Error running controller: %s", err.Error())
+	// }
+
+}