Initial commit

[ric-plt/o1.git] / agent / cmd / o1agent.go

diff --git a/agent/cmd/o1agent.go b/agent/cmd/o1agent.go
new file mode 100755 (executable)
index 0000000..76d19b6
--- /dev/null
+++ b/agent/cmd/o1agent.go
@@ -0,0 +1,99 @@
+/*
+==================================================================================
+  Copyright (c) 2019 AT&T Intellectual Property.
+  Copyright (c) 2019 Nokia
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+==================================================================================
+*/
+
+package main
+
+import (
+       "os"
+    "os/signal"
+    "syscall"
+
+       "github.com/spf13/viper"
+       "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
+       "gerrit.oran-osc.org/r/ric-plt/o1mediator/pkg/sbi"
+       "gerrit.oran-osc.org/r/ric-plt/o1mediator/pkg/nbi"
+)
+
+type O1Agent struct {
+       rmrReady        bool
+       nbiClient       *nbi.Nbi
+       sigChan         chan os.Signal
+}
+
+func (o O1Agent) Consume(rp *xapp.RMRParams) (err error) {
+       xapp.Logger.Debug("Message received!")
+       return nil
+}
+
+func (o *O1Agent) ConfigChangeHandler(f string) {
+       xapp.Logger.Debug("Config changed!")
+}
+
+func (o *O1Agent) StatusCB() bool {
+       if !o.rmrReady {
+               xapp.Logger.Info("RMR not ready yet!")
+       }
+       return true
+}
+
+func (o *O1Agent) Run() {
+       xapp.Logger.SetMdc("o1agent", "0.3.1")
+       xapp.SetReadyCB(func(d interface{}) { o.rmrReady = true }, true)
+       xapp.AddConfigChangeListener(o.ConfigChangeHandler)
+       xapp.Resource.InjectStatusCb(o.StatusCB)
+
+       signal.Notify(o.sigChan, syscall.SIGINT, syscall.SIGTERM)
+       go o.Sighandler()
+
+       xapp.Run(o)
+}
+
+func (o *O1Agent) Sighandler() {
+       xapp.Logger.Info("Signal handler installed!")
+
+       <- o.sigChan
+       o.nbiClient.Stop()
+       os.Exit(1)
+}
+
+func NewO1Agent() *O1Agent {
+       host := viper.GetString("sbi.appmgrService")
+       baseUrl := viper.GetString("sbi.baseUrl")
+       prot := viper.GetString("sbi.proto")
+       timeout := viper.GetInt("sbi.timeout")
+
+       sbiClient := sbi.NewSBIClient(host, baseUrl, []string{prot}, timeout)
+
+       return &O1Agent{
+               rmrReady: false,
+               nbiClient: nbi.NewNbi(sbiClient),
+               sigChan: make(chan os.Signal, 1),
+       }
+}
+
+func main() {
+       o1Agent := NewO1Agent()
+
+       if ok := o1Agent.nbiClient.Start(); !ok {
+               xapp.Logger.Error("NBI initialization failed!")
+               return
+       }
+       
+       o1Agent.Run()
+}