Add node env variable and revert cri-api to v1alpha2

Signed-off-by: ghokun <[email protected]>

Author: ghokun

manifests/device-plugin.yml

@@ -45,7 +45,7 @@ spec:
             capabilities:
               add: ["SYS_ADMIN"]
       containers:
-        - image: ghcr.io/kuartis/kuartis-virtual-gpu-device-plugin:0.4.1
+        - image: ghcr.io/kuartis/kuartis-virtual-gpu-device-plugin:0.4.2
           name: kuartis-virtual-gpu-device-plugin-ctr
           command:
             - /usr/bin/virtual-gpu-device-plugin
@@ -55,6 +55,11 @@ spec:
             allowPrivilegeEscalation: false
             capabilities:
               drop: ["ALL"]
+          env:
+            - name: NODE_NAME
+              valueFrom:
+                fieldRef:
+                  fieldPath: spec.nodeName
           ports:
             - containerPort: 8080
               name: metrics

pkg/gpu/nvidia/metrics.go

@@ -16,14 +16,16 @@ import (
 
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials/insecure"
-	pb "k8s.io/cri-api/pkg/apis/runtime/v1"
+	pb "k8s.io/cri-api/pkg/apis/runtime/v1alpha2"
 )
 
 const (
 	containerdsocket = "/var/run/containerd/containerd.sock"
 	timeout          = 10 * time.Second
 )
 
+var node = os.Getenv("NODE_NAME")
+
 var metricsFormat = `
 # HELP container_per_gpu
 # TYPE container_per_gpu gauge
@@ -75,7 +77,7 @@ func collectMetrics(w http.ResponseWriter, r *http.Request) {
 	containerMap := make(map[string]containerInfo)
 	for _, container := range containers.GetContainers() {
 		containerMap[container.Id] = containerInfo{
-			Node:        "",
+			Node:        node,
 			Namespace:   container.Labels["io.kubernetes.pod.namespace"],
 			Pod:         container.Labels["io.kubernetes.pod.name"],
 			PodUid:      container.Labels["io.kubernetes.pod.uid"],