nfd-worker: Watch features.d changes

Signed-off-by: Oleg Zhurakivskyy <[email protected]>

Author: ozhuraki

pkg/nfd-worker/nfd-worker.go

@@ -30,6 +30,7 @@ import (
 
 	"maps"
 
+	"github.com/fsnotify/fsnotify"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"golang.org/x/net/context"
@@ -55,7 +56,7 @@ import (
 	_ "sigs.k8s.io/node-feature-discovery/source/custom"
 	_ "sigs.k8s.io/node-feature-discovery/source/fake"
 	_ "sigs.k8s.io/node-feature-discovery/source/kernel"
-	_ "sigs.k8s.io/node-feature-discovery/source/local"
+	"sigs.k8s.io/node-feature-discovery/source/local"
 	_ "sigs.k8s.io/node-feature-discovery/source/memory"
 	_ "sigs.k8s.io/node-feature-discovery/source/network"
 	_ "sigs.k8s.io/node-feature-discovery/source/pci"
@@ -121,6 +122,8 @@ type nfdWorker struct {
 	k8sClient           k8sclient.Interface
 	nfdClient           nfdclient.Interface
 	stop                chan struct{} // channel for signaling stop
+	fsEvent             chan fsnotify.Event
+	fsWatcher           *fsnotify.Watcher
 	featureSources      []source.FeatureSource
 	labelSources        []source.LabelSource
 	ownerReference      []metav1.OwnerReference
@@ -304,6 +307,13 @@ func (w *nfdWorker) Run() error {
 	labelTrigger.Reset(w.config.Core.SleepInterval.Duration)
 	defer labelTrigger.Stop()
 
+	if s := source.GetFeatureSource("local"); s != nil {
+		w.fsWatcher = local.FSWatcher
+		if w.fsWatcher != nil {
+			defer w.fsWatcher.Close()
+		}
+	}
+
 	httpMux := http.NewServeMux()
 
 	// Register to metrics server
@@ -341,6 +351,13 @@ func (w *nfdWorker) Run() error {
 				return err
 			}
 
+		case event := <-w.fsWatcher.Events:
+			if event.Op&fsnotify.Create == fsnotify.Create || event.Op&fsnotify.Write == fsnotify.Write || event.Op&fsnotify.Remove == fsnotify.Remove || event.Op&fsnotify.Rename == fsnotify.Rename || event.Op&fsnotify.Chmod == fsnotify.Chmod {
+				err = w.runFeatureDiscovery()
+				if err != nil {
+					return err
+				}
+			}
 		case <-w.stop:
 			klog.InfoS("shutting down nfd-worker")
 			return nil

source/local/local.go

@@ -26,6 +26,7 @@ import (
 
 	"k8s.io/klog/v2"
 
+	"github.com/fsnotify/fsnotify"
 	nfdv1alpha1 "sigs.k8s.io/node-feature-discovery/api/nfd/v1alpha1"
 	"sigs.k8s.io/node-feature-discovery/pkg/utils"
 	"sigs.k8s.io/node-feature-discovery/source"
@@ -63,6 +64,7 @@ const MaxFeatureFileSize = 65536
 // Config
 var (
 	featureFilesDir = "/etc/kubernetes/node-feature-discovery/features.d/"
+	FSWatcher       *fsnotify.Watcher
 )
 
 // localSource implements the FeatureSource and LabelSource interfaces.
@@ -126,6 +128,13 @@ func (s *localSource) GetLabels() (source.FeatureLabels, error) {
 func (s *localSource) Discover() error {
 	s.features = nfdv1alpha1.NewFeatures()
 
+	if FSWatcher == nil {
+		err := watch(s)
+		if err != nil {
+			klog.ErrorS(err, "failed to watch features.d directory")
+		}
+	}
+
 	featuresFromFiles, labelsFromFiles, err := getFeaturesFromFiles()
 	if err != nil {
 		klog.ErrorS(err, "failed to read feature files")
@@ -318,6 +327,30 @@ func getFileContent(fileName string) ([][]byte, error) {
 	return lines, nil
 }
 
+func watch(s *localSource) error {
+	info, err := os.Stat(featureFilesDir)
+	if err != nil {
+		if !os.IsNotExist(err) {
+			return err
+		}
+	}
+
+	if info != nil && info.IsDir() {
+		watcher, err := fsnotify.NewWatcher()
+		if err != nil {
+			return err
+		}
+
+		err = watcher.Add(featureFilesDir)
+		if err != nil {
+			return fmt.Errorf("unable to access %v: %w", featureFilesDir, err)
+		}
+		FSWatcher = watcher
+	}
+
+	return nil
+}
+
 func init() {
 	source.Register(&src)
 }