X-Git-Url: https://gerrit.akraino.org/r/gitweb?a=blobdiff_plain;f=mecm%2Fmepm%2Fapplcm%2Fk8shelm%2Fcmd%2Fhelm%2Fmain.go;h=84b2a123e50f1a825f4c1a44a9f377fc4e0888cc;hb=650f6ee3da7d8af68e3cbd58914731892e4cb70b;hp=adcba84d11cf046cb57e8502ae39dad5011d2f32;hpb=e0237ae53c473ceb84d9f84d85d4631714cdc292;p=ealt-edge.git

diff --git a/mecm/mepm/applcm/k8shelm/cmd/helm/main.go b/mecm/mepm/applcm/k8shelm/cmd/helm/main.go
index adcba84..84b2a12 100644
--- a/mecm/mepm/applcm/k8shelm/cmd/helm/main.go
+++ b/mecm/mepm/applcm/k8shelm/cmd/helm/main.go
@@ -18,19 +18,40 @@ package main
 
 import (
 	"k8shelm/pkg/plugin"
-	"log"
 	"os"
+	"strconv"
+
+	"github.com/sirupsen/logrus"
+)
+
+// Variables to be defined in deployment file
+var (
+	serverPort  = os.Getenv("HELM_PLUGIN_PORT")
+	logFile     = os.Getenv("LOGFILE_PATH")
+	loggerLevel = os.Getenv("LOGGER_LEVEL")
+	certificate = os.Getenv("CERTIFICATE_PATH")
+	key         = os.Getenv("KEY_PATH")
 )
 
 func main() {
-	logger := log.New(os.Stdout, "helm ", log.LstdFlags|log.Lshortfile)
-	serverConfig := plugin.ServerGRPCConfig{Certificate:"", Port:50051, Key:""}
-	server, err := plugin.NewServerGRPC(serverConfig)
+	// Prepare logger
+	file, err := os.Create(logFile)
 	if err != nil {
-		logger.Fatalf("failed to create server: %v", err)
+		logrus.Fatal(err)
 	}
-	error := server.Listen()
-	if error != nil {
+	defer file.Close()
+
+	level, err := logrus.ParseLevel(loggerLevel)
+	var logger = plugin.GetLogger(logFile, level, file)
+
+	// Create GRPC server
+	sp, err := strconv.Atoi(serverPort)
+	serverConfig := plugin.ServerGRPCConfig{Certificate: certificate, Port: sp, Key: key, Logger: logger}
+	server := plugin.NewServerGRPC(serverConfig)
+
+	// Start listening
+	err = server.Listen()
+	if err != nil {
 		logger.Fatalf("failed to listen: %v", err)
 	}
 }