-
Notifications
You must be signed in to change notification settings - Fork 0
/
handler.go
111 lines (93 loc) · 1.77 KB
/
handler.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
package vsrpc
import (
"strings"
"sync"
)
type Handler interface {
Handle(*Call) error
}
type HandlerFunc func(*Call) error
func (fn HandlerFunc) Handle(call *Call) error {
if fn == nil {
method := call.Method()
return NoSuchMethodError{Method: method}
}
return fn(call)
}
var _ Handler = HandlerFunc(nil)
type HandlerMux struct {
mu sync.Mutex
db map[Method]Handler
}
func (mux *HandlerMux) Reset() {
if mux == nil {
return
}
mux.mu.Lock()
if mux.db != nil {
for method := range mux.db {
delete(mux.db, method)
}
}
mux.mu.Unlock()
}
func (mux *HandlerMux) AddFunc(fn func(*Call) error, methods ...Method) {
mux.Add(HandlerFunc(fn), methods...)
}
func (mux *HandlerMux) Add(h Handler, methods ...Method) {
if mux == nil || len(methods) <= 0 {
return
}
if h == nil {
h = HandlerFunc(nil)
}
mux.mu.Lock()
if mux.db == nil {
mux.db = make(map[Method]Handler, 64)
}
for _, method := range methods {
mux.db[method] = h
}
mux.mu.Unlock()
}
func (mux *HandlerMux) Remove(methods ...Method) {
if mux == nil || len(methods) <= 0 {
return
}
mux.mu.Lock()
if mux.db != nil {
for _, method := range methods {
delete(mux.db, method)
}
}
mux.mu.Unlock()
}
func (mux *HandlerMux) Find(method Method, inexact bool) Handler {
if mux == nil {
return nil
}
mux.mu.Lock()
h := mux.db[method]
for inexact && h == nil {
i := strings.LastIndex(string(method), ".")
if i < 0 {
break
}
method = method[:i]
h = mux.db[method+".*"]
}
if inexact && h == nil {
h = mux.db["*"]
}
mux.mu.Unlock()
return h
}
func (mux *HandlerMux) Handle(call *Call) error {
method := call.Method()
h := mux.Find(method, true)
if h == nil {
return NoSuchMethodError{Method: method}
}
return h.Handle(call)
}
var _ Handler = (*HandlerMux)(nil)