cmd/rtmgr.go

   1 /*
   2 ==================================================================================
   3    Copyright (c) 2019 AT&T Intellectual Property.
   4    Copyright (c) 2019 Nokia
   5
   6    Licensed under the Apache License, Version 2.0 (the "License");
   7    you may not use this file except in compliance with the License.
   8    You may obtain a copy of the License at
   9
  10        http://www.apache.org/licenses/LICENSE-2.0
  11
  12    Unless required by applicable law or agreed to in writing, software
  13    distributed under the License is distributed on an "AS IS" BASIS,
  14    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15    See the License for the specific language governing permissions and
  16    limitations under the License.
  17
  18
  19    This source code is part of the near-RT RIC (RAN Intelligent Controller)
  20    platform project (RICP).
  21
  22
  23 ==================================================================================
  24 */
  25 /*
  26         Mnemonic:       rtmgr.go
  27         Abstract:       Routing Manager Main file. Implemets the following functions:
  28                         - parseArgs: reading command line arguments
  29                         - init:Rtmgr initializing the service modules
  30                         - serve: running the loop
  31         Date:           12 March 2019
  32 */
  33 package main
  34
  35 //TODO: change flag to pflag (won't need any argument parse)
  36
  37 import (
  38         "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
  39         "os"
  40         "os/signal"
  41         "routing-manager/pkg/nbi"
  42         "routing-manager/pkg/rpe"
  43         "routing-manager/pkg/rtmgr"
  44         "routing-manager/pkg/sbi"
  45         "routing-manager/pkg/sdl"
  46         "syscall"
  47         "time"
  48         "sync"
  49 )
  50
  51 const SERVICENAME = "rtmgr"
  52 const INTERVAL time.Duration = 60
  53
  54 func initRtmgr() (nbiEngine nbi.Engine, sbiEngine sbi.Engine, sdlEngine sdl.Engine, rpeEngine rpe.Engine, err error) {
  55         if nbiEngine, err = nbi.GetNbi(xapp.Config.GetString("nbi")); err == nil && nbiEngine != nil {
  56                 if sbiEngine, err = sbi.GetSbi(xapp.Config.GetString("sbi")); err == nil && sbiEngine != nil {
  57                         if sdlEngine, err = sdl.GetSdl(xapp.Config.GetString("sdl")); err == nil && sdlEngine != nil {
  58                                 if rpeEngine, err = rpe.GetRpe(xapp.Config.GetString("rpe")); err == nil && rpeEngine != nil {
  59                                         return nbiEngine, sbiEngine, sdlEngine, rpeEngine, nil
  60                                 }
  61                         }
  62                 }
  63         }
  64         return nil, nil, nil, nil, err
  65 }
  66
  67 func serveSBI(triggerSBI <-chan bool, sbiEngine sbi.Engine, sdlEngine sdl.Engine, rpeEngine rpe.Engine, m *sync.Mutex) {
  68         for {
  69                 if <-triggerSBI {
  70                         m.Lock()
  71                         data, err := sdlEngine.ReadAll(xapp.Config.GetString("rtfile"))
  72                         m.Unlock()
  73                         if err != nil || data == nil {
  74                                 xapp.Logger.Error("Cannot get data from sdl interface due to: " + err.Error())
  75                                 continue
  76                         }
  77                         sbiEngine.UpdateEndpoints(data)
  78                         policies := rpeEngine.GeneratePolicies(rtmgr.Eps, data)
  79                         err = sbiEngine.DistributeAll(policies)
  80                         if err != nil {
  81                                 xapp.Logger.Error("Routing table cannot be published due to: " + err.Error())
  82                         }
  83                 }
  84         }
  85 }
  86
  87 func serve(nbiEngine nbi.Engine, sbiEngine sbi.Engine, sdlEngine sdl.Engine, rpeEngine rpe.Engine, m *sync.Mutex) {
  88
  89         triggerSBI := make(chan bool)
  90
  91         nbiErr := nbiEngine.Initialize(xapp.Config.GetString("xmurl"), xapp.Config.GetString("nbiurl"), xapp.Config.GetString("rtfile"), xapp.Config.GetString("cfgfile"),
  92                 sdlEngine, rpeEngine, triggerSBI, m)
  93         if nbiErr != nil {
  94                 xapp.Logger.Error("Failed to initialize nbi due to: " + nbiErr.Error())
  95                 return
  96         }
  97
  98         err := sbiEngine.Initialize(xapp.Config.GetString("sbiurl"))
  99         if err != nil {
 100                 xapp.Logger.Info("Failed to open push socket due to: " + err.Error())
 101                 return
 102         }
 103         defer nbiEngine.Terminate()
 104         defer sbiEngine.Terminate()
 105
 106         // This SBI Go routine is trtiggered by periodic main loop and when data is recieved on REST interface.
 107         go serveSBI(triggerSBI, sbiEngine, sdlEngine, rpeEngine, m)
 108
 109         for {
 110                 if xapp.Config.GetString("nbi") == "httpGetter" {
 111                         data, err := nbiEngine.(*nbi.HttpGetter).FetchAllXApps(xapp.Config.GetString("xmurl"))
 112                         if err != nil {
 113                                 xapp.Logger.Error("Cannot fetch xapp data due to: " + err.Error())
 114                         } else if data != nil {
 115                                 sdlEngine.WriteXApps(xapp.Config.GetString("rtfile"), data)
 116                         }
 117                 }
 118
 119                 triggerSBI <- true
 120
 121                 time.Sleep(INTERVAL * time.Second)
 122                 xapp.Logger.Debug("Periodic loop timed out. Setting triggerSBI flag to distribute updated routes.")
 123         }
 124 }
 125
 126 func SetupCloseHandler() {
 127         c := make(chan os.Signal, 2)
 128         signal.Notify(c, os.Interrupt, syscall.SIGTERM)
 129         go func() {
 130                 <-c
 131                 xapp.Logger.Info("\r- Ctrl+C pressed in Terminal")
 132                 os.Exit(0)
 133         }()
 134 }
 135
 136 func main() {
 137         nbiEngine, sbiEngine, sdlEngine, rpeEngine, err := initRtmgr()
 138         if err != nil {
 139                 xapp.Logger.Error(err.Error())
 140                 os.Exit(1)
 141         }
 142         SetupCloseHandler()
 143         xapp.Logger.Info("Start " + SERVICENAME + " service")
 144         rtmgr.Eps = make(rtmgr.Endpoints)
 145
 146         var m sync.Mutex
 147
 148         serve(nbiEngine, sbiEngine, sdlEngine, rpeEngine, &m)
 149         os.Exit(0)
 150 }