Fetch RIC active alarms via O1

[ric-plt/o1.git] / agent / pkg / sbi / sbi.go

diff --git a/agent/pkg/sbi/sbi.go b/agent/pkg/sbi/sbi.go
index 09dae0e..5120f5e 100755 (executable)
--- a/agent/pkg/sbi/sbi.go
+++ b/agent/pkg/sbi/sbi.go
@@ -20,24 +20,41 @@
 package sbi
 
 import (
+       "bytes"
+       "fmt"
        httptransport "github.com/go-openapi/runtime/client"
        "github.com/go-openapi/strfmt"
+       "os/exec"
+       "regexp"
+       "strings"
        "time"
 
-       "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
+       clientruntime "github.com/go-openapi/runtime/client"
+       "github.com/prometheus/alertmanager/api/v2/client"
+       "github.com/prometheus/alertmanager/api/v2/client/alert"
+
        apiclient "gerrit.oran-osc.org/r/ric-plt/o1mediator/pkg/appmgrclient"
        apixapp "gerrit.oran-osc.org/r/ric-plt/o1mediator/pkg/appmgrclient/xapp"
        apimodel "gerrit.oran-osc.org/r/ric-plt/o1mediator/pkg/appmgrmodel"
+
+       "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
+
 )
 
+type PodStatus struct {
+       Name   string
+       Health string
+       Status string
+}
+
 var log = xapp.Logger
 
-func NewSBIClient(host, baseUrl string, prot []string, timo int) *SBIClient {
-       return &SBIClient{host, baseUrl, prot, time.Duration(timo) * time.Second}
+func NewSBIClient(appmgrAddr, alertmgrAddr string, timo int) *SBIClient {
+       return &SBIClient{appmgrAddr, alertmgrAddr, time.Duration(timo) * time.Second}
 }
 
-func (s *SBIClient) CreateTransport() *apiclient.RICAppmgr {
-       return apiclient.New(httptransport.New(s.host, s.baseUrl, s.prot), strfmt.Default)
+func (s *SBIClient) CreateTransport(host string) *apiclient.RICAppmgr {
+       return apiclient.New(httptransport.New(host, "/ric/v1/", []string{"http"}), strfmt.Default)
 }
 
 func (s *SBIClient) BuildXappDescriptor(name, namespace, release, version string) *apimodel.XappDescriptor {
@@ -53,7 +70,7 @@ func (s *SBIClient) DeployXapp(xappDesc *apimodel.XappDescriptor) error {
        params := apixapp.NewDeployXappParamsWithTimeout(s.timeout).WithXappDescriptor(xappDesc)
        log.Info("SBI: DeployXapp=%v", params)
 
-       result, err := s.CreateTransport().Xapp.DeployXapp(params)
+       result, err := s.CreateTransport(s.appmgrAddr).Xapp.DeployXapp(params)
        if err != nil {
                log.Error("SBI: DeployXapp unsuccessful: %v", err)
        } else {
@@ -71,7 +88,7 @@ func (s *SBIClient) UndeployXapp(xappDesc *apimodel.XappDescriptor) error {
        params := apixapp.NewUndeployXappParamsWithTimeout(s.timeout).WithXAppName(name)
        log.Info("SBI: UndeployXapp=%v", params)
 
-       result, err := s.CreateTransport().Xapp.UndeployXapp(params)
+       result, err := s.CreateTransport(s.appmgrAddr).Xapp.UndeployXapp(params)
        if err != nil {
                log.Error("SBI: UndeployXapp unsuccessful: %v", err)
        } else {
@@ -82,7 +99,7 @@ func (s *SBIClient) UndeployXapp(xappDesc *apimodel.XappDescriptor) error {
 
 func (s *SBIClient) GetDeployedXapps() error {
        params := apixapp.NewGetAllXappsParamsWithTimeout(s.timeout)
-       result, err := s.CreateTransport().Xapp.GetAllXapps(params)
+       result, err := s.CreateTransport(s.appmgrAddr).Xapp.GetAllXapps(params)
        if err != nil {
                log.Error("GET unsuccessful: %v", err)
        } else {
@@ -105,7 +122,7 @@ func (s *SBIClient) BuildXappConfig(name, namespace string, configData interface
 
 func (s *SBIClient) ModifyXappConfig(xappConfig *apimodel.XAppConfig) error {
        params := apixapp.NewModifyXappConfigParamsWithTimeout(s.timeout).WithXAppConfig(xappConfig)
-       result, err := s.CreateTransport().Xapp.ModifyXappConfig(params)
+       result, err := s.CreateTransport(s.appmgrAddr).Xapp.ModifyXappConfig(params)
        if err != nil {
                log.Error("SBI: ModifyXappConfig unsuccessful: %v", err)
        } else {
@@ -113,3 +130,73 @@ func (s *SBIClient) ModifyXappConfig(xappConfig *apimodel.XAppConfig) error {
        }
        return err
 }
+
+func (s *SBIClient) GetAllPodStatus(namespace string) ([]PodStatus, error) {
+       output, err := s.RunCommand(fmt.Sprintf("/usr/local/bin/kubectl get pod -n %s", namespace))
+       if err != nil {
+               return []PodStatus{}, err
+       }
+
+       podStatusList := []PodStatus{}
+       var readyStr string
+       re := regexp.MustCompile(fmt.Sprintf(`%s-.*`, namespace))
+       podList := re.FindAllStringSubmatch(string(output), -1)
+       if podList != nil {
+               for _, pod := range podList {
+                       p := PodStatus{}
+                       fmt.Sscanf(pod[0], "%s %s %s", &p.Name, &readyStr, &p.Status)
+                       p.Name = strings.Split(p.Name, "-")[1]
+                       p.Health = s.GetHealthState(readyStr)
+
+                       podStatusList = append(podStatusList, p)
+               }
+       }
+       return podStatusList, nil
+}
+
+func (s *SBIClient) GetHealthState(ready string) (state string) {
+       result := strings.Split(ready, "/")
+       if len(result) < 2 {
+               return "unavailable"
+       }
+
+       if result[0] == result[1] {
+               state = "healthy"
+       } else {
+               state = "unhealthy"
+       }
+       return
+}
+
+func (s *SBIClient) RunCommand(args string) (string, error) {
+       return CommandExec(args)
+}
+
+var CommandExec = func(args string) (string, error) {
+       cmd := exec.Command("/bin/sh", "-c", args)
+       var stdout bytes.Buffer
+       var stderr bytes.Buffer
+       cmd.Stdout = &stdout
+       cmd.Stderr = &stderr
+
+       xapp.Logger.Debug("Running command: '%s'", cmd)
+       if err := cmd.Run(); err != nil {
+               xapp.Logger.Error("Command failed (%s): %v - %s", cmd, err.Error(), stderr.String())
+               return "", err
+       }
+       xapp.Logger.Debug("Command executed successfully!")
+       return stdout.String(), nil
+}
+
+func (s *SBIClient) GetAlerts() (*alert.GetAlertsOK, error) {
+       xapp.Logger.Info("Fetching alerts ...")
+
+       cr := clientruntime.New(s.alertmgrAddr, "/api/v2", []string{"http"})
+       resp, err := client.New(cr, strfmt.Default).Alert.GetAlerts(nil)
+       if err != nil {
+               xapp.Logger.Error("Fetching alerts failed with error: %v", err)
+               return nil, err
+       }
+
+       return resp, nil
+}
\ No newline at end of file