This repository has been archived by the owner on Dec 20, 2022. It is now read-only.
-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mock.go
139 lines (110 loc) · 2.32 KB
/
mock.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
package mock
import (
"errors"
"sync"
"github.com/google/uuid"
"github.com/crypto-zero/go-micro/v2/server"
)
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.New().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...)
}