Add '--kubelet-root-dir' flag to configure the root directory of kubelet

Author: cyclinder

cmd/k8s-rdma-shared-dp/main.go

@@ -44,12 +44,13 @@ func main() {
 
 	// add version flag
 	versionOpt := false
-	var configFilePath string
+	useCdi := false
+	var configFilePath, kubeletRootDir string
 	flag.BoolVar(&versionOpt, "version", false, "Show application version")
 	flag.BoolVar(&versionOpt, "v", false, "Show application version")
 	flag.StringVar(
 		&configFilePath, "config-file", resources.DefaultConfigFilePath, "path to device plugin config file")
-	useCdi := false
+	flag.StringVar(&kubeletRootDir, "kubelet-root-dir", "/var/lib/kubelet", "root directory of kubelet")
 	flag.BoolVar(&useCdi, "use-cdi", false,
 		"Use Container Device Interface to expose devices in containers")
 	flag.Parse()
@@ -64,7 +65,7 @@ func main() {
 
 	log.Println("Starting K8s RDMA Shared Device Plugin version=", version)
 
-	rm := resources.NewResourceManager(configFilePath, useCdi)
+	rm := resources.NewResourceManager(configFilePath, kubeletRootDir, useCdi)
 
 	log.Println("resource manager reading configs")
 	if err := rm.ReadConfig(); err != nil {

pkg/resources/resources_manager.go

@@ -23,6 +23,7 @@ import (
 	"fmt"
 	"log"
 	"os"
+	"path"
 	"regexp"
 	"strconv"
 	"time"
@@ -74,7 +75,12 @@ type resourceManager struct {
 	useCdi                 bool
 }
 
-func NewResourceManager(configFile string, useCdi bool) types.ResourceManager {
+func NewResourceManager(configFile, kubeletRootDir string, useCdi bool) types.ResourceManager {
+	if kubeletRootDir != "" {
+		activeSockDir = path.Join(kubeletRootDir, "plugins_registry")
+		deprecatedSockDir = path.Join(kubeletRootDir, "device-plugins")
+	}
+
 	watcherMode := detectPluginWatchMode(activeSockDir)
 	if watcherMode {
 		fmt.Println("Using Kubelet Plugin Registry Mode")