signalr.go

package signalr

// SignalR -
type SignalR struct {
	hub *Hub
	t   *Transport

	url string
}

// NewSignalR -
func NewSignalR(url string) *SignalR {
	return &SignalR{
		t:   NewTransport(url),
		url: url,
	}
}

// Connect - connect to server
func (s *SignalR) Connect(version Version) error {
	resp, err := s.t.Negotiate(version)
	if err != nil {
		return err
	}
	var id string
	switch version {
	case Version0:
		id = resp.ConnectionID
	case Version1:
		id = resp.ConnectionToken
	}

	hub, err := NewHub(s.url, id)
	if err != nil {
		return err
	}
	s.hub = hub

	return s.hub.Connect()
}

// Messages - listens message channel
func (s *SignalR) Messages() <-chan interface{} {
	return s.hub.msgs
}

// Close - close connection
func (s *SignalR) Close() error {
	return s.hub.Close()
}

// Send - send message to server
func (s *SignalR) Send(msg interface{}) error {
	return s.hub.Send(msg)
}

// SetOnReconnect -
func (s *SignalR) SetOnReconnect(onReconnect func() error) {
	s.hub.onReconnect = onReconnect
}