Support for time-to-live

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

diff --git a/alarm/alarm.go b/alarm/alarm.go
index cb18dd7..f8d909a 100755 (executable)
--- a/alarm/alarm.go
+++ b/alarm/alarm.go
@@ -21,10 +21,14 @@
 package alarm
 
 import (
+       "bytes"
        "encoding/json"
        "errors"
        "fmt"
+       "io/ioutil"
        "log"
+       "net/http"
+       "os"
        "time"
        "unsafe"
 )
@@ -42,10 +46,20 @@ import "C"
 // The identities are used when raising/clearing alarms, unless provided by the applications.
 func InitAlarm(mo, id string) (*RICAlarm, error) {
        r := &RICAlarm{
-               moId:  mo,
-               appId: id,
+               moId:       mo,
+               appId:      id,
+               managerUrl: ALARM_MANAGER_HTTP_URL,
+       }
+
+       if os.Getenv("ALARM_MANAGER_URL") != "" {
+               r.managerUrl = os.Getenv("ALARM_MANAGER_URL")
+       }
+
+       if os.Getenv("ALARM_IF_RMR") == "" {
+               go InitRMR(r, "")
+       } else {
+               go InitRMR(r, ALARM_MANAGER_RMR_URL)
        }
-       go InitRMR(r)
 
        return r, nil
 }
@@ -64,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}
 }
 
@@ -124,15 +138,23 @@ func (r *RICAlarm) AlarmString(a AlarmMessage) string {
 }
 
 func (r *RICAlarm) sendAlarmUpdateReq(a AlarmMessage) error {
-       if r.rmrCtx == nil || !r.rmrReady {
-               return errors.New("RMR no ready yet!")
-       }
 
-       log.Println("Sending alarm: ", r.AlarmString(a))
        payload, err := json.Marshal(a)
        if err != nil {
+               log.Println("json.Marshal failed with error: ", err)
                return err
        }
+       log.Println("Sending alarm: ", fmt.Sprintf("%s", payload))
+
+       if r.rmrCtx == nil || !r.rmrReady {
+               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)
+               }
+               log.Printf("Alarm posted to %s [status=%d]", url, resp.StatusCode)
+               return nil
+       }
 
        datap := C.CBytes(payload)
        defer C.free(datap)
@@ -143,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
 }
 
@@ -157,7 +180,27 @@ 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
+       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"
+
+       if err := ioutil.WriteFile(alarmRTFile, []byte(alarmRT), 0644); err != nil {
+               log.Println("ioutil.WriteFile failed with error: ", err)
+               return err
+       }
+
+       os.Setenv("RMR_SEED_RT", alarmRTFile)
+       os.Setenv("RMR_RTG_SVC", "-1")
+
        if ctx := C.rmrInit(); ctx != nil {
                r.rmrCtx = ctx
                r.rmrReady = true
@@ -166,3 +209,7 @@ func InitRMR(r *RICAlarm) error {
 
        return errors.New("rmrInit failed!")
 }
+
+func (r *RICAlarm) IsRMRReady() bool {
+       return r.rmrReady
+}