subscription_event.go

package actioncable

import (
	"encoding/json"
)

type SubscriptionEvent struct {
	Type       SubscriptionEventType
	Message    map[string]interface{}
	RawMessage *json.RawMessage
}

type SubscriptionEventType string

const (
	Connected    SubscriptionEventType = SubscriptionEventType("connected")
	Disconnected SubscriptionEventType = SubscriptionEventType("disconnected")
	Rejected     SubscriptionEventType = SubscriptionEventType("rejected")
	Received     SubscriptionEventType = SubscriptionEventType("received")
)

func createSubscriptionEvent(eventType SubscriptionEventType, event *Event) *SubscriptionEvent {
	if event == nil {
		event = NewNilEvent()
	}
	e := newSubscriptionEvent(eventType, &event.Message)

	return e
}

func newSubscriptionEvent(eventType SubscriptionEventType, raw *json.RawMessage) *SubscriptionEvent {
	return &SubscriptionEvent{
		Type:       eventType,
		Message:    map[string]interface{}{},
		RawMessage: raw,
	}
}

func (s *SubscriptionEvent) ReadJSON(v interface{}) error {
	b, err := json.Marshal(s.RawMessage)
	if err != nil {
		return err
	}

	return json.Unmarshal(b, v)
}