Added events decoder helpers.

[src/xds/xds-agent.git] / lib / apiv1 / events.go

diff --git a/lib/apiv1/events.go b/lib/apiv1/events.go
index dde2710..cdd0889 100644 (file)
--- a/lib/apiv1/events.go
+++ b/lib/apiv1/events.go
@@ -1,5 +1,10 @@
 package apiv1
 
+import (
+       "encoding/json"
+       "fmt"
+)
+
 // EventRegisterArgs is the parameters (json format) of /events/register command
 type EventRegisterArgs struct {
        Name      string `json:"name"`
@@ -12,17 +17,17 @@ type EventUnRegisterArgs struct {
        ID   int    `json:"id"`
 }
 
-// Events definitions
+// Events Type definitions
 const (
        // EventTypePrefix Used as event prefix
        EventTypePrefix = "event:" // following by event type
 
        // Supported Events type
        EVTAll           = EventTypePrefix + "all"
-       EVTServerConfig  = EventTypePrefix + "server-config"        // data type apiv1.ServerCfg
-       EVTProjectAdd    = EventTypePrefix + "project-add"          // data type apiv1.ProjectConfig
-       EVTProjectDelete = EventTypePrefix + "project-delete"       // data type apiv1.ProjectConfig
-       EVTProjectChange = EventTypePrefix + "project-state-change" // data type apiv1.ProjectConfig
+       EVTServerConfig  = EventTypePrefix + "server-config"        // type EventMsg with Data type apiv1.ServerCfg
+       EVTProjectAdd    = EventTypePrefix + "project-add"          // type EventMsg with Data type apiv1.ProjectConfig
+       EVTProjectDelete = EventTypePrefix + "project-delete"       // type EventMsg with Data type apiv1.ProjectConfig
+       EVTProjectChange = EventTypePrefix + "project-state-change" // type EventMsg with Data type apiv1.ProjectConfig
 )
 
 // EVTAllList List of all supported events
@@ -33,9 +38,39 @@ var EVTAllList = []string{
        EVTProjectChange,
 }
 
-// EventMsg Message send
+// EventMsg Event message send over Websocket, data format depend to Type (see DecodeXXX function)
 type EventMsg struct {
        Time string      `json:"time"`
        Type string      `json:"type"`
        Data interface{} `json:"data"`
 }
+
+// DecodeServerCfg Helper to decode Data field type ServerCfg
+func (e *EventMsg) DecodeServerCfg() (ServerCfg, error) {
+       p := ServerCfg{}
+       if e.Type != EVTServerConfig {
+               return p, fmt.Errorf("Invalid type")
+       }
+       d, err := json.Marshal(e.Data)
+       if err == nil {
+               err = json.Unmarshal(d, &p)
+       }
+       return p, err
+}
+
+// DecodeProjectConfig Helper to decode Data field type ProjectConfig
+func (e *EventMsg) DecodeProjectConfig() (ProjectConfig, error) {
+       var err error
+       p := ProjectConfig{}
+       switch e.Type {
+       case EVTProjectAdd, EVTProjectChange, EVTProjectDelete:
+               d := []byte{}
+               d, err = json.Marshal(e.Data)
+               if err == nil {
+                       err = json.Unmarshal(d, &p)
+               }
+       default:
+               err = fmt.Errorf("Invalid type")
+       }
+       return p, err
+}