archive/core/nebula/controller/main.go

package main

import (
	"flag"
	"time"

	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
	kubeinformers "k8s.io/client-go/informers"
	"k8s.io/client-go/kubernetes"
	"k8s.io/client-go/tools/clientcmd"

	controllers "github.com/giolekva/pcloud/core/nebula/controller/controllers"
	clientset "github.com/giolekva/pcloud/core/nebula/controller/generated/clientset/versioned"
	"github.com/giolekva/pcloud/core/nebula/controller/generated/clientset/versioned/scheme"
	informers "github.com/giolekva/pcloud/core/nebula/controller/generated/informers/externalversions"

	nebulascheme "k8s.io/sample-controller/pkg/generated/clientset/versioned/scheme"
)

var kubeConfig = flag.String("kubeconfig", "", "Path to a kubeconfig. Only required if out-of-cluster.")
var masterURL = flag.String("master", "", "The address of the Kubernetes API server. Overrides any value in kubeconfig. Only required if out-of-cluster.")

func main() {
	flag.Parse()
	cfg, err := clientcmd.BuildConfigFromFlags(*masterURL, *kubeConfig)
	if err != nil {
		panic(err)
	}
	kubeClient, err := kubernetes.NewForConfig(cfg)
	if err != nil {
		panic(err)
	}
	nebulaClient := clientset.NewForConfigOrDie(cfg)
	utilruntime.Must(nebulascheme.AddToScheme(scheme.Scheme))
	kubeInformerFactory := kubeinformers.NewSharedInformerFactory(kubeClient, time.Second*30)
	nebulaInformerFactory := informers.NewSharedInformerFactory(nebulaClient, 5*time.Second)
	c := controllers.NewNebulaController(
		kubeClient,
		nebulaClient,
		nebulaInformerFactory.Lekva().V1().NebulaCAs(),
		nebulaInformerFactory.Lekva().V1().NebulaNodes(),
		kubeInformerFactory.Core().V1().Secrets())
	stopCh := make(chan struct{})
	kubeInformerFactory.Start(stopCh)
	nebulaInformerFactory.Start(stopCh)
	if err := c.Run(1, stopCh); err != nil {
		panic(err)
	}
}