diff --git a/root.go b/root.go index 3b70357..7675338 100644 --- a/root.go +++ b/root.go @@ -36,7 +36,7 @@ var nodeName string var operatingSystem string var provider string var providerConfig string -var taint string +var disableTaint bool // RootCmd represents the base command when called without any subcommands var RootCmd = &cobra.Command{ @@ -47,7 +47,7 @@ backend implementation allowing users to create kubernetes nodes without running This allows users to schedule kubernetes workloads on nodes that aren't running Kubernetes.`, Run: func(cmd *cobra.Command, args []string) { fmt.Println(kubeConfig) - f, err := vkubelet.New(nodeName, operatingSystem, kubeNamespace, kubeConfig, taint, provider, providerConfig) + f, err := vkubelet.New(nodeName, operatingSystem, kubeNamespace, kubeConfig, provider, providerConfig, disableTaint) if err != nil { log.Fatal(err) } @@ -82,7 +82,7 @@ func init() { RootCmd.PersistentFlags().StringVar(&nodeName, "nodename", defaultNodeName, "kubernetes node name") RootCmd.PersistentFlags().StringVar(&operatingSystem, "os", "Linux", "Operating System (Linux/Windows)") RootCmd.PersistentFlags().StringVar(&provider, "provider", "", "cloud provider") - RootCmd.PersistentFlags().StringVar(&taint, "taint", "", "apply taint to node, making scheduling explicit") + RootCmd.PersistentFlags().BoolVar(&disableTaint, "disable-taint", false, "disable the virtual-kubelet node taint") RootCmd.PersistentFlags().StringVar(&providerConfig, "provider-config", "", "cloud provider configuration file") // Cobra also supports local flags, which will only run