micro/server/mock/mock.go

140 lines
2.3 KiB
Go
Raw Normal View History

2016-04-16 02:02:55 +03:00
package mock
import (
"errors"
"sync"
"github.com/micro/go-micro/server"
"github.com/pborman/uuid"
)
type MockServer struct {
sync.Mutex
Running bool
Opts server.Options
Handlers map[string]server.Handler
Subscribers map[string][]server.Subscriber
}
var (
_ server.Server = NewServer()
)
func newMockServer(opts ...server.Option) *MockServer {
var options server.Options
for _, o := range opts {
o(&options)
}
return &MockServer{
Opts: options,
Handlers: make(map[string]server.Handler),
Subscribers: make(map[string][]server.Subscriber),
}
}
func (m *MockServer) Options() server.Options {
m.Lock()
defer m.Unlock()
return m.Opts
}
func (m *MockServer) Init(opts ...server.Option) error {
m.Lock()
defer m.Unlock()
for _, o := range opts {
o(&m.Opts)
}
return nil
}
func (m *MockServer) Handle(h server.Handler) error {
m.Lock()
defer m.Unlock()
if _, ok := m.Handlers[h.Name()]; ok {
return errors.New("Handler " + h.Name() + " already exists")
}
m.Handlers[h.Name()] = h
return nil
}
func (m *MockServer) NewHandler(h interface{}, opts ...server.HandlerOption) server.Handler {
var options server.HandlerOptions
for _, o := range opts {
o(&options)
}
return &MockHandler{
Id: uuid.NewUUID().String(),
Hdlr: h,
Opts: options,
}
}
func (m *MockServer) NewSubscriber(topic string, fn interface{}, opts ...server.SubscriberOption) server.Subscriber {
var options server.SubscriberOptions
for _, o := range opts {
o(&options)
}
return &MockSubscriber{
Id: topic,
Sub: fn,
Opts: options,
}
}
func (m *MockServer) Subscribe(sub server.Subscriber) error {
m.Lock()
defer m.Unlock()
subs := m.Subscribers[sub.Topic()]
subs = append(subs, sub)
m.Subscribers[sub.Topic()] = subs
return nil
}
func (m *MockServer) Register() error {
return nil
}
func (m *MockServer) Deregister() error {
return nil
}
func (m *MockServer) Start() error {
m.Lock()
defer m.Unlock()
if m.Running {
return errors.New("already running")
}
m.Running = true
return nil
}
func (m *MockServer) Stop() error {
m.Lock()
defer m.Unlock()
if !m.Running {
return errors.New("not running")
}
m.Running = false
return nil
}
func (m *MockServer) String() string {
return "mock"
}
func NewServer(opts ...server.Option) *MockServer {
return newMockServer(opts...)
}