-
Notifications
You must be signed in to change notification settings - Fork 138
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Introduce basic functionality for a v1alpha2 Workflow Controller. This implementation is independent of the existing controller as we won't support the v1alpha1 API when we move to v1alpha2. The first commit moves deprecated code into /internal/deprecated and isn't that interesting: the work is mostly in the [second commit](eb5eb09). The implementation renders the template actions onto the Workflow only which will aid in E2E tests with Tink Server. Logic such as remediating Workflow states is deferred to a later PR.
- Loading branch information
Showing
51 changed files
with
2,006 additions
and
1,139 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,10 @@ | ||
| FROM alpine:3.15 | ||
|
|
||
| ARG TARGETOS | ||
| ARG TARGETARCH | ||
|
|
||
| RUN apk add --no-cache --update --upgrade ca-certificates | ||
|
|
||
| COPY bin/tink-controller-v1alpha2-${TARGETOS}-${TARGETARCH} /usr/bin/tink-controller | ||
|
|
||
| ENTRYPOINT ["/usr/bin/tink-controller"] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,179 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "os" | ||
| "strings" | ||
|
|
||
| "github.com/go-logr/logr" | ||
| "github.com/go-logr/zapr" | ||
| "github.com/spf13/cobra" | ||
| "github.com/spf13/pflag" | ||
| "github.com/spf13/viper" | ||
| tinkv1 "github.com/tinkerbell/tink/api/v1alpha2" | ||
| "github.com/tinkerbell/tink/internal/workflow" | ||
| "go.uber.org/zap" | ||
| "k8s.io/apimachinery/pkg/runtime" | ||
| amruntimeutil "k8s.io/apimachinery/pkg/util/runtime" | ||
| "k8s.io/client-go/tools/clientcmd" | ||
| clientcmdapi "k8s.io/client-go/tools/clientcmd/api" | ||
| ctrl "sigs.k8s.io/controller-runtime" | ||
| "sigs.k8s.io/controller-runtime/pkg/healthz" | ||
| "sigs.k8s.io/controller-runtime/pkg/metrics/server" | ||
| ) | ||
|
|
||
| // version is set at build time. | ||
| var version = "devel" | ||
|
|
||
| // scheme is passed to the manager. | ||
| var scheme = runtime.NewScheme() | ||
|
|
||
| func init() { | ||
| amruntimeutil.Must(tinkv1.AddToScheme(scheme)) | ||
|
|
||
| //+kubebuilder:scaffold:scheme | ||
| } | ||
|
|
||
| type Config struct { | ||
| K8sAPI string | ||
| Kubeconfig string // only applies to out of cluster | ||
| MetricsAddr string | ||
| ProbeAddr string | ||
| EnableLeaderElection bool | ||
| } | ||
|
|
||
| func (c *Config) AddFlags(fs *pflag.FlagSet) { | ||
| fs.StringVar(&c.K8sAPI, "kubernetes", "", | ||
| "The Kubernetes API URL, used for in-cluster client construction.") | ||
| fs.StringVar(&c.Kubeconfig, "kubeconfig", "", "Absolute path to the kubeconfig file") | ||
| fs.StringVar(&c.MetricsAddr, "metrics-bind-address", ":8080", | ||
| "The address the metric endpoint binds to.") | ||
| fs.StringVar(&c.ProbeAddr, "health-probe-bind-address", ":8081", | ||
| "The address the probe endpoint binds to.") | ||
| fs.BoolVar(&c.EnableLeaderElection, "leader-elect", false, | ||
| "Enable leader election for controller manager. "+ | ||
| "Enabling this will ensure there is only one active controller manager.") | ||
| } | ||
|
|
||
| func main() { | ||
| cmd := NewRootCommand() | ||
| if err := cmd.Execute(); err != nil { | ||
| os.Exit(1) | ||
| } | ||
| } | ||
|
|
||
| func NewRootCommand() *cobra.Command { | ||
| var config Config | ||
|
|
||
| zlog, err := zap.NewProduction() | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| logger := zapr.NewLogger(zlog).WithName("github.com/tinkerbell/tink") | ||
|
|
||
| cmd := &cobra.Command{ | ||
| Use: "tink-controller", | ||
| PreRunE: func(cmd *cobra.Command, args []string) error { | ||
| viper, err := createViper(logger) | ||
| if err != nil { | ||
| return fmt.Errorf("config init: %w", err) | ||
| } | ||
| return applyViper(viper, cmd) | ||
| }, | ||
| RunE: func(cmd *cobra.Command, args []string) error { | ||
| logger.Info("Starting controller version " + version) | ||
|
|
||
| ccfg := clientcmd.NewNonInteractiveDeferredLoadingClientConfig( | ||
| &clientcmd.ClientConfigLoadingRules{ExplicitPath: config.Kubeconfig}, | ||
| &clientcmd.ConfigOverrides{ClusterInfo: clientcmdapi.Cluster{Server: config.K8sAPI}}) | ||
|
|
||
| cfg, err := ccfg.ClientConfig() | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| namespace, _, err := ccfg.Namespace() | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| opts := ctrl.Options{ | ||
| Logger: logger, | ||
| LeaderElection: config.EnableLeaderElection, | ||
| LeaderElectionID: "tink.tinkerbell.org", | ||
| LeaderElectionNamespace: namespace, | ||
| Metrics: server.Options{ | ||
| BindAddress: config.MetricsAddr, | ||
| }, | ||
| HealthProbeBindAddress: config.ProbeAddr, | ||
| Scheme: scheme, | ||
| } | ||
|
|
||
| mgr, err := ctrl.NewManager(cfg, opts) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if err := mgr.AddHealthzCheck("healthz", healthz.Ping); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if err := mgr.AddReadyzCheck("readyz", healthz.Ping); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if err := workflow.NewReconciler(mgr.GetClient()).SetupWithManager(mgr); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return mgr.Start(cmd.Context()) | ||
| }, | ||
| } | ||
| config.AddFlags(cmd.Flags()) | ||
| return cmd | ||
| } | ||
|
|
||
| func createViper(logger logr.Logger) (*viper.Viper, error) { | ||
| v := viper.New() | ||
| v.AutomaticEnv() | ||
| v.SetConfigName("tink-controller") | ||
| v.AddConfigPath("/etc/tinkerbell") | ||
| v.AddConfigPath(".") | ||
| v.SetEnvKeyReplacer(strings.NewReplacer("-", "_")) | ||
|
|
||
| // If a config file is found, read it in. | ||
| if err := v.ReadInConfig(); err != nil { | ||
| if _, ok := err.(viper.ConfigFileNotFoundError); !ok { | ||
| return nil, fmt.Errorf("loading config file: %w", err) | ||
| } | ||
| logger.Info("no config file found") | ||
| } else { | ||
| logger.Info("loaded config file", "configFile", v.ConfigFileUsed()) | ||
| } | ||
|
|
||
| return v, nil | ||
| } | ||
|
|
||
| func applyViper(v *viper.Viper, cmd *cobra.Command) error { | ||
| errors := []error{} | ||
|
|
||
| cmd.Flags().VisitAll(func(f *pflag.Flag) { | ||
| if !f.Changed && v.IsSet(f.Name) { | ||
| val := v.Get(f.Name) | ||
| if err := cmd.Flags().Set(f.Name, fmt.Sprintf("%v", val)); err != nil { | ||
| errors = append(errors, err) | ||
| return | ||
| } | ||
| } | ||
| }) | ||
|
|
||
| if len(errors) > 0 { | ||
| errs := []string{} | ||
| for _, err := range errors { | ||
| errs = append(errs, err.Error()) | ||
| } | ||
| return fmt.Errorf(strings.Join(errs, ", ")) | ||
| } | ||
|
|
||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.