LN0739_FM_RF13: support interface to external Prometheus alert manager

[ric-plt/alarm-go.git] / alarm / alarm.go

diff --git a/alarm/alarm.go b/alarm/alarm.go
index 4059b0d..3856fce 100755 (executable)
--- a/alarm/alarm.go
+++ b/alarm/alarm.go
@@ -48,16 +48,18 @@ func InitAlarm(mo, id string) (*RICAlarm, error) {
        r := &RICAlarm{
                moId:       mo,
                appId:      id,
-               adapterUrl: ALARM_ADAPTER_HTTP_URL,
+               managerUrl: ALARM_MANAGER_HTTP_URL,
        }
 
-       if os.Getenv("ALARM_ADAPTER_URL") != "" {
-               r.adapterUrl = os.Getenv("ALARM_ADAPTER_URL")
+       if os.Getenv("ALARM_MANAGER_URL") != "" {
+               r.managerUrl = os.Getenv("ALARM_MANAGER_URL")
        }
 
-       if os.Getenv("ALARM_IF_RMR") != "" {
-               go InitRMR(r)
-       }
+       if os.Getenv("ALARM_IF_RMR") == "" {
+                go InitRMR(r, "")
+        } else {
+                go InitRMR(r, ALARM_MANAGER_RMR_URL)
+        }
 
        return r, nil
 }
@@ -76,7 +78,7 @@ func (r *RICAlarm) NewAlarm(sp int, severity Severity, ainfo, iinfo string) Alar
 
 // Create a new AlarmMessage instance
 func (r *RICAlarm) NewAlarmMessage(a Alarm, alarmAction AlarmAction) AlarmMessage {
-       alarmTime := time.Now().UnixNano() / 1000
+       alarmTime := time.Now().UnixNano()
        return AlarmMessage{a, alarmAction, alarmTime}
 }
 
@@ -136,6 +138,7 @@ func (r *RICAlarm) AlarmString(a AlarmMessage) string {
 }
 
 func (r *RICAlarm) sendAlarmUpdateReq(a AlarmMessage) error {
+
        payload, err := json.Marshal(a)
        if err != nil {
                log.Println("json.Marshal failed with error: ", err)
@@ -144,7 +147,7 @@ func (r *RICAlarm) sendAlarmUpdateReq(a AlarmMessage) error {
        log.Println("Sending alarm: ", fmt.Sprintf("%s", payload))
 
        if r.rmrCtx == nil || !r.rmrReady {
-               url := fmt.Sprintf("%s/%s", r.adapterUrl, "ric/v1/alarms")
+               url := fmt.Sprintf("%s/%s", r.managerUrl, "ric/v1/alarms")
                resp, err := http.Post(url, "application/json", bytes.NewReader(payload))
                if err != nil || resp == nil {
                        return fmt.Errorf("Unable to send alarm: %v", err)
@@ -162,6 +165,7 @@ func (r *RICAlarm) sendAlarmUpdateReq(a AlarmMessage) error {
                log.Println("rmrSend failed with error: ", state)
                return errors.New(fmt.Sprintf("rmrSend failed with error: %d", state))
        }
+
        return nil
 }
 
@@ -176,14 +180,15 @@ func (r *RICAlarm) ReceiveMessage(cb func(AlarmMessage)) error {
        return errors.New("rmrRcv failed!")
 }
 
-func InitRMR(r *RICAlarm) error {
+func InitRMR(r *RICAlarm,  endpoint string) error {
        // Setup static RT for alarm system
-       endpoint := ALARM_ADAPTER_RMR_URL
-       if r.moId == "my-pod" {
-               endpoint = "localhost:4560"
-       } else if r.moId == "my-pod-lib" {
-               endpoint = "localhost:4588"
-       }
+       if endpoint == "" {
+               if r.moId == "my-pod" {
+                        endpoint = "127.0.0.1:4560"
+                } else if r.moId == "my-pod-lib" {
+                        endpoint = "127.0.0.1:4588"
+                }
+        }
 
        alarmRT := fmt.Sprintf("newrt|start\nrte|13111|%s\nnewrt|end\n", endpoint)
        alarmRTFile := "/tmp/alarm.rt"
@@ -204,3 +209,7 @@ func InitRMR(r *RICAlarm) error {
 
        return errors.New("rmrInit failed!")
 }
+
+func (r *RICAlarm) IsRMRReady() bool {
+       return r.rmrReady
+}