diff options
Diffstat (limited to 'plugins/websockets/memory/inMemory.go')
| -rw-r--r-- | plugins/websockets/memory/inMemory.go | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/plugins/websockets/memory/inMemory.go b/plugins/websockets/memory/inMemory.go index deb927ed..cef28182 100644 --- a/plugins/websockets/memory/inMemory.go +++ b/plugins/websockets/memory/inMemory.go @@ -4,8 +4,8 @@ import ( "sync" "github.com/spiral/roadrunner/v2/pkg/bst" + websocketsv1 "github.com/spiral/roadrunner/v2/pkg/proto/websockets/v1beta" "github.com/spiral/roadrunner/v2/pkg/pubsub" - "github.com/spiral/roadrunner/v2/pkg/pubsub/message" "github.com/spiral/roadrunner/v2/plugins/logger" "google.golang.org/protobuf/proto" ) @@ -67,7 +67,7 @@ func (p *Plugin) Connections(topic string, res map[string]struct{}) { } } -func (p *Plugin) Next() (*message.Message, error) { +func (p *Plugin) Next() (*websocketsv1.Message, error) { msg := <-p.pushCh if msg == nil { return nil, nil @@ -76,7 +76,7 @@ func (p *Plugin) Next() (*message.Message, error) { p.RLock() defer p.RUnlock() - m := &message.Message{} + m := &websocketsv1.Message{} err := proto.Unmarshal(msg, m) if err != nil { return nil, err |