Improve UT coverage
[ric-plt/alarm-go.git] / manager / cmd / restapi.go
index c455e87..a4668aa 100755 (executable)
@@ -23,12 +23,33 @@ package main
 import (
        "encoding/json"
        "net/http"
+       "strconv"
        "time"
 
        "gerrit.o-ran-sc.org/r/ric-plt/alarm-go/alarm"
        app "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
+       "github.com/gorilla/mux"
 )
 
+func (a *AlarmManager) InjectRoutes() {
+       app.Resource.InjectRoute("/ric/v1/alarms", a.RaiseAlarm, "POST")
+       app.Resource.InjectRoute("/ric/v1/alarms", a.ClearAlarm, "DELETE")
+       app.Resource.InjectRoute("/ric/v1/alarms/active", a.GetActiveAlarms, "GET")
+       app.Resource.InjectRoute("/ric/v1/alarms/history", a.GetAlarmHistory, "GET")
+       app.Resource.InjectRoute("/ric/v1/alarms/config", a.SetAlarmConfig, "POST")
+       app.Resource.InjectRoute("/ric/v1/alarms/config", a.GetAlarmConfig, "GET")
+       app.Resource.InjectRoute("/ric/v1/alarms/define", a.SetAlarmDefinition, "POST")
+       app.Resource.InjectRoute("/ric/v1/alarms/define/{alarmId}", a.DeleteAlarmDefinition, "DELETE")
+       app.Resource.InjectRoute("/ric/v1/alarms/define", a.GetAlarmDefinition, "GET")
+       app.Resource.InjectRoute("/ric/v1/alarms/define/{alarmId}", a.GetAlarmDefinition, "GET")
+
+       app.Resource.InjectRoute("/ric/v1/symptomdata", a.SymptomDataHandler, "GET")
+}
+
+func (a *AlarmManager) respondWithError(w http.ResponseWriter, code int, message string) {
+       a.respondWithJSON(w, code, map[string]string{"error": message})
+}
+
 func (a *AlarmManager) respondWithJSON(w http.ResponseWriter, code int, payload interface{}) {
        w.Header().Set("Content-Type", "application/json")
        w.WriteHeader(code)
@@ -60,6 +81,100 @@ func (a *AlarmManager) ClearAlarm(w http.ResponseWriter, r *http.Request) {
        }
 }
 
+func (a *AlarmManager) SetAlarmDefinition(w http.ResponseWriter, r *http.Request) {
+
+       app.Logger.Debug("POST arrived for creating alarm definition ")
+       /* If body is nil then return error */
+       if r.Body == nil {
+               app.Logger.Error("POST - body is empty")
+               a.respondWithError(w, http.StatusBadRequest, "No data in request body.")
+               return
+       }
+       defer r.Body.Close()
+
+       /* Parameters are available. Check if they are valid */
+       var alarmDefinitions RicAlarmDefinitions
+       err := json.NewDecoder(r.Body).Decode(&alarmDefinitions)
+       if err != nil {
+               app.Logger.Error("POST - received alarm definition  parameters are invalid - " + err.Error())
+               a.respondWithError(w, http.StatusBadRequest, "Invalid data in request body.")
+               return
+       }
+
+       for _, alarmDefinition := range alarmDefinitions.AlarmDefinitions {
+               _, exists := alarm.RICAlarmDefinitions[alarmDefinition.AlarmId]
+               if exists {
+                       app.Logger.Error("POST - alarm definition already exists for %v", alarmDefinition.AlarmId)
+               } else {
+                       ricAlarmDefintion := new(alarm.AlarmDefinition)
+                       ricAlarmDefintion.AlarmId = alarmDefinition.AlarmId
+                       ricAlarmDefintion.AlarmText = alarmDefinition.AlarmText
+                       ricAlarmDefintion.EventType = alarmDefinition.EventType
+                       ricAlarmDefintion.OperationInstructions = alarmDefinition.OperationInstructions
+                       ricAlarmDefintion.RaiseDelay = alarmDefinition.RaiseDelay
+                       ricAlarmDefintion.ClearDelay = alarmDefinition.ClearDelay
+                       ricAlarmDefintion.TimeToLive = alarmDefinition.TimeToLive
+                       alarm.RICAlarmDefinitions[alarmDefinition.AlarmId] = ricAlarmDefintion
+                       app.Logger.Debug("POST - alarm definition added for alarm id %v", alarmDefinition.AlarmId)
+               }
+       }
+
+       a.respondWithJSON(w, http.StatusOK, nil)
+       return
+}
+
+func (a *AlarmManager) DeleteAlarmDefinition(w http.ResponseWriter, r *http.Request) {
+       pathParams := mux.Vars(r)
+       alarmId, alarmIdok := pathParams["alarmId"]
+       if alarmIdok {
+               if ialarmId, err := strconv.Atoi(alarmId); err == nil {
+                       delete(alarm.RICAlarmDefinitions, ialarmId)
+                       app.Logger.Debug("DELETE - alarm definition deleted for alarmId %v", ialarmId)
+               } else {
+                       app.Logger.Error("DELETE - alarmId string to int conversion failed %v", alarmId)
+                       a.respondWithError(w, http.StatusBadRequest, "Invalid path parameter")
+                       return
+               }
+       } else {
+               app.Logger.Error("DELETE - alarmId does not exist %v", alarmId)
+               a.respondWithError(w, http.StatusBadRequest, "Invalid path parameter")
+               return
+
+       }
+}
+
+func (a *AlarmManager) GetAlarmDefinition(w http.ResponseWriter, r *http.Request) {
+       var ricAlarmDefinitions RicAlarmDefinitions
+       pathParams := mux.Vars(r)
+       alarmId, alarmIdok := pathParams["alarmId"]
+       if alarmIdok {
+               if ialarmId, err := strconv.Atoi(alarmId); err == nil {
+                       alarmDefinition, ok := alarm.RICAlarmDefinitions[ialarmId]
+                       if ok {
+                               app.Logger.Debug("Successfully returned alarm defintion for alarm id %v", ialarmId)
+                               a.respondWithJSON(w, http.StatusOK, alarmDefinition)
+                               return
+
+                       } else {
+                               app.Logger.Error("Requested alarm id not found %v", ialarmId)
+                               a.respondWithError(w, http.StatusBadRequest, "Non existent alarmId")
+                               return
+                       }
+               } else {
+                       app.Logger.Error("alarmId string to int conversion failed %v", alarmId)
+                       a.respondWithError(w, http.StatusBadRequest, "Invalid alarmId")
+                       return
+               }
+       } else {
+               app.Logger.Debug("GET arrived for all alarm definitions ")
+               for _, alarmDefinition := range alarm.RICAlarmDefinitions {
+                       ricAlarmDefinitions.AlarmDefinitions = append(ricAlarmDefinitions.AlarmDefinitions, alarmDefinition)
+               }
+               app.Logger.Debug("Successfully returned all alarm definitions")
+               a.respondWithJSON(w, http.StatusOK, ricAlarmDefinitions)
+       }
+}
+
 func (a *AlarmManager) doAction(w http.ResponseWriter, r *http.Request, isRaiseAlarm bool) error {
        app.Logger.Info("doAction: request received = %t", isRaiseAlarm)
 
@@ -84,10 +199,12 @@ func (a *AlarmManager) doAction(w http.ResponseWriter, r *http.Request, isRaiseA
                m.AlarmTime = time.Now().UnixNano()
        }
 
-       _, err := a.ProcessAlarm(&m)
+       _, err := a.ProcessAlarm(&AlarmNotification{m, alarm.AlarmDefinition{}})
        return err
 }
 
+// HandleViaRmr is defined but not used as of now 
+/*
 func (a *AlarmManager) HandleViaRmr(d alarm.Alarm, isRaiseAlarm bool) error {
        alarmClient, err := alarm.InitAlarm(d.ManagedObjectId, d.ApplicationId)
        if err != nil {
@@ -103,7 +220,7 @@ func (a *AlarmManager) HandleViaRmr(d alarm.Alarm, isRaiseAlarm bool) error {
        }
 
        return nil
-}
+}*/
 
 func (a *AlarmManager) SetAlarmConfig(w http.ResponseWriter, r *http.Request) {
        var m alarm.AlarmConfigParams
@@ -127,3 +244,49 @@ func (a *AlarmManager) GetAlarmConfig(w http.ResponseWriter, r *http.Request) {
        a.respondWithJSON(w, http.StatusOK, m)
        return
 }
+
+func (a *AlarmManager) SymptomDataHandler(w http.ResponseWriter, r *http.Request) {
+       baseDir := "/tmp/symptomdata/"
+       if err := app.Util.CreateDir(baseDir); err != nil {
+               app.Resource.SendSymptomDataError(w, r, "CreateDir failed: "+err.Error())
+               return
+       }
+
+       if b, err := json.Marshal(a.activeAlarms); err == nil {
+               if err := app.Util.WriteToFile(baseDir+"active_alarms.json", string(b)); err != nil {
+                       app.Resource.SendSymptomDataError(w, r, "writeToFile failed: "+err.Error())
+                       return
+               }
+       }
+
+       if b, err := json.Marshal(a.alarmHistory); err == nil {
+               if err := app.Util.WriteToFile(baseDir+"alarm_history.json", string(b)); err != nil {
+                       app.Resource.SendSymptomDataError(w, r, "writeToFile failed: "+err.Error())
+                       return
+               }
+       }
+
+       var ac alarm.AlarmConfigParams
+       ac.MaxActiveAlarms = a.maxActiveAlarms
+       ac.MaxAlarmHistory = a.maxAlarmHistory
+
+       if b, err := json.Marshal(ac); err == nil {
+               if err := app.Util.WriteToFile(baseDir+"alarm_config.json", string(b)); err != nil {
+                       app.Resource.SendSymptomDataError(w, r, "writeToFile failed: "+err.Error())
+                       return
+               }
+       }
+
+       var ad RicAlarmDefinitions
+       for _, alarmDefinition := range alarm.RICAlarmDefinitions {
+               ad.AlarmDefinitions = append(ad.AlarmDefinitions, alarmDefinition)
+       }
+       if b, err := json.Marshal(ad); err == nil {
+               if err := app.Util.WriteToFile(baseDir+"alarm_defs.json", string(b)); err != nil {
+                       app.Resource.SendSymptomDataError(w, r, "writeToFile failed: "+err.Error())
+                       return
+               }
+       }
+
+       app.Resource.SendSymptomDataFile(w, r, baseDir, "symptomdata.zip")
+}