E2Manager/managers/kubernetes_manager.go

   1 //
   2 // Copyright 2019 AT&T Intellectual Property
   3 // Copyright 2019 Nokia
   4 //
   5 // Licensed under the Apache License, Version 2.0 (the "License");
   6 // you may not use this file except in compliance with the License.
   7 // You may obtain a copy of the License at
   8 //
   9 //      http://www.apache.org/licenses/LICENSE-2.0
  10 //
  11 // Unless required by applicable law or agreed to in writing, software
  12 // distributed under the License is distributed on an "AS IS" BASIS,
  13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 // See the License for the specific language governing permissions and
  15 // limitations under the License.
  16 //
  17
  18 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
  19 //  platform project (RICP).
  20
  21 package managers
  22
  23 import (
  24         "e2mgr/configuration"
  25         "e2mgr/e2managererrors"
  26         "e2mgr/logger"
  27         metaV1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  28         "k8s.io/client-go/kubernetes"
  29         "k8s.io/client-go/tools/clientcmd"
  30         "path/filepath"
  31 )
  32
  33 type KubernetesManager struct {
  34         Logger    *logger.Logger
  35         ClientSet kubernetes.Interface
  36         Config    *configuration.Configuration
  37 }
  38
  39 func NewKubernetesManager(logger *logger.Logger, config *configuration.Configuration) *KubernetesManager {
  40         return &KubernetesManager{
  41                 Logger:    logger,
  42                 ClientSet: createClientSet(logger, config),
  43                 Config:    config,
  44         }
  45 }
  46
  47 /*func (km KubernetesManager) GetAndDeletePod(namespace string, podName string) {
  48         km.logger.Infof("#KubernetesManager.GetAndDeletePod - namespace: %s, POD name: %s ", namespace, podName)
  49
  50         config, err := clientcmd.BuildConfigFromFlags("", "kubeConfigPath")
  51         if err != nil {
  52                 log.Fatal(err)
  53         }
  54
  55         clientSet, _ := kubernetesManager.NewForConfig(config)
  56
  57         podInterface := km.GetPodInterface(clientSet.CoreV1(), namespace, podName)
  58
  59         if podInterface == nil{
  60                 return
  61         }
  62
  63         km.DeletePod(podInterface, podName)
  64 }*/
  65
  66 func createClientSet(logger *logger.Logger, config *configuration.Configuration) kubernetes.Interface {
  67         ////path := os.Getenv("HOME") + "/.kube/config"
  68
  69         absConfigPath,err := filepath.Abs(config.Kubernetes.ConfigPath)
  70         if err != nil {
  71                 logger.Errorf("#KubernetesManager.init - error: %s", err)
  72                 return nil
  73         }
  74
  75         kubernetesConfig, err := clientcmd.BuildConfigFromFlags("", absConfigPath)
  76         if err != nil {
  77                 logger.Errorf("#KubernetesManager.init - error: %s", err)
  78                 return nil
  79         }
  80
  81         clientSet, err := kubernetes.NewForConfig(kubernetesConfig)
  82         if err != nil {
  83                 logger.Errorf("#KubernetesManager.init - error: %s", err)
  84                 return nil
  85         }
  86         return clientSet
  87 }
  88
  89 /*func (km KubernetesManager) DeletePod(podInterface v1.PodInterface, podName string) {
  90         km.logger.Infof("#KubernetesManager.DeletePod - POD name %s ", podName)
  91
  92         err := podInterface.Delete(podName, &metaV1.DeleteOptions{})
  93
  94         if err != nil{
  95                 km.logger.Warnf("#KubernetesManager.DeletePod - POD %s can't be deleted", podName)
  96                 return
  97         }
  98         km.logger.Infof("#KubernetesManager.DeletePod - POD %s was deleted", podName)
  99 }
 100 */
 101 func (km KubernetesManager) DeletePod(podName string) error {
 102         km.Logger.Infof("#KubernetesManager.DeletePod - POD name: %s ", podName)
 103
 104         if km.ClientSet == nil {
 105                 km.Logger.Errorf("#KubernetesManager.DeletePod - no kubernetesManager connection")
 106                 return e2managererrors.NewInternalError()
 107         }
 108
 109         if len(podName) == 0 {
 110                 km.Logger.Warnf("#KubernetesManager.DeletePod - empty pod name")
 111                 return e2managererrors.NewInternalError()
 112         }
 113
 114         err := km.ClientSet.CoreV1().Pods(km.Config.Kubernetes.Namespace).Delete(podName, &metaV1.DeleteOptions{})
 115
 116         if err != nil {
 117                 km.Logger.Errorf("#KubernetesManager.DeletePod - POD %s can't be deleted, error: %s", podName, err)
 118                 return err
 119         }
 120
 121         km.Logger.Infof("#KubernetesManager.DeletePod - POD %s was deleted", podName)
 122         return nil
 123 }
 124
 125 /*func (km KubernetesManager) GetPodInterface(client v1.CoreV1Interface, namespace string, podName string) v1.PodInterface{
 126         km.logger.Infof("#KubernetesManager.GetPodInterface - namespace: %s, POD name: %s ", namespace, podName)
 127
 128
 129         podInterface := client.Pods(namespace)
 130         pod, err := podInterface.Get(podName, metaV1.GetOptions{})
 131
 132         if err != nil{
 133                 km.logger.Warnf("#KubernetesManager.GetPodInterface - POD name: %s not found", podName)
 134                 return nil
 135         }
 136
 137         km.logger.Infof("#KubernetesManager.GetPodInterface - POD status: %s ", pod.Status.String())
 138
 139         return podInterface
 140 }*/