mirror of
https://github.com/stakater/Reloader.git
synced 2026-03-02 01:10:19 +00:00
111 lines
3.2 KiB
Go
111 lines
3.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
"github.com/stakater/Reloader/internal/pkg/controller"
|
|
"github.com/stakater/Reloader/internal/pkg/options"
|
|
"github.com/stakater/Reloader/internal/pkg/util"
|
|
"github.com/stakater/Reloader/pkg/kube"
|
|
v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
// NewReloaderCommand starts the reloader controller
|
|
func NewReloaderCommand() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "reloader",
|
|
Short: "A watcher for your Kubernetes cluster",
|
|
Run: startReloader,
|
|
}
|
|
|
|
// options
|
|
cmd.PersistentFlags().StringVar(&options.ConfigmapUpdateOnChangeAnnotation, "configmap-annotation", "configmap.reloader.stakater.com/reload", "annotation to detect changes in configmaps")
|
|
cmd.PersistentFlags().StringVar(&options.SecretUpdateOnChangeAnnotation, "secret-annotation", "secret.reloader.stakater.com/reload", "annotation to detect changes in secrets")
|
|
cmd.PersistentFlags().StringVar(&options.ReloaderAutoAnnotation, "auto-annotation", "reloader.stakater.com/auto", "annotation to detect changes in secrets")
|
|
cmd.PersistentFlags().StringSlice("resources-to-ignore", []string{}, "list of resources to ignore (valid options 'configMaps' or 'secrets')")
|
|
cmd.PersistentFlags().StringSlice("namespaces-to-ignore", []string{}, "list of namespaces to ignore")
|
|
return cmd
|
|
}
|
|
|
|
func startReloader(cmd *cobra.Command, args []string) {
|
|
logrus.Info("Starting Reloader")
|
|
currentNamespace := os.Getenv("KUBERNETES_NAMESPACE")
|
|
if len(currentNamespace) == 0 {
|
|
currentNamespace = v1.NamespaceAll
|
|
logrus.Warnf("KUBERNETES_NAMESPACE is unset, will detect changes in all namespaces.")
|
|
}
|
|
|
|
// create the clientset
|
|
clientset, err := kube.GetKubernetesClient()
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
|
|
ignoredResourcesList, err := getIgnoredResourcesList(cmd)
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
|
|
ignoredNamespacesList, err := getIgnoredNamespacesList(cmd)
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
|
|
for k := range kube.ResourceMap {
|
|
if ignoredResourcesList.Contains(k) {
|
|
continue
|
|
}
|
|
|
|
c, err := controller.NewController(clientset, k, currentNamespace, ignoredNamespacesList)
|
|
if err != nil {
|
|
logrus.Fatalf("%s", err)
|
|
}
|
|
|
|
// Now let's start the controller
|
|
stop := make(chan struct{})
|
|
defer close(stop)
|
|
logrus.Infof("Starting Controller to watch resource type: %s", k)
|
|
go c.Run(1, stop)
|
|
}
|
|
|
|
// Wait forever
|
|
select {}
|
|
}
|
|
|
|
func getIgnoredNamespacesList(cmd *cobra.Command) (util.List, error) {
|
|
return getStringSliceFromFlags(cmd, "namespaces-to-ignore")
|
|
}
|
|
|
|
func getStringSliceFromFlags(cmd *cobra.Command, flag string) ([]string, error) {
|
|
slice, err := cmd.Flags().GetStringSlice(flag)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return slice, nil
|
|
}
|
|
|
|
func getIgnoredResourcesList(cmd *cobra.Command) (util.List, error) {
|
|
|
|
ignoredResourcesList, err := getStringSliceFromFlags(cmd, "resources-to-ignore")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, v := range ignoredResourcesList {
|
|
if v != "configMaps" && v != "secrets" {
|
|
return nil, fmt.Errorf("'resources-to-ignore' only accepts 'configMaps' or 'secrets', not '%s'", v)
|
|
}
|
|
}
|
|
|
|
if len(ignoredResourcesList) > 1 {
|
|
return nil, errors.New("'resources-to-ignore' only accepts 'configMaps' or 'secrets', not both")
|
|
}
|
|
|
|
return ignoredResourcesList, nil
|
|
}
|