mirror of
https://github.com/kubernetes-sigs/node-feature-discovery.git
synced 2024-12-15 17:50:49 +00:00
a00cdc2b61
Refactor the code, moving the hostpath helper functionality to new "pkg/utils/hostpath" package. This breaks odd-ish dependency "pkg/utils" -> "source".
124 lines
4.4 KiB
Go
124 lines
4.4 KiB
Go
/*
|
|
Copyright 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"
|
|
"time"
|
|
|
|
"k8s.io/klog/v2"
|
|
|
|
"sigs.k8s.io/node-feature-discovery/pkg/kubeconf"
|
|
topology "sigs.k8s.io/node-feature-discovery/pkg/nfd-client/topology-updater"
|
|
"sigs.k8s.io/node-feature-discovery/pkg/resourcemonitor"
|
|
"sigs.k8s.io/node-feature-discovery/pkg/topologypolicy"
|
|
"sigs.k8s.io/node-feature-discovery/pkg/utils"
|
|
"sigs.k8s.io/node-feature-discovery/pkg/utils/hostpath"
|
|
"sigs.k8s.io/node-feature-discovery/pkg/version"
|
|
)
|
|
|
|
const (
|
|
// ProgramName is the canonical name of this program
|
|
ProgramName = "nfd-topology-updater"
|
|
)
|
|
|
|
func main() {
|
|
flags := flag.NewFlagSet(ProgramName, flag.ExitOnError)
|
|
|
|
printVersion := flags.Bool("version", false, "Print version and exit.")
|
|
|
|
args, resourcemonitorArgs := 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()
|
|
|
|
klConfig, err := kubeconf.GetKubeletConfigFromLocalFile(resourcemonitorArgs.KubeletConfigFile)
|
|
if err != nil {
|
|
klog.Exitf("error reading kubelet config: %v", err)
|
|
}
|
|
tmPolicy := string(topologypolicy.DetectTopologyPolicy(klConfig.TopologyManagerPolicy, klConfig.TopologyManagerScope))
|
|
klog.Infof("detected kubelet Topology Manager policy %q", tmPolicy)
|
|
|
|
// Get new TopologyUpdater instance
|
|
instance, err := topology.NewTopologyUpdater(*args, *resourcemonitorArgs, tmPolicy)
|
|
if err != nil {
|
|
klog.Exitf("failed to initialize TopologyUpdater instance: %v", err)
|
|
}
|
|
|
|
if err = instance.Run(); err != nil {
|
|
klog.Exit(err)
|
|
}
|
|
}
|
|
|
|
func parseArgs(flags *flag.FlagSet, osArgs ...string) (*topology.Args, *resourcemonitor.Args) {
|
|
args, resourcemonitorArgs := 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)
|
|
}
|
|
|
|
return args, resourcemonitorArgs
|
|
}
|
|
|
|
func initFlags(flagset *flag.FlagSet) (*topology.Args, *resourcemonitor.Args) {
|
|
args := &topology.Args{}
|
|
resourcemonitorArgs := &resourcemonitor.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.KeyFile, "key-file", "",
|
|
"Private key matching -cert-file")
|
|
flagset.BoolVar(&args.Oneshot, "oneshot", false,
|
|
"Update once and exit")
|
|
flagset.BoolVar(&args.NoPublish, "no-publish", false,
|
|
"Do not publish discovered features to the cluster-local Kubernetes API server.")
|
|
flagset.StringVar(&args.KubeConfigFile, "kubeconfig", "",
|
|
"Kube config file.")
|
|
flagset.DurationVar(&resourcemonitorArgs.SleepInterval, "sleep-interval", time.Duration(60)*time.Second,
|
|
"Time to sleep between CR updates. Non-positive value implies no CR updatation (i.e. infinite sleep). [Default: 60s]")
|
|
flagset.StringVar(&resourcemonitorArgs.Namespace, "watch-namespace", "*",
|
|
"Namespace to watch pods (for testing/debugging purpose). Use * for all namespaces.")
|
|
flagset.StringVar(&resourcemonitorArgs.KubeletConfigFile, "kubelet-config-file", hostpath.VarDir.Path("lib/kubelet/config.yaml"),
|
|
"Kubelet config file path.")
|
|
flagset.StringVar(&resourcemonitorArgs.PodResourceSocketPath, "podresources-socket", hostpath.VarDir.Path("lib/kubelet/pod-resources/kubelet.sock"),
|
|
"Pod Resource Socket path to use.")
|
|
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")
|
|
|
|
klog.InitFlags(flagset)
|
|
|
|
return args, resourcemonitorArgs
|
|
}
|