enable socket communication also for GPU plugin

Signed-off-by: Diego Ciangottini <[email protected]>
interTwin-eu · Oct 16, 2024 · 2e70a73 · 2e70a73
1 parent d875119
commit 2e70a73
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -2,9 +2,13 @@ package main
 
 import (
 	"context"
+	"net"
 	"net/http"
 	"os"
+	"os/signal"
 	"strconv"
+	"strings"
+	"syscall"
 
 	"github.com/sirupsen/logrus"
 	"github.com/virtual-kubelet/virtual-kubelet/log"
@@ -114,6 +118,7 @@ func main() {
 		if err != nil {
 			log.G(Ctx).Fatal(err)
 		}
+	}
 
 	if err != nil {
 		log.G(Ctx).Fatal(err)