X-Git-Url: https://gerrit.akraino.org/r/gitweb?a=blobdiff_plain;ds=sidebyside;f=mecm%2Fmepm%2Fapplcm%2Fk8shelm%2Fcmd%2Fhelm%2Fmain.go;h=e2ba78b36ee9be642480d4ce8ed2c3ba78eafede;hb=3de2bab9378d427f1bb43d9c517c898cdc936461;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..e2ba78b 100644
--- a/mecm/mepm/applcm/k8shelm/cmd/helm/main.go
+++ b/mecm/mepm/applcm/k8shelm/cmd/helm/main.go
@@ -17,20 +17,40 @@
 package main
 
 import (
+	"github.com/sirupsen/logrus"
 	"k8shelm/pkg/plugin"
-	"log"
 	"os"
+	"strconv"
+)
+
+// 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)
 	}
 }