mecm/mepm/applcm/k8shelm/cmd/helm/main.go

   1 /*
   2  * Copyright 2020 Huawei Technologies Co., Ltd.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package main
  18
  19 import (
  20         "k8shelm/pkg/plugin"
  21         "os"
  22         "strconv"
  23
  24         "github.com/sirupsen/logrus"
  25 )
  26
  27 // Variables to be defined in deployment file
  28 var (
  29         serverPort  = os.Getenv("HELM_PLUGIN_PORT")
  30         logFile     = os.Getenv("LOGFILE_PATH")
  31         loggerLevel = os.Getenv("LOGGER_LEVEL")
  32         certificate = os.Getenv("CERTIFICATE_PATH")
  33         key         = os.Getenv("KEY_PATH")
  34 )
  35
  36 func main() {
  37         // Prepare logger
  38         file, err := os.Create(logFile)
  39         if err != nil {
  40                 logrus.Fatal(err)
  41         }
  42         defer file.Close()
  43
  44         level, err := logrus.ParseLevel(loggerLevel)
  45         var logger = plugin.GetLogger(logFile, level, file)
  46
  47         // Create GRPC server
  48         sp, err := strconv.Atoi(serverPort)
  49         serverConfig := plugin.ServerGRPCConfig{Certificate: certificate, Port: sp, Key: key, Logger: logger}
  50         server := plugin.NewServerGRPC(serverConfig)
  51
  52         // Start listening
  53         err = server.Listen()
  54         if err != nil {
  55                 logger.Fatalf("failed to listen: %v", err)
  56         }
  57 }