import (
"flag"
- "nbi"
"os"
- "rpe"
- "rtmgr"
- "sbi"
- "sdl"
+ "os/signal"
+ "routing-manager/pkg/nbi"
+ "routing-manager/pkg/rpe"
+ "routing-manager/pkg/rtmgr"
+ "routing-manager/pkg/sbi"
+ "routing-manager/pkg/sdl"
+ "syscall"
"time"
)
)
func parseArgs() {
- // TODO: arguments should be validated (filename; xm-url; sbi-if)
+ // TODO: arguments should be validated (filename; xm-url; sbi-if; rest-url; rest-port)
args = make(map[string]*string)
- args["nbi"] = flag.String("nbi", "httpGetter", "Northbound interface module to be used. Valid values are: 'httpGetter'")
+ args["configfile"] = flag.String("configfile", "/etc/rtmgrcfg.json", "Routing manager's configuration file path")
+ args["nbi"] = flag.String("nbi", "httpGetter", "Northbound interface module to be used. Valid values are: 'httpGetter | httpRESTful'")
args["sbi"] = flag.String("sbi", "nngpush", "Southbound interface module to be used. Valid values are: 'nngpush | nngpub'")
args["rpe"] = flag.String("rpe", "rmrpush", "Route Policy Engine to be used. Valid values are: 'rmrpush | rmrpub'")
args["sdl"] = flag.String("sdl", "file", "Datastore enginge to be used. Valid values are: 'file'")
args["xm-url"] = flag.String("xm-url", "http://localhost:3000/xapps", "HTTP URL where xApp Manager exposes the entire xApp List")
+ args["nbi-if"] = flag.String("nbi-if", "http://localhost:8888", "Base HTTP URL where routing manager will be listening on")
args["sbi-if"] = flag.String("sbi-if", "0.0.0.0", "IPv4 address of interface where Southbound socket to be opened")
args["filename"] = flag.String("filename", "/db/rt.json", "Absolute path of file where the route information to be stored")
args["loglevel"] = flag.String("loglevel", "INFO", "INFO | WARN | ERROR | DEBUG")
flag.Parse()
}
-func initRtmgr() (*nbi.NbiEngineConfig, *sbi.SbiEngineConfig, *sdl.SdlEngineConfig, *rpe.RpeEngineConfig, error) {
+func initRtmgr() (nbi.NbiEngine, sbi.SbiEngine, sdl.SdlEngine, rpe.RpeEngine, error) {
var err error
- var nbii *nbi.NbiEngineConfig
- var sbii *sbi.SbiEngineConfig
- var sdli *sdl.SdlEngineConfig
- var rpei *rpe.RpeEngineConfig
+ var nbii nbi.NbiEngine
+ var sbii sbi.SbiEngine
+ var sdli sdl.SdlEngine
+ var rpei rpe.RpeEngine
if nbii, err = nbi.GetNbi(*args["nbi"]); err == nil && nbii != nil {
if sbii, err = sbi.GetSbi(*args["sbi"]); err == nil && sbii != nil {
if sdli, err = sdl.GetSdl(*args["sdl"]); err == nil && sdli != nil {
return nil, nil, nil, nil, err
}
-func serve(nbii *nbi.NbiEngineConfig, sbii *sbi.SbiEngineConfig, sdli *sdl.SdlEngineConfig, rpei *rpe.RpeEngineConfig) {
- err := sbii.OpenSocket(*args["sbi-if"])
+func serveSBI(triggerSBI <-chan bool, sbiEngine sbi.SbiEngine, sdlEngine sdl.SdlEngine, rpeEngine rpe.RpeEngine) {
+ for {
+ if <-triggerSBI {
+ data, err := sdlEngine.ReadAll(*args["filename"])
+ if err != nil || data == nil {
+ rtmgr.Logger.Error("cannot get data from sdl interface due to: " + err.Error())
+ continue
+ }
+ sbiEngine.UpdateEndpoints(data)
+ policies := rpeEngine.GeneratePolicies(rtmgr.Eps)
+ err = sbiEngine.DistributeAll(policies)
+ if err != nil {
+ rtmgr.Logger.Error("routing rable cannot be published due to: " + err.Error())
+ }
+ }
+ }
+}
+
+func serve(nbiEngine nbi.NbiEngine, sbiEngine sbi.SbiEngine, sdlEngine sdl.SdlEngine, rpeEngine rpe.RpeEngine) {
+
+ triggerSBI := make(chan bool)
+
+ nbiErr := nbiEngine.Initialize(*args["xm-url"], *args["nbi-if"], *args["filename"], *args["configfile"],
+ sdlEngine, rpeEngine, triggerSBI)
+ if nbiErr != nil {
+ rtmgr.Logger.Error("fail to initialize nbi due to: " + nbiErr.Error())
+ return
+ }
+
+ err := sbiEngine.Initialize(*args["sbi-if"])
if err != nil {
rtmgr.Logger.Info("fail to open pub socket due to: " + err.Error())
return
}
- defer sbii.CloseSocket()
+ defer nbiEngine.Terminate()
+ defer sbiEngine.Terminate()
+
+ // This SBI Go routine is trtiggered by periodic main loop and when data is recieved on REST interface.
+ go serveSBI(triggerSBI, sbiEngine, sdlEngine, rpeEngine)
+
for {
time.Sleep(INTERVAL * time.Second)
- data, err := nbii.BatchFetch(*args["xm-url"])
- if err != nil {
- rtmgr.Logger.Error("cannot get data from " + nbii.Engine.Name + " interface dute to: " + err.Error())
- } else {
- sdli.WriteAll(*args["filename"], data)
- }
- data, err = sdli.ReadAll(*args["filename"])
- if err != nil || data == nil {
- rtmgr.Logger.Error("cannot get data from " + sdli.Engine.Name + " interface dute to: " + err.Error())
- continue
- }
- sbi.UpdateEndpointList(data, sbii)
- policies := rpei.GeneratePolicies(rtmgr.Eps)
- err = sbii.DistributeAll(policies)
- if err != nil {
- rtmgr.Logger.Error("routing rable cannot be published due to: " + err.Error())
+ if *args["nbi"] == "httpGetter" {
+ data, err := nbiEngine.(*nbi.HttpGetter).FetchAllXapps(*args["xm-url"])
+ if err != nil {
+ rtmgr.Logger.Error("cannot fetch xapp data dute to: " + err.Error())
+ } else if data != nil {
+ sdlEngine.WriteXapps(*args["filename"], data)
+ }
}
+
+ triggerSBI <- true
}
}
+func SetupCloseHandler() {
+ c := make(chan os.Signal, 2)
+ signal.Notify(c, os.Interrupt, syscall.SIGTERM)
+ go func() {
+ <-c
+ rtmgr.Logger.Info("\r- Ctrl+C pressed in Terminal")
+ os.Exit(0)
+ }()
+}
+
func main() {
parseArgs()
rtmgr.SetLogLevel(*args["loglevel"])
- nbii, sbii, sdli, rpei, err := initRtmgr()
+ nbiEngine, sbiEngine, sdlEngine, rpeEngine, err := initRtmgr()
if err != nil {
rtmgr.Logger.Error(err.Error())
os.Exit(1)
}
+ SetupCloseHandler()
rtmgr.Logger.Info("Start " + SERVICENAME + " service")
rtmgr.Eps = make(rtmgr.Endpoints)
- serve(nbii, sbii, sdli, rpei)
+ serve(nbiEngine, sbiEngine, sdlEngine, rpeEngine)
os.Exit(0)
}
Code Review / ric-plt / rtmgr.git / blobdiff