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

Signed-off-by: Cyclinder Kuo <kuocyclinder@gmail.com>

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")

pkg/resources/resources_manager_test.go

@@ -53,6 +53,7 @@ func (l *FakeLink) Type() string {
 var _ = Describe("ResourcesManger", func() {
 	Context("NewResourceManager", func() {
 		const activeSockDirBackUP = "/var/lib/kubelet/plugins_registry"
+		const kubeletRootDir = "/var/lib/kubelet"
 
 		It("Resource Manager with watcher mode", func() {
 			fs := utils.FakeFilesystem{
@@ -64,7 +65,7 @@ var _ = Describe("ResourcesManger", func() {
 				activeSockDir = activeSockDirBackUP
 			}()
 
-			obj := NewResourceManager(DefaultConfigFilePath, false)
+			obj := NewResourceManager(DefaultConfigFilePath, kubeletRootDir, false)
 			rm := obj.(*resourceManager)
 			Expect(rm.watchMode).To(Equal(true))
 		})
@@ -76,7 +77,7 @@ var _ = Describe("ResourcesManger", func() {
 				activeSockDir = activeSockDirBackUP
 			}()
 
-			obj := NewResourceManager(DefaultConfigFilePath, false)
+			obj := NewResourceManager(DefaultConfigFilePath, kubeletRootDir, false)
 			rm := obj.(*resourceManager)
 			Expect(rm.watchMode).To(Equal(false))
 		})