Merge "Special version for alarm manager" into alarm_wa

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

diff --git a/pkg/xapp/restapi.go b/pkg/xapp/restapi.go
index 40ad8d2..205f5f6 100755 (executable)
--- a/pkg/xapp/restapi.go
+++ b/pkg/xapp/restapi.go
@@ -22,12 +22,13 @@ package xapp
 import (
        "encoding/json"
        "fmt"
-       "github.com/gorilla/mux"
-       "github.com/spf13/viper"
        "io/ioutil"
        "net/http"
        "os"
 
+       "github.com/gorilla/mux"
+       "github.com/spf13/viper"
+
        "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/models"
 )
 
@@ -103,9 +104,13 @@ func (r *Router) CheckStatus() (status bool) {
 }
 
 func (r *Router) GetSymptomDataParams(w http.ResponseWriter, req *http.Request) SymptomDataParams {
+       Logger.Info("GetSymptomDataParams ...")
+
        params := SymptomDataParams{}
        queryParams := req.URL.Query()
 
+       Logger.Info("GetSymptomDataParams: %+v", queryParams)
+
        for p := range queryParams {
                if p == "timeout" {
                        fmt.Sscanf(p, "%d", &params.Timeout)
@@ -120,12 +125,50 @@ func (r *Router) GetSymptomDataParams(w http.ResponseWriter, req *http.Request)
        return params
 }
 
+func (r *Router) CollectDefaultSymptomData(fileName string, data interface{}) string {
+       baseDir := Config.GetString("controls.symptomdata.baseDir")
+       if baseDir == "" {
+               baseDir = "/tmp/xapp/"
+       }
+
+       if err := Util.CreateDir(baseDir); err != nil {
+               Logger.Error("CreateDir failed: %v", err)
+               return ""
+       }
+
+       if metrics, err := r.GetLocalMetrics(GetPortData("http").Port); err == nil {
+               if err := Util.WriteToFile(baseDir+"metrics.json", metrics); err != nil {
+                       Logger.Error("writeToFile failed for metrics.json: %v", err)
+               }
+       }
+
+       if data != nil {
+               if b, err := json.MarshalIndent(data, "", "  "); err == nil {
+                       Util.WriteToFile(baseDir+fileName, string(b))
+               }
+       }
+
+       rtPath := os.Getenv("RMR_STASH_RT")
+       if rtPath == "" {
+               return baseDir
+       }
+
+       input, err := ioutil.ReadFile(rtPath)
+       if err != nil {
+               Logger.Error("ioutil.ReadFile failed: %v", err)
+               return baseDir
+       }
+
+       Util.WriteToFile(baseDir+"rttable.txt", string(input))
+       return baseDir
+}
+
 func (r *Router) SendSymptomDataJson(w http.ResponseWriter, req *http.Request, data interface{}, n string) {
        w.Header().Set("Content-Type", "application/json")
        w.Header().Set("Content-Disposition", "attachment; filename="+n)
        w.WriteHeader(http.StatusOK)
        if data != nil {
-               response, _ := json.Marshal(data)
+               response, _ := json.MarshalIndent(data, "", " ")
                w.Write(response)
        }
 }
@@ -156,6 +199,21 @@ func (r *Router) SendSymptomDataError(w http.ResponseWriter, req *http.Request,
        http.Error(w, message, http.StatusInternalServerError)
 }
 
+func (r *Router) GetLocalMetrics(port int) (string, error) {
+       resp, err := http.Get(fmt.Sprintf("http://localhost:%d/ric/v1/metrics", port))
+       if err != nil {
+               return "", err
+       }
+       defer resp.Body.Close()
+
+       metrics, err := ioutil.ReadAll(resp.Body)
+       if err != nil {
+               return "", err
+       }
+
+       return string(metrics), nil
+}
+
 func IsHealthProbeReady() bool {
        return healthReady
 }