-
Notifications
You must be signed in to change notification settings - Fork 0
/
register.go
121 lines (107 loc) · 3.64 KB
/
register.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
/*
Nging is a toolbox for webmasters
Copyright (C) 2018-present Wenhui Shen <swh@admpub.com>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published
by the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
package route
import (
"strings"
"github.com/webx-top/echo"
)
func NewRegister(e *echo.Echo, groupNamers ...func(string) string) *Register {
return &Register{
Echo: e,
Handlers: []func(echo.RouteRegister){},
GroupHandlers: map[string][]func(echo.RouteRegister){},
GroupMiddlewares: map[string][]interface{}{},
PrefixMiddlewares: map[string][]interface{}{},
GroupNamers: groupNamers,
}
}
type Register struct {
Echo *echo.Echo
RootGroup string
Handlers []func(echo.RouteRegister)
GroupHandlers map[string][]func(echo.RouteRegister)
GroupNamers []func(string) string
GroupMiddlewares map[string][]interface{}
PrefixMiddlewares map[string][]interface{}
}
func (r *Register) AddGroupNamer(namers ...func(string) string) {
r.GroupNamers = append(r.GroupNamers, namers...)
}
func (r *Register) SetGroupNamer(namers ...func(string) string) {
r.GroupNamers = namers
}
func (r *Register) Apply() {
e := r.Echo
for _, register := range r.Handlers {
register(e)
}
var groupDefaultMiddlewares []interface{}
middlewares, ok := r.GroupMiddlewares[`*`]
if ok {
groupDefaultMiddlewares = append(groupDefaultMiddlewares, middlewares...)
}
for group, handlers := range r.GroupHandlers {
for _, namer := range r.GroupNamers {
group = namer(group)
}
g := e.Group(group)
if group != r.RootGroup { // 组名为空时,为顶层组
g.Use(groupDefaultMiddlewares...)
}
for prefix, middlewares := range r.PrefixMiddlewares {
if strings.HasPrefix(group, prefix) {
g.Use(middlewares...)
}
}
middlewares, ok := r.GroupMiddlewares[group]
if ok {
g.Use(middlewares...)
}
for _, register := range handlers {
register(g)
}
}
}
func (r *Register) Use(groupName string, middlewares ...interface{}) {
if groupName != `*` && strings.HasSuffix(groupName, `*`) {
groupName = strings.TrimRight(groupName, `*`)
if _, ok := r.PrefixMiddlewares[groupName]; !ok {
r.PrefixMiddlewares[groupName] = []interface{}{}
}
r.PrefixMiddlewares[groupName] = append(r.PrefixMiddlewares[groupName], middlewares...)
return
}
if _, ok := r.GroupMiddlewares[groupName]; !ok {
r.GroupMiddlewares[groupName] = []interface{}{}
}
r.GroupMiddlewares[groupName] = append(r.GroupMiddlewares[groupName], middlewares...)
}
func (r *Register) Register(fn func(echo.RouteRegister)) {
r.Handlers = append(r.Handlers, fn)
}
func (r *Register) RegisterToGroup(groupName string, fn func(echo.RouteRegister), middlewares ...interface{}) {
_, ok := r.GroupHandlers[groupName]
if !ok {
r.GroupHandlers[groupName] = []func(echo.RouteRegister){}
}
if len(middlewares) > 0 {
_, ok = r.GroupMiddlewares[groupName]
if !ok {
r.GroupMiddlewares[groupName] = []interface{}{}
}
r.GroupMiddlewares[groupName] = append(r.GroupMiddlewares[groupName], middlewares...)
}
r.GroupHandlers[groupName] = append(r.GroupHandlers[groupName], fn)
}