unistack-org/micro
3
0
Fork 3
Code Issues 16 Pull Requests 2 Actions Packages Projects Releases 239 Wiki Activity

broker/memory: add codec support (#1276)

Browse Source 
allow easy testing of other services with memory broker
and also allows to more deeply simulate real brokers

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2020-02-29 23:00:29 +03:00 committed by GitHub
parent 6b8930a960
commit 0754229878
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 35 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
broker/memory/memory.go
View File

@ -2,6 +2,7 @@
package memory
import (
"context"
"errors"
"math/rand"
"sync"
@ -9,6 +10,7 @@ import (
"github.com/google/uuid"
"github.com/micro/go-micro/v2/broker"
log "github.com/micro/go-micro/v2/logger"
maddr "github.com/micro/go-micro/v2/util/addr"
mnet "github.com/micro/go-micro/v2/util/net"
)
@ -23,8 +25,9 @@ type memoryBroker struct {
}
type memoryEvent struct {
opts broker.Options
topic string
message *broker.Message
message interface{}
}
type memorySubscriber struct {
@ -85,7 +88,7 @@ func (m *memoryBroker) Init(opts ...broker.Option) error {
return nil
}
func (m *memoryBroker) Publish(topic string, message *broker.Message, opts ...broker.PublishOption) error {
func (m *memoryBroker) Publish(topic string, msg *broker.Message, opts ...broker.PublishOption) error {
m.RLock()
if !m.connected {
m.RUnlock()
@ -98,9 +101,21 @@ func (m *memoryBroker) Publish(topic string, message *broker.Message, opts ...br
return nil
}
var v interface{}
if m.opts.Codec != nil {
buf, err := m.opts.Codec.Marshal(msg)
if err != nil {
return err
}
v = buf
} else {
v = msg
}
p := &memoryEvent{
topic: topic,
message: message,
message: v,
opts: m.opts,
}
for _, sub := range subs {
@ -163,7 +178,19 @@ func (m *memoryEvent) Topic() string {
}
func (m *memoryEvent) Message() *broker.Message {
return m.message
switch v := m.message.(type) {
case *broker.Message:
return v
case []byte:
msg := &broker.Message{}
if err := m.opts.Codec.Unmarshal(v, msg); err != nil {
log.Errorf("[memory]: failed to unmarshal: %v\n", err)
return nil
}
return msg
}
return nil
}
func (m *memoryEvent) Ack() error {
@ -184,7 +211,10 @@ func (m *memorySubscriber) Unsubscribe() error {
}
func NewBroker(opts ...broker.Option) broker.Broker {
var options broker.Options
options := broker.Options{
Context: context.Background(),
}
rand.Seed(time.Now().UnixNano())
for _, o := range opts {
o(&options)