mirror of
https://github.com/kubernetes-sigs/node-feature-discovery.git
synced 2024-12-14 11:57:51 +00:00
1026d91d12
Simplify code bu dropping the unnecessary base client package.
156 lines
5 KiB
Go
156 lines
5 KiB
Go
/*
|
|
Copyright 2019-2021 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"k8s.io/klog/v2"
|
|
|
|
worker "sigs.k8s.io/node-feature-discovery/pkg/nfd-worker"
|
|
"sigs.k8s.io/node-feature-discovery/pkg/utils"
|
|
"sigs.k8s.io/node-feature-discovery/pkg/version"
|
|
)
|
|
|
|
const (
|
|
// ProgramName is the canonical name of this program
|
|
ProgramName = "nfd-worker"
|
|
)
|
|
|
|
func main() {
|
|
flags := flag.NewFlagSet(ProgramName, flag.ExitOnError)
|
|
|
|
printVersion := flags.Bool("version", false, "Print version and exit.")
|
|
|
|
args := parseArgs(flags, os.Args[1:]...)
|
|
|
|
if *printVersion {
|
|
fmt.Println(ProgramName, version.Get())
|
|
os.Exit(0)
|
|
}
|
|
|
|
// Assert that the version is known
|
|
if version.Undefined() {
|
|
klog.Warningf("version not set! Set -ldflags \"-X sigs.k8s.io/node-feature-discovery/pkg/version.version=`git describe --tags --dirty --always`\" during build or run.")
|
|
}
|
|
|
|
// Plug klog into grpc logging infrastructure
|
|
utils.ConfigureGrpcKlog()
|
|
|
|
// Get new NfdWorker instance
|
|
instance, err := worker.NewNfdWorker(args)
|
|
if err != nil {
|
|
klog.Exitf("failed to initialize NfdWorker instance: %v", err)
|
|
}
|
|
|
|
if err = instance.Run(); err != nil {
|
|
klog.Exit(err)
|
|
}
|
|
}
|
|
|
|
func parseArgs(flags *flag.FlagSet, osArgs ...string) *worker.Args {
|
|
args, overrides := initFlags(flags)
|
|
|
|
_ = flags.Parse(osArgs)
|
|
if len(flags.Args()) > 0 {
|
|
fmt.Fprintf(flags.Output(), "unknown command line argument: %s\n", flags.Args()[0])
|
|
flags.Usage()
|
|
os.Exit(2)
|
|
}
|
|
|
|
// Handle overrides
|
|
flags.Visit(func(f *flag.Flag) {
|
|
switch f.Name {
|
|
case "no-publish":
|
|
args.Overrides.NoPublish = overrides.NoPublish
|
|
case "feature-sources":
|
|
args.Overrides.FeatureSources = overrides.FeatureSources
|
|
case "label-sources":
|
|
args.Overrides.LabelSources = overrides.LabelSources
|
|
}
|
|
})
|
|
|
|
return args
|
|
}
|
|
|
|
func initFlags(flagset *flag.FlagSet) (*worker.Args, *worker.ConfigOverrideArgs) {
|
|
args := &worker.Args{}
|
|
|
|
flagset.StringVar(&args.CaFile, "ca-file", "",
|
|
"Root certificate for verifying connections")
|
|
flagset.StringVar(&args.CertFile, "cert-file", "",
|
|
"Certificate used for authenticating connections")
|
|
flagset.StringVar(&args.ConfigFile, "config", "/etc/kubernetes/node-feature-discovery/nfd-worker.conf",
|
|
"Config file to use.")
|
|
flagset.StringVar(&args.KeyFile, "key-file", "",
|
|
"Private key matching -cert-file")
|
|
flagset.BoolVar(&args.EnableNodeFeatureApi, "enable-nodefeature-api", false,
|
|
"Enable the NodeFeature CRD API for communicating with nfd-master. This will automatically disable the gRPC communication.")
|
|
flagset.StringVar(&args.Kubeconfig, "kubeconfig", "",
|
|
"Kubeconfig to use")
|
|
flagset.BoolVar(&args.Oneshot, "oneshot", false,
|
|
"Do not publish feature labels")
|
|
flagset.StringVar(&args.Options, "options", "",
|
|
"Specify config options from command line. Config options are specified "+
|
|
"in the same format as in the config file (i.e. json or yaml). These options")
|
|
flagset.StringVar(&args.Server, "server", "localhost:8080",
|
|
"NFD server address to connecto to.")
|
|
flagset.StringVar(&args.ServerNameOverride, "server-name-override", "",
|
|
"Hostname expected from server certificate, useful in testing")
|
|
|
|
initKlogFlags(flagset, args)
|
|
|
|
// Flags overlapping with config file options
|
|
overrides := &worker.ConfigOverrideArgs{
|
|
FeatureSources: &utils.StringSliceVal{},
|
|
LabelSources: &utils.StringSliceVal{},
|
|
}
|
|
overrides.NoPublish = flagset.Bool("no-publish", false,
|
|
"Do not publish discovered features, disable connection to nfd-master and don't create NodeFeature object.")
|
|
flagset.Var(overrides.FeatureSources, "feature-sources",
|
|
"Comma separated list of feature sources. Special value 'all' enables all sources. "+
|
|
"Prefix the source name with '-' to disable it.")
|
|
flagset.Var(overrides.LabelSources, "label-sources",
|
|
"Comma separated list of label sources. Special value 'all' enables all sources. "+
|
|
"Prefix the source name with '-' to disable it.")
|
|
|
|
return args, overrides
|
|
}
|
|
|
|
func initKlogFlags(flagset *flag.FlagSet, args *worker.Args) {
|
|
args.Klog = make(map[string]*utils.KlogFlagVal)
|
|
|
|
flags := flag.NewFlagSet("klog flags", flag.ContinueOnError)
|
|
//flags.SetOutput(ioutil.Discard)
|
|
klog.InitFlags(flags)
|
|
flags.VisitAll(func(f *flag.Flag) {
|
|
name := klogConfigOptName(f.Name)
|
|
args.Klog[name] = utils.NewKlogFlagVal(f)
|
|
flagset.Var(args.Klog[name], f.Name, f.Usage)
|
|
})
|
|
}
|
|
|
|
func klogConfigOptName(flagName string) string {
|
|
split := strings.Split(flagName, "_")
|
|
for i, v := range split[1:] {
|
|
split[i+1] = strings.ToUpper(v[0:1]) + v[1:]
|
|
}
|
|
return strings.Join(split, "")
|
|
}
|