pkg/nbi/control.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    This source code is part of the near-RT RIC (RAN Intelligent Controller)
  19    platform project (RICP).
  20
  21 ==================================================================================
  22 */
  23 package nbi
  24
  25 import "C"
  26
  27 import (
  28         "errors"
  29         "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
  30         "routing-manager/pkg/rpe"
  31         "routing-manager/pkg/rtmgr"
  32         "routing-manager/pkg/sbi"
  33         "routing-manager/pkg/sdl"
  34         "strconv"
  35         "sync"
  36         "time"
  37         "os"
  38         "fmt"
  39 )
  40
  41 var m sync.Mutex
  42
  43 var nbiEngine Engine
  44 var sbiEngine sbi.Engine
  45 var sdlEngine sdl.Engine
  46 var rpeEngine rpe.Engine
  47
  48 const INTERVAL time.Duration = 60
  49
  50 func NewControl() Control {
  51         return Control{make(chan *xapp.RMRParams)}
  52 }
  53
  54 type Control struct {
  55         rcChan chan *xapp.RMRParams
  56 }
  57
  58
  59 func (c *Control) Run() {
  60         var err error
  61         go c.controlLoop()
  62         nbiEngine, sbiEngine, sdlEngine, rpeEngine, err = initRtmgr()
  63         if err != nil {
  64                 xapp.Logger.Error(err.Error())
  65                 os.Exit(1)
  66         }
  67         xapp.Run(c)
  68 }
  69
  70 func (c *Control) Consume(rp *xapp.RMRParams) (err error) {
  71         c.rcChan <- rp
  72         return
  73 }
  74
  75 func initRtmgr() (nbiEngine Engine, sbiEngine sbi.Engine, sdlEngine sdl.Engine, rpeEngine rpe.Engine, err error) {
  76         if nbiEngine, err = GetNbi(xapp.Config.GetString("nbi")); err == nil && nbiEngine != nil {
  77                 if sbiEngine, err = sbi.GetSbi(xapp.Config.GetString("sbi")); err == nil && sbiEngine != nil {
  78                         if sdlEngine, err = sdl.GetSdl(xapp.Config.GetString("sdl")); err == nil && sdlEngine != nil {
  79                                 if rpeEngine, err = rpe.GetRpe(xapp.Config.GetString("rpe")); err == nil && rpeEngine != nil {
  80         fmt.Printf("%v,%v,%v,%v",nbiEngine, sbiEngine, sdlEngine, rpeEngine)
  81                                         return nbiEngine, sbiEngine, sdlEngine, rpeEngine, nil
  82                                 }
  83                         }
  84                 }
  85         }
  86         return nil, nil, nil, nil, err
  87 }
  88
  89 func (c *Control) controlLoop() {
  90         for {
  91                 msg := <-c.rcChan
  92                 c.recievermr(msg)
  93                 /*
  94                 xapp_msg := sbi.RMRParams{msg}
  95                 switch msg.Mtype {
  96                 case xapp.RICMessageTypes["RMRRM_REQ_TABLE"]:
  97                         if rtmgr.Rtmgr_ready == false {
  98                                 xapp.Logger.Info("Update Route Table Request(RMR to RM), message discarded as routing manager is not ready")
  99                         } else {
 100                                 xapp.Logger.Info("Update Route Table Request(RMR to RM)")
 101                                 go c.handleUpdateToRoutingManagerRequest(msg)
 102                         }
 103                 case xapp.RICMessageTypes["RMRRM_TABLE_STATE"]:
 104                         xapp.Logger.Info("state of table to route mgr %s,payload %s", xapp_msg.String(), msg.Payload)
 105
 106                 default:
 107                         err := errors.New("Message Type " + strconv.Itoa(msg.Mtype) + " is discarded")
 108                         xapp.Logger.Error("Unknown message type: %v", err)
 109                 }
 110                 xapp.Rmr.Free(msg.Mbuf)*/
 111         }
 112 }
 113
 114 func (c *Control) recievermr(msg *xapp.RMRParams) {
 115         xapp_msg := sbi.RMRParams{msg}
 116         switch msg.Mtype {
 117         case xapp.RICMessageTypes["RMRRM_REQ_TABLE"]:
 118         if rtmgr.Rtmgr_ready == false {
 119                 xapp.Logger.Info("Update Route Table Request(RMR to RM), message discarded as routing manager is not ready")
 120         } else {
 121                 xapp.Logger.Info("Update Route Table Request(RMR to RM)")
 122                 go c.handleUpdateToRoutingManagerRequest(msg)
 123         }
 124         case xapp.RICMessageTypes["RMRRM_TABLE_STATE"]:
 125                 xapp.Logger.Info("state of table to route mgr %s,payload %s", xapp_msg.String(), msg.Payload)
 126         default:
 127                 err := errors.New("Message Type " + strconv.Itoa(msg.Mtype) + " is discarded")
 128                 xapp.Logger.Error("Unknown message type: %v", err)
 129         }
 130         xapp.Rmr.Free(msg.Mbuf)
 131 }
 132
 133 func (c *Control) handleUpdateToRoutingManagerRequest(params *xapp.RMRParams) {
 134
 135         msg := sbi.RMRParams{params}
 136
 137         xapp.Logger.Info("Update Route Table Request, msg.String() : %s", msg.String())
 138         xapp.Logger.Info("Update Route Table Request, params.Payload : %s", string(params.Payload))
 139
 140         m.Lock()
 141         data, err := sdlEngine.ReadAll(xapp.Config.GetString("rtfile"))
 142         m.Unlock()
 143         if err != nil || data == nil {
 144                 xapp.Logger.Error("Cannot get data from sdl interface due to: " + err.Error())
 145                 return
 146         }
 147
 148         ep := sbiEngine.CreateEndpoint(string(params.Payload))
 149         if ep == nil {
 150                 xapp.Logger.Error("Update Routing Table Request can't handle due to end point %s is not avail in complete ep list: ", string(params.Payload))
 151                 return
 152         }
 153
 154         policies := rpeEngine.GeneratePolicies(rtmgr.Eps, data)
 155         err = sbiEngine.DistributeToEp(policies, ep)
 156         if err != nil {
 157                 xapp.Logger.Error("Routing table cannot be published due to: " + err.Error())
 158                 return
 159         }
 160 }
 161
 162 func sendRoutesToAll() (err error) {
 163
 164         m.Lock()
 165         data, err := sdlEngine.ReadAll(xapp.Config.GetString("rtfile"))
 166         fmt.Printf("data = %v,%v,%v",data,sdlEngine,sbiEngine)
 167         m.Unlock()
 168         if err != nil || data == nil {
 169                 return errors.New("Cannot get data from sdl interface due to: " + err.Error())
 170         }
 171         if sbiEngine == nil {
 172                 fmt.Printf("SBI is nil")
 173         }
 174         sbiEngine.UpdateEndpoints(data)
 175         policies := rpeEngine.GeneratePolicies(rtmgr.Eps, data)
 176         err = sbiEngine.DistributeAll(policies)
 177         if err != nil {
 178                 return errors.New("Routing table cannot be published due to: " + err.Error())
 179         }
 180
 181         return nil
 182 }
 183
 184 func Serve() {
 185
 186         nbiErr := nbiEngine.Initialize(xapp.Config.GetString("xmurl"), xapp.Config.GetString("nbiurl"), xapp.Config.GetString("rtfile"), xapp.Config.GetString("cfgfile"), xapp.Config.GetString("e2murl"), sdlEngine, rpeEngine, &m)
 187         if nbiErr != nil {
 188                 xapp.Logger.Error("Failed to initialize nbi due to: " + nbiErr.Error())
 189                 return
 190         }
 191
 192         err := sbiEngine.Initialize(xapp.Config.GetString("sbiurl"))
 193         if err != nil {
 194                 xapp.Logger.Info("Failed to open push socket due to: " + err.Error())
 195                 return
 196         }
 197         defer nbiEngine.Terminate()
 198         defer sbiEngine.Terminate()
 199
 200         for {
 201                 sendRoutesToAll()
 202
 203                 rtmgr.Rtmgr_ready = true
 204                 time.Sleep(INTERVAL * time.Second)
 205                 xapp.Logger.Debug("Periodic loop timed out. Setting triggerSBI flag to distribute updated routes.")
 206         }
 207 }