do not expose low level message.Message to publish

This commit is contained in:
Marcelo Pires
2018-09-06 13:03:21 +02:00
parent c8bfbbd8fd
commit eed039f351
4 changed files with 13 additions and 11 deletions

View File

@ -18,9 +18,9 @@ type Transport interface {
Options() *Options
Handshake() error
Connect() error
Subscribe(subscription string, onMessage func(message *message.Message)) error
Subscribe(subscription string, onMessage func(message message.Data)) error
Unsubscribe(subscription string) error
Publish(subscription string, message *message.Message) error
Publish(subscription string, message message.Data) error
}
type Event = string

View File

@ -160,7 +160,7 @@ func (w *Websocket) Connect() error {
return w.sendMessage(&m)
}
func (w *Websocket) Subscribe(subscription string, onMessage func(message *message.Message)) error {
func (w *Websocket) Subscribe(subscription string, onMessage func(data message.Data)) error {
m := &message.Message{
Channel: transport.Subscribe,
ClientId: w.clientID,
@ -184,7 +184,7 @@ func (w *Websocket) Subscribe(subscription string, onMessage func(message *messa
if inMsg.GetError() != nil {
return inMsg.GetError()
}
onMessage(inMsg)
onMessage(inMsg.Data)
}
return nil
}
@ -200,7 +200,7 @@ func (w *Websocket) Unsubscribe(subscription string) error {
return w.sendMessage(m)
}
func (w *Websocket) Publish(subscription string, message *message.Message) error {
func (w *Websocket) Publish(subscription string, data message.Data) error {
panic("not implemented")
}