getAllSubscriptions API (RM -> SM) during restart of routing manager handled
[ric-plt/rtmgr.git] / 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"), xapp.Config.GetString("e2murl"), 
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         c := sbi.NewControl()
149         go c.Run()
150
151         serve(nbiEngine, sbiEngine, sdlEngine, rpeEngine, &m)
152         os.Exit(0)
153 }