unistack-org/micro
4
0
Fork 3
Code Issues 15 Pull Requests 2 Actions Packages Projects Releases 246 Wiki Activity

We don't need to export those things

Browse Source
This commit is contained in:
Asim 2016-03-17 22:52:49 +00:00
parent 771ff7fbff
commit 01a642410f
1 changed files with 23 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
broker/mock/mock.go
View File

@ -8,20 +8,20 @@ import (
"github.com/pborman/uuid" "github.com/pborman/uuid"
) )
type MockBroker struct { type mockBroker struct {
opts broker.Options opts broker.Options
sync.RWMutex sync.RWMutex
connected bool connected bool
Subscribers map[string][]*MockSubscriber Subscribers map[string][]*mockSubscriber
} }
type MockPublication struct { type mockPublication struct {
topic string topic string
message *broker.Message message *broker.Message
} }
type MockSubscriber struct { type mockSubscriber struct {
id string id string
topic string topic string
exit chan bool exit chan bool
@ -29,15 +29,15 @@ type MockSubscriber struct {
opts broker.SubscribeOptions opts broker.SubscribeOptions
} }
func (m *MockBroker) Options() broker.Options { func (m *mockBroker) Options() broker.Options {
return m.opts return m.opts
} }
func (m *MockBroker) Address() string { func (m *mockBroker) Address() string {
return "" return ""
} }
func (m *MockBroker) Connect() error { func (m *mockBroker) Connect() error {
m.Lock() m.Lock()
defer m.Unlock() defer m.Unlock()
@ -50,7 +50,7 @@ func (m *MockBroker) Connect() error {
return nil return nil
} }
func (m *MockBroker) Disconnect() error { func (m *mockBroker) Disconnect() error {
m.Lock() m.Lock()
defer m.Unlock() defer m.Unlock()
@ -63,14 +63,14 @@ func (m *MockBroker) Disconnect() error {
return nil return nil
} }
func (m *MockBroker) Init(opts ...broker.Option) error { func (m *mockBroker) Init(opts ...broker.Option) error {
for _, o := range opts { for _, o := range opts {
o(&m.opts) o(&m.opts)
} }
return nil return nil
} }
func (m *MockBroker) Publish(topic string, message *broker.Message, opts ...broker.PublishOption) error { func (m *mockBroker) Publish(topic string, message *broker.Message, opts ...broker.PublishOption) error {
m.Lock() m.Lock()
defer m.Unlock() defer m.Unlock()
@ -83,7 +83,7 @@ func (m *MockBroker) Publish(topic string, message *broker.Message, opts ...brok
return nil return nil
} }
p := &MockPublication{ p := &mockPublication{
topic: topic, topic: topic,
message: message, message: message,
} }
@ -97,7 +97,7 @@ func (m *MockBroker) Publish(topic string, message *broker.Message, opts ...brok
return nil return nil
} }
func (m *MockBroker) Subscribe(topic string, handler broker.Handler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { func (m *mockBroker) Subscribe(topic string, handler broker.Handler, opts ...broker.SubscribeOption) (broker.Subscriber, error) {
m.Lock() m.Lock()
defer m.Unlock() defer m.Unlock()
@ -110,7 +110,7 @@ func (m *MockBroker) Subscribe(topic string, handler broker.Handler, opts ...bro
o(&options) o(&options)
} }
sub := &MockSubscriber{ sub := &mockSubscriber{
exit: make(chan bool, 1), exit: make(chan bool, 1),
id: uuid.NewUUID().String(), id: uuid.NewUUID().String(),
topic: topic, topic: topic,
@ -123,7 +123,7 @@ func (m *MockBroker) Subscribe(topic string, handler broker.Handler, opts ...bro
go func() { go func() {
<-sub.exit <-sub.exit
m.Lock() m.Lock()
var newSubscribers []*MockSubscriber var newSubscribers []*mockSubscriber
for _, sb := range m.Subscribers[topic] { for _, sb := range m.Subscribers[topic] {
if sb.id == sub.id { if sb.id == sub.id {
continue continue
@ -137,31 +137,31 @@ func (m *MockBroker) Subscribe(topic string, handler broker.Handler, opts ...bro
return sub, nil return sub, nil
} }
func (m *MockBroker) String() string { func (m *mockBroker) String() string {
return "mock" return "mock"
} }
func (m *MockPublication) Topic() string { func (m *mockPublication) Topic() string {
return m.topic return m.topic
} }
func (m *MockPublication) Message() *broker.Message { func (m *mockPublication) Message() *broker.Message {
return m.message return m.message
} }
func (m *MockPublication) Ack() error { func (m *mockPublication) Ack() error {
return nil return nil
} }
func (m *MockSubscriber) Options() broker.SubscribeOptions { func (m *mockSubscriber) Options() broker.SubscribeOptions {
return m.opts return m.opts
} }
func (m *MockSubscriber) Topic() string { func (m *mockSubscriber) Topic() string {
return m.topic return m.topic
} }
func (m *MockSubscriber) Unsubscribe() error { func (m *mockSubscriber) Unsubscribe() error {
m.exit <- true m.exit <- true
return nil return nil
} }
@ -172,8 +172,8 @@ func NewBroker(opts ...broker.Option) broker.Broker {
o(&options) o(&options)
} }
return &MockBroker{ return &mockBroker{
opts: options, opts: options,
Subscribers: make(map[string][]*MockSubscriber), Subscribers: make(map[string][]*mockSubscriber),
} }
} }