Add UT for REST interface & code refactor
[ric-plt/alarm-go.git] / adapter / cmd / restapi.go
1 /*
2  *  Copyright (c) 2020 AT&T Intellectual Property.
3  *  Copyright (c) 2020 Nokia.
4  *
5  *  Licensed under the Apache License, Version 2.0 (the "License");
6  *  you may not use this file except in compliance with the License.
7  *  You may obtain a copy of the License at
8  *
9  *     http://www.apache.org/licenses/LICENSE-2.0
10  *
11  *  Unless required by applicable law or agreed to in writing, software
12  *  distributed under the License is distributed on an "AS IS" BASIS,
13  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  *  See the License for the specific language governing permissions and
15  *  limitations under the License.
16  *
17  * This source code is part of the near-RT RIC (RAN Intelligent Controller)
18  * platform project (RICP).
19  */
20
21 package main
22
23 import (
24         "encoding/json"
25         "net/http"
26
27         "gerrit.o-ran-sc.org/r/ric-plt/alarm-go/alarm"
28         app "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
29 )
30
31 var alarmClient *alarm.RICAlarm
32
33 func (a *AlarmAdapter) GetActiveAlarms(w http.ResponseWriter, r *http.Request) {
34         app.Logger.Info("GetActiveAlarms: request received!")
35
36         w.Header().Set("Content-Type", "application/json")
37         w.WriteHeader(http.StatusOK)
38         response, _ := json.Marshal(a.activeAlarms)
39         w.Write(response)
40 }
41
42 func (a *AlarmAdapter) RaiseAlarm(w http.ResponseWriter, r *http.Request) {
43         a.doAction(w, r, true)
44 }
45
46 func (a *AlarmAdapter) ClearAlarm(w http.ResponseWriter, r *http.Request) {
47         a.doAction(w, r, false)
48 }
49
50 func (a *AlarmAdapter) doAction(w http.ResponseWriter, r *http.Request, raiseAlarm bool) {
51         app.Logger.Info("doAction: request received!")
52
53         if r.Body == nil {
54                 return
55         }
56         defer r.Body.Close()
57
58         var d alarm.Alarm
59         err := json.NewDecoder(r.Body).Decode(&d)
60         if err != nil {
61                 app.Logger.Error("json.NewDecoder failed: %v", err)
62                 return
63         }
64
65         if alarmClient == nil {
66                 alarmClient, err = alarm.InitAlarm("RIC", "UEEC")
67                 if err != nil {
68                         app.Logger.Error("json.NewDecoder failed: %v", err)
69                         return
70                 }
71         }
72
73         alarmData := alarmClient.NewAlarm(d.SpecificProblem, d.PerceivedSeverity, d.AdditionalInfo, d.IdentifyingInfo)
74         if raiseAlarm {
75                 alarmClient.Raise(alarmData)
76         } else {
77                 alarmClient.Clear(alarmData)
78         }
79 }