Add mock broker
This commit is contained in:
parent
7126dc1238
commit
771ff7fbff
179
broker/mock/mock.go
Normal file
179
broker/mock/mock.go
Normal file
@ -0,0 +1,179 @@
|
|||||||
|
package mock
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/micro/go-micro/broker"
|
||||||
|
"github.com/pborman/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
type MockBroker struct {
|
||||||
|
opts broker.Options
|
||||||
|
|
||||||
|
sync.RWMutex
|
||||||
|
connected bool
|
||||||
|
Subscribers map[string][]*MockSubscriber
|
||||||
|
}
|
||||||
|
|
||||||
|
type MockPublication struct {
|
||||||
|
topic string
|
||||||
|
message *broker.Message
|
||||||
|
}
|
||||||
|
|
||||||
|
type MockSubscriber struct {
|
||||||
|
id string
|
||||||
|
topic string
|
||||||
|
exit chan bool
|
||||||
|
handler broker.Handler
|
||||||
|
opts broker.SubscribeOptions
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockBroker) Options() broker.Options {
|
||||||
|
return m.opts
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockBroker) Address() string {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockBroker) Connect() error {
|
||||||
|
m.Lock()
|
||||||
|
defer m.Unlock()
|
||||||
|
|
||||||
|
if m.connected {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
m.connected = true
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockBroker) Disconnect() error {
|
||||||
|
m.Lock()
|
||||||
|
defer m.Unlock()
|
||||||
|
|
||||||
|
if !m.connected {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
m.connected = false
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockBroker) Init(opts ...broker.Option) error {
|
||||||
|
for _, o := range opts {
|
||||||
|
o(&m.opts)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockBroker) Publish(topic string, message *broker.Message, opts ...broker.PublishOption) error {
|
||||||
|
m.Lock()
|
||||||
|
defer m.Unlock()
|
||||||
|
|
||||||
|
if !m.connected {
|
||||||
|
return errors.New("not connected")
|
||||||
|
}
|
||||||
|
|
||||||
|
subs, ok := m.Subscribers[topic]
|
||||||
|
if !ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
p := &MockPublication{
|
||||||
|
topic: topic,
|
||||||
|
message: message,
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, sub := range subs {
|
||||||
|
if err := sub.handler(p); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockBroker) Subscribe(topic string, handler broker.Handler, opts ...broker.SubscribeOption) (broker.Subscriber, error) {
|
||||||
|
m.Lock()
|
||||||
|
defer m.Unlock()
|
||||||
|
|
||||||
|
if !m.connected {
|
||||||
|
return nil, errors.New("not connected")
|
||||||
|
}
|
||||||
|
|
||||||
|
var options broker.SubscribeOptions
|
||||||
|
for _, o := range opts {
|
||||||
|
o(&options)
|
||||||
|
}
|
||||||
|
|
||||||
|
sub := &MockSubscriber{
|
||||||
|
exit: make(chan bool, 1),
|
||||||
|
id: uuid.NewUUID().String(),
|
||||||
|
topic: topic,
|
||||||
|
handler: handler,
|
||||||
|
opts: options,
|
||||||
|
}
|
||||||
|
|
||||||
|
m.Subscribers[topic] = append(m.Subscribers[topic], sub)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
<-sub.exit
|
||||||
|
m.Lock()
|
||||||
|
var newSubscribers []*MockSubscriber
|
||||||
|
for _, sb := range m.Subscribers[topic] {
|
||||||
|
if sb.id == sub.id {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
newSubscribers = append(newSubscribers, sb)
|
||||||
|
}
|
||||||
|
m.Subscribers[topic] = newSubscribers
|
||||||
|
m.Unlock()
|
||||||
|
}()
|
||||||
|
|
||||||
|
return sub, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockBroker) String() string {
|
||||||
|
return "mock"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockPublication) Topic() string {
|
||||||
|
return m.topic
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockPublication) Message() *broker.Message {
|
||||||
|
return m.message
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockPublication) Ack() error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockSubscriber) Options() broker.SubscribeOptions {
|
||||||
|
return m.opts
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockSubscriber) Topic() string {
|
||||||
|
return m.topic
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MockSubscriber) Unsubscribe() error {
|
||||||
|
m.exit <- true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewBroker(opts ...broker.Option) broker.Broker {
|
||||||
|
var options broker.Options
|
||||||
|
for _, o := range opts {
|
||||||
|
o(&options)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &MockBroker{
|
||||||
|
opts: options,
|
||||||
|
Subscribers: make(map[string][]*MockSubscriber),
|
||||||
|
}
|
||||||
|
}
|
53
broker/mock/mock_test.go
Normal file
53
broker/mock/mock_test.go
Normal file
@ -0,0 +1,53 @@
|
|||||||
|
package mock
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/micro/go-micro/broker"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestBroker(t *testing.T) {
|
||||||
|
b := NewBroker()
|
||||||
|
|
||||||
|
if err := b.Connect(); err != nil {
|
||||||
|
t.Fatal("Unexpected connect error %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
topic := "test"
|
||||||
|
count := 10
|
||||||
|
|
||||||
|
fn := func(p broker.Publication) error {
|
||||||
|
m := p.Message()
|
||||||
|
t.Logf("Received message id %s %+v for topic %s", m.Header["id"], m, p.Topic())
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
sub, err := b.Subscribe(topic, fn)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Unexpected error subscribing %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := 0; i < count; i++ {
|
||||||
|
message := &broker.Message{
|
||||||
|
Header: map[string]string{
|
||||||
|
"foo": "bar",
|
||||||
|
"id": fmt.Sprintf("%d", i),
|
||||||
|
},
|
||||||
|
Body: []byte(`hello world`),
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Logf("Sending message %d %+v for topic %s", i, message, topic)
|
||||||
|
if err := b.Publish(topic, message); err != nil {
|
||||||
|
t.Fatalf("Unexpected error publishing %d", i)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := sub.Unsubscribe(); err != nil {
|
||||||
|
t.Fatalf("Unexpected error unsubscribing from %s: %v", topic, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := b.Disconnect(); err != nil {
|
||||||
|
t.Fatalf("Unexpected connect error %v", err)
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user