Resetting E2Term connection flag after connection succes
[ric-plt/rtmgr.git] / pkg / nbi / control.go
index 5483aba..98d7821 100644 (file)
@@ -26,18 +26,23 @@ import "C"
 
 import (
        "errors"
+       //"fmt"
        "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
+       "net/http"
+       "os"
+       "routing-manager/pkg/models"
        "routing-manager/pkg/rpe"
        "routing-manager/pkg/rtmgr"
        "routing-manager/pkg/sbi"
        "routing-manager/pkg/sdl"
        "strconv"
+       "strings"
        "sync"
        "time"
-       "os"
 )
 
 var m sync.Mutex
+var EndpointLock sync.Mutex
 
 var nbiEngine Engine
 var sbiEngine sbi.Engine
@@ -54,57 +59,67 @@ type Control struct {
        rcChan chan *xapp.RMRParams
 }
 
-
 func (c *Control) Run() {
        var err error
        go c.controlLoop()
        nbiEngine, sbiEngine, sdlEngine, rpeEngine, err = initRtmgr()
        if err != nil {
-                xapp.Logger.Error(err.Error())
-                os.Exit(1)
-        }
+               xapp.Logger.Error(err.Error())
+               os.Exit(1)
+       }
+
+       xapp.Resource.InjectRoute("/ric/v1/symptomdata", c.SymptomDataHandler, "GET")
+
        xapp.Run(c)
 }
 
+func (c *Control) SymptomDataHandler(w http.ResponseWriter, r *http.Request) {
+       resp, _ := DumpDebugData()
+       xapp.Resource.SendSymptomDataJson(w, r, resp, "platform/rttable.json")
+}
+
 func (c *Control) Consume(rp *xapp.RMRParams) (err error) {
        c.rcChan <- rp
        return
 }
 
 func initRtmgr() (nbiEngine Engine, sbiEngine sbi.Engine, sdlEngine sdl.Engine, rpeEngine rpe.Engine, err error) {
-        if nbiEngine, err = GetNbi(xapp.Config.GetString("nbi")); err == nil && nbiEngine != nil {
-                if sbiEngine, err = sbi.GetSbi(xapp.Config.GetString("sbi")); err == nil && sbiEngine != nil {
-                        if sdlEngine, err = sdl.GetSdl(xapp.Config.GetString("sdl")); err == nil && sdlEngine != nil {
-                                if rpeEngine, err = rpe.GetRpe(xapp.Config.GetString("rpe")); err == nil && rpeEngine != nil {
-                                        return nbiEngine, sbiEngine, sdlEngine, rpeEngine, nil
-                                }
-                        }
-                }
-        }
-        return nil, nil, nil, nil, err
+       if nbiEngine, err = GetNbi(xapp.Config.GetString("nbi")); err == nil && nbiEngine != nil {
+               if sbiEngine, err = sbi.GetSbi(xapp.Config.GetString("sbi")); err == nil && sbiEngine != nil {
+                       if sdlEngine, err = sdl.GetSdl(xapp.Config.GetString("sdl")); err == nil && sdlEngine != nil {
+                               if rpeEngine, err = rpe.GetRpe(xapp.Config.GetString("rpe")); err == nil && rpeEngine != nil {
+                                       return nbiEngine, sbiEngine, sdlEngine, rpeEngine, nil
+                               }
+                       }
+               }
+       }
+       return nil, nil, nil, nil, err
 }
 
 func (c *Control) controlLoop() {
        for {
                msg := <-c.rcChan
-               xapp_msg := sbi.RMRParams{msg}
-               switch msg.Mtype {
-               case xapp.RICMessageTypes["RMRRM_REQ_TABLE"]:
-                       if rtmgr.Rtmgr_ready == false {
-                               xapp.Logger.Info("Update Route Table Request(RMR to RM), message discarded as routing manager is not ready")
-                       } else {
-                               xapp.Logger.Info("Update Route Table Request(RMR to RM)")
-                               go c.handleUpdateToRoutingManagerRequest(msg)
-                       }
-               case xapp.RICMessageTypes["RMRRM_TABLE_STATE"]:
-                       xapp.Logger.Info("state of table to route mgr %s,payload %s", xapp_msg.String(), msg.Payload)
+               c.recievermr(msg)
+       }
+}
 
-               default:
-                       err := errors.New("Message Type " + strconv.Itoa(msg.Mtype) + " is discarded")
-                       xapp.Logger.Error("Unknown message type: %v", err)
+func (c *Control) recievermr(msg *xapp.RMRParams) {
+       xapp_msg := sbi.RMRParams{msg}
+       switch msg.Mtype {
+       case xapp.RICMessageTypes["RMRRM_REQ_TABLE"]:
+               if rtmgr.Rtmgr_ready == false {
+                       xapp.Logger.Info("Update route Table Request(RMR -> RM), message discarded as routing manager is not ready")
+               } else {
+                       xapp.Logger.Info("Update Route Table Request(RMR -> RM)")
+                       go c.handleUpdateToRoutingManagerRequest(msg)
                }
-               xapp.Rmr.Free(msg.Mbuf)
+       case xapp.RICMessageTypes["RMRRM_TABLE_STATE"]:
+               xapp.Logger.Info("State of route table to route mgr %s,payload %s", xapp_msg.String(), msg.Payload)
+       default:
+               err := errors.New("Message Type " + strconv.Itoa(msg.Mtype) + " is discarded")
+               xapp.Logger.Error("Unknown message type: %v", err)
        }
+       xapp.Rmr.Free(msg.Mbuf)
 }
 
 func (c *Control) handleUpdateToRoutingManagerRequest(params *xapp.RMRParams) {
@@ -117,64 +132,132 @@ func (c *Control) handleUpdateToRoutingManagerRequest(params *xapp.RMRParams) {
        m.Lock()
        data, err := sdlEngine.ReadAll(xapp.Config.GetString("rtfile"))
        m.Unlock()
-       if err != nil || data == nil {
-               xapp.Logger.Error("Cannot get data from sdl interface due to: " + err.Error())
-               return
+       if data == nil {
+               if err != nil {
+                       xapp.Logger.Error("Cannot get data from sdl interface due to: " + err.Error())
+                       return
+               } else {
+                       xapp.Logger.Debug("Cannot get data from sdl interface")
+                       return
+               }
+       }
+
+       /* hack with WA only for mcxapp in near future */
+       if strings.Contains(msg.String(), "ricxapp") {
+               ep := sbiEngine.CheckEndpoint(string(params.Payload))
+               if ep == nil {
+                       xapp.Logger.Error("Update Routing Table Request, can't handle due to end point %s is not available in complete ep list: ", string(params.Payload))
+                       return
+               }
        }
 
-       ep := sbiEngine.CreateEndpoint(string(params.Payload))
-       if ep == nil {
-               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))
+       epstr, whid := sbiEngine.CreateEndpoint(msg.String())
+       if epstr == nil || whid < 0 {
+               xapp.Logger.Error("Wormhole Id creation failed %d for %s", whid, msg.String())
                return
        }
 
+       /*This is to ensure the latest routes are sent.
+       Assumption is that in this time interval the routes are built for this endpoint */
+       time.Sleep(100 * time.Millisecond)
        policies := rpeEngine.GeneratePolicies(rtmgr.Eps, data)
-       err = sbiEngine.DistributeToEp(policies, ep)
+       err = sbiEngine.DistributeToEp(policies, *epstr, whid)
        if err != nil {
-               xapp.Logger.Error("Routing table cannot be published due to: " + err.Error())
+               xapp.Logger.Error("Not able to publish the routing table due to: " + err.Error())
                return
        }
 }
 
+func getConfigData() (*rtmgr.RicComponents, error) {
+       var data *rtmgr.RicComponents
+       m.Lock()
+       data, err := sdlEngine.ReadAll(xapp.Config.GetString("rtfile"))
+
+       m.Unlock()
+       if data == nil {
+               if err != nil {
+                       return nil, errors.New("Cannot get data from sdl interface due to: " + err.Error())
+               } else {
+                       xapp.Logger.Debug("Cannot get data from sdl interface due to data is null")
+                       return nil, errors.New("Cannot get data from sdl interface")
+               }
+       }
+
+       return data, nil
+}
+
+func updateEp() (err error) {
+       data, err := getConfigData()
+       if err != nil {
+               return errors.New("Routing table cannot be published due to: " + err.Error())
+       }
+       EndpointLock.Lock()
+       sbiEngine.UpdateEndpoints(data)
+       EndpointLock.Unlock()
+
+       return nil
+}
+
+func sendPartialRoutesToAll(xappSubData *models.XappSubscriptionData, updatetype rtmgr.RMRUpdateType) (err error) {
+       policies := rpeEngine.GeneratePartialPolicies(rtmgr.Eps, xappSubData, updatetype)
+       err = sbiEngine.DistributeAll(policies)
+       if err != nil {
+               return errors.New("Routing table cannot be published due to: " + err.Error())
+       }
+
+       return nil
+}
+
 func sendRoutesToAll() (err error) {
 
-        m.Lock()
-        data, err := sdlEngine.ReadAll(xapp.Config.GetString("rtfile"))
-        m.Unlock()
-        if err != nil || data == nil {
-                return errors.New("Cannot get data from sdl interface due to: " + err.Error())
-        }
-        sbiEngine.UpdateEndpoints(data)
-        policies := rpeEngine.GeneratePolicies(rtmgr.Eps, data)
-        err = sbiEngine.DistributeAll(policies)
-        if err != nil {
-                return errors.New("Routing table cannot be published due to: " + err.Error())
-        }
+       data, err := getConfigData()
+       if err != nil {
+               return errors.New("Routing table cannot be published due to: " + err.Error())
+       }
+
+       policies := rpeEngine.GeneratePolicies(rtmgr.Eps, data)
+       err = sbiEngine.DistributeAll(policies)
+       if err != nil {
+               return errors.New("Routing table cannot be published due to: " + err.Error())
+       }
 
        return nil
 }
 
 func Serve() {
 
-        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)
-        if nbiErr != nil {
-                xapp.Logger.Error("Failed to initialize nbi due to: " + nbiErr.Error())
-                return
-        }
-
-        err := sbiEngine.Initialize(xapp.Config.GetString("sbiurl"))
-        if err != nil {
-                xapp.Logger.Info("Failed to open push socket due to: " + err.Error())
-                return
-        }
-        defer nbiEngine.Terminate()
-        defer sbiEngine.Terminate()
-
-        for {
-                sendRoutesToAll()
-
-                rtmgr.Rtmgr_ready = true
-                time.Sleep(INTERVAL * time.Second)
-                xapp.Logger.Debug("Periodic loop timed out. Setting triggerSBI flag to distribute updated routes.")
-        }
+       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)
+       if nbiErr != nil {
+               xapp.Logger.Error("Initialization of nbi failed due to: " + nbiErr.Error())
+               return
+       }
+
+       err := sbiEngine.Initialize(xapp.Config.GetString("sbiurl"))
+       if err != nil {
+               xapp.Logger.Info("Failed to open push socket due to: " + err.Error())
+               return
+       }
+       defer nbiEngine.Terminate()
+       defer sbiEngine.Terminate()
+
+       /* used for rtmgr restart case to connect to Endpoints */
+       go updateEp()
+       time.Sleep(5 * time.Second)
+       sendRoutesToAll()
+       /* Sometimes first message  fails, retry after 5 sec */
+       time.Sleep(5 * time.Second)
+       sendRoutesToAll()
+
+       for {
+               xapp.Logger.Debug("Periodic Routes value = %s", xapp.Config.GetString("periodicroutes"))
+               if xapp.Config.GetString("periodicroutes") == "enable" {
+                       go updateEp()
+                       time.Sleep(5 * time.Second)
+                       sendRoutesToAll()
+               }
+
+               rtmgr.Rtmgr_ready = true
+               time.Sleep(INTERVAL * time.Second)
+               xapp.Logger.Debug("Periodic loop timed out. Setting triggerSBI flag to distribute updated routes.")
+       }
 }