Backward compatible with old xApp descriptor

[ric-plt/xapp-frame.git] / pkg / xapp / xapp.go

diff --git a/pkg/xapp/xapp.go b/pkg/xapp/xapp.go
index 124fbcf..a9a130e 100755 (executable)
--- a/pkg/xapp/xapp.go
+++ b/pkg/xapp/xapp.go
@@ -23,9 +23,15 @@ import (
        "fmt"
        "github.com/spf13/viper"
        "net/http"
+       "os"
+       "os/signal"
+       "sync/atomic"
+       "syscall"
+       "time"
 )
 
 type ReadyCB func(interface{})
+type ShutdownCB func()
 
 var (
        // XApp is an application instance
@@ -37,8 +43,12 @@ var (
        Logger        *Log
        Config        Configurator
        Subscription  *Subscriber
+       Alarm         *AlarmClient
        readyCb       ReadyCB
        readyCbParams interface{}
+       shutdownCb    ShutdownCB
+       shutdownFlag  int32
+       shutdownCnt   int32
 )
 
 func IsReady() bool {
@@ -50,40 +60,91 @@ func SetReadyCB(cb ReadyCB, params interface{}) {
        readyCbParams = params
 }
 
-func xappReadyCb(params interface{}) {
+func XappReadyCb(params interface{}) {
+       Alarm = NewAlarmClient(viper.GetString("moId"), viper.GetString("name"))
        if readyCb != nil {
                readyCb(readyCbParams)
        }
 }
 
+func SetShutdownCB(cb ShutdownCB) {
+       shutdownCb = cb
+}
+
+func InstallSignalHandler() {
+       //
+       // Signal handlers to really exit program.
+       // shutdownCb can hang until application has
+       // made all needed gracefull shutdown actions
+       // hardcoded limit for shutdown is 20 seconds
+       //
+       interrupt := make(chan os.Signal, 1)
+       signal.Notify(interrupt, syscall.SIGINT, syscall.SIGTERM)
+       //signal handler function
+       go func() {
+               for range interrupt {
+                       if atomic.CompareAndSwapInt32(&shutdownFlag, 0, 1) {
+                               // close function
+                               go func() {
+                                       timeout := int(20)
+                                       sentry := make(chan struct{})
+                                       defer close(sentry)
+
+                                       // close callback
+                                       go func() {
+                                               if shutdownCb != nil {
+                                                       shutdownCb()
+                                               }
+                                               sentry <- struct{}{}
+                                       }()
+                                       select {
+                                       case <-time.After(time.Duration(timeout) * time.Second):
+                                               Logger.Info("xapp-frame shutdown callback took more than %d seconds", timeout)
+                                       case <-sentry:
+                                               Logger.Info("xapp-frame shutdown callback handled within %d seconds", timeout)
+                                       }
+                                       os.Exit(0)
+                               }()
+                       } else {
+                               newCnt := atomic.AddInt32(&shutdownCnt, 1)
+                               Logger.Info("xapp-frame shutdown already ongoing. Forced exit counter %d/%d ", newCnt, 5)
+                               if newCnt >= 5 {
+                                       Logger.Info("xapp-frame shutdown forced exit")
+                                       os.Exit(0)
+                               }
+                               continue
+                       }
+
+               }
+       }()
+}
+
 func init() {
        // Load xapp configuration
        Logger = LoadConfig()
 
-       Logger.SetLevel(viper.GetInt("logger.level"))
+       if viper.IsSet("controls.logger.level") {
+               Logger.SetLevel(viper.GetInt("controls.logger.level"))
+       } else {
+               Logger.SetLevel(viper.GetInt("logger.level"))
+       }
        Resource = NewRouter()
        Config = Configurator{}
        Metric = NewMetrics(viper.GetString("metrics.url"), viper.GetString("metrics.namespace"), Resource.router)
        Subscription = NewSubscriber(viper.GetString("subscription.host"), viper.GetInt("subscription.timeout"))
+       Sdl = NewSDLClient(viper.GetString("controls.db.namespace"))
+       Rnib = NewRNIBClient()
 
-       if viper.IsSet("db.namespaces") {
-               namespaces := viper.GetStringSlice("db.namespaces")
-               if namespaces[0] != "" {
-                       Sdl = NewSDLClient(viper.GetStringSlice("db.namespaces")[0])
-               }
-               if namespaces[1] != "" {
-                       Rnib = NewRNIBClient(viper.GetStringSlice("db.namespaces")[1])
-               }
-       } else {
-               Sdl = NewSDLClient(viper.GetString("db.namespace"))
-       }
+       InstallSignalHandler()
 }
 
 func RunWithParams(c MessageConsumer, sdlcheck bool) {
        Rmr = NewRMRClient()
-       Rmr.SetReadyCB(xappReadyCb, nil)
-       go http.ListenAndServe(viper.GetString("local.host"), Resource.router)
-       Logger.Info(fmt.Sprintf("Xapp started, listening on: %s", viper.GetString("local.host")))
+       Rmr.SetReadyCB(XappReadyCb, nil)
+
+       host := fmt.Sprintf(":%d", GetPortData("http").Port)
+       go http.ListenAndServe(host, Resource.router)
+       Logger.Info(fmt.Sprintf("Xapp started, listening on: %s", host))
        if sdlcheck {
                Sdl.TestConnection()
        }