FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
dotweb/middleware.go at develop · devfeel/dotweb · GitHub
devfeel
/
dotweb
Public
Notifications
You must be signed in to change notification settings
Fork
177
Star
1.4k
Code
Issues
13
Pull requests
4
Actions
Projects
Wiki
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Actions
Projects
Wiki
Security and quality
Insights
Expand file tree
Breadcrumbs
dotweb
/
middleware.go
Copy path
More file actions
More file actions
Latest commit
History
History
History
227 lines (203 loc) · 5.34 KB
Breadcrumbs
dotweb
/
middleware.go
Copy path
File metadata and controls
227 lines (203 loc) · 5.34 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
package
dotweb
import
(
"time"
"github.com/devfeel/dotweb/framework/convert"
)
const
(
middleware_App
=
"app"
middleware_Group
=
"group"
middleware_Router
=
"router"
)
type
(
// HttpModule global module in http server
// it will be no effect when websocket request or use offline mode
HttpModule
struct
{
Name
string
// OnBeginRequest is the first event in the execution chain
OnBeginRequest
func
(
Context
)
// OnEndRequest is the last event in the execution chain
OnEndRequest
func
(
Context
)
}
MiddlewareFunc
func
()
Middleware
// middleware execution priority:
// app > group > router
// Middleware middleware interface
Middleware
interface
{
Handle
(
ctx
Context
)
error
SetNext
(
m
Middleware
)
Next
(
ctx
Context
)
error
Exclude
(
routers
...
string
)
HasExclude
()
bool
ExistsExcludeRouter
(
router
string
)
bool
}
// BaseMiddlware is a shortcut for BaseMiddleware
// Deprecated: 由于该struct命名有误，将在2.0版本弃用，请大家尽快修改自己的middleware
BaseMiddlware
struct
{
BaseMiddleware
}
// BaseMiddleware is the base struct, user defined middleware should extend this
BaseMiddleware
struct
{
next
Middleware
excludeRouters
map
[
string
]
struct
{}
}
xMiddleware
struct
{
BaseMiddleware
IsEnd
bool
}
)
func
(
bm
*
BaseMiddleware
)
SetNext
(
m
Middleware
) {
bm
.
next
=
m
}
func
(
bm
*
BaseMiddleware
)
Next
(
ctx
Context
)
error
{
if
ctx
.
getMiddlewareStep
()
==
""
{
ctx
.
setMiddlewareStep
(
middleware_App
)
}
if
bm
.
next
==
nil
{
if
ctx
.
getMiddlewareStep
()
==
middleware_App
{
ctx
.
setMiddlewareStep
(
middleware_Group
)
if
len
(
ctx
.
RouterNode
().
GroupMiddlewares
())
>
0
{
return
ctx
.
RouterNode
().
GroupMiddlewares
()[
0
].
Handle
(
ctx
)
}
}
if
ctx
.
getMiddlewareStep
()
==
middleware_Group
{
ctx
.
setMiddlewareStep
(
middleware_Router
)
if
len
(
ctx
.
RouterNode
().
Middlewares
())
>
0
{
return
ctx
.
RouterNode
().
Middlewares
()[
0
].
Handle
(
ctx
)
}
}
if
ctx
.
getMiddlewareStep
()
==
middleware_Router
{
return
ctx
.
Handler
()(
ctx
)
}
}
else
{
// check exclude config
if
ctx
.
RouterNode
().
Node
().
hasExcludeMiddleware
&&
bm
.
next
.
HasExclude
() {
if
bm
.
next
.
ExistsExcludeRouter
(
ctx
.
RouterNode
().
Node
().
fullPath
) {
return
bm
.
next
.
Next
(
ctx
)
}
}
return
bm
.
next
.
Handle
(
ctx
)
}
return
nil
}
// Exclude Exclude this middleware with router
func
(
bm
*
BaseMiddleware
)
Exclude
(
routers
...
string
) {
if
bm
.
excludeRouters
==
nil
{
bm
.
excludeRouters
=
make
(
map
[
string
]
struct
{})
}
for
_
,
v
:=
range
routers
{
bm
.
excludeRouters
[
v
]
=
struct
{}{}
}
}
// HasExclude check has set exclude router
func
(
bm
*
BaseMiddleware
)
HasExclude
()
bool
{
if
bm
.
excludeRouters
==
nil
{
return
false
}
if
len
(
bm
.
excludeRouters
)
>
0
{
return
true
}
else
{
return
false
}
}
// ExistsExcludeRouter check is exists router in exclude map
func
(
bm
*
BaseMiddleware
)
ExistsExcludeRouter
(
router
string
)
bool
{
if
bm
.
excludeRouters
==
nil
{
return
false
}
_
,
exists
:=
bm
.
excludeRouters
[
router
]
return
exists
}
func
getIgnoreFaviconModule
()
*
HttpModule
{
return
&
HttpModule
{
Name
:
"IgnoreFavicon"
,
OnBeginRequest
:
func
(
ctx
Context
) {
if
ctx
.
Request
().
Path
()
==
"/favicon.ico"
{
ctx
.
End
()
}
},
}
}
func
(
x
*
xMiddleware
)
Handle
(
ctx
Context
)
error
{
if
ctx
.
getMiddlewareStep
()
==
""
{
ctx
.
setMiddlewareStep
(
middleware_App
)
}
if
x
.
IsEnd
{
return
ctx
.
Handler
()(
ctx
)
}
return
x
.
Next
(
ctx
)
}
type
RequestLogMiddleware
struct
{
BaseMiddleware
}
func
(
m
*
RequestLogMiddleware
)
Handle
(
ctx
Context
)
error
{
var
timeDuration
time.
Duration
var
timeTaken
uint64
err
:=
m
.
Next
(
ctx
)
if
ctx
.
Items
().
Exists
(
ItemKeyHandleDuration
) {
var
errParse
error
timeDuration
,
errParse
=
time
.
ParseDuration
(
ctx
.
Items
().
GetString
(
ItemKeyHandleDuration
))
if
errParse
!=
nil
{
timeTaken
=
0
}
else
{
timeTaken
=
uint64
(
timeDuration
/
time
.
Millisecond
)
}
}
else
{
var
begin
time.
Time
beginVal
,
exists
:=
ctx
.
Items
().
Get
(
ItemKeyHandleStartTime
)
if
!
exists
{
begin
=
time
.
Now
()
}
else
{
begin
=
beginVal
.(time.
Time
)
}
timeTaken
=
uint64
(
time
.
Now
().
Sub
(
begin
)
/
time
.
Millisecond
)
}
log
:=
ctx
.
Request
().
Url
()
+
" "
+
logContext
(
ctx
,
timeTaken
)
ctx
.
HttpServer
().
Logger
().
Debug
(
log
,
LogTarget_HttpRequest
)
return
err
}
// get default log string
func
logContext
(
ctx
Context
,
timetaken
uint64
)
string
{
var
reqbytelen
,
resbytelen
,
method
,
proto
,
status
,
userip
string
if
ctx
!=
nil
{
reqbytelen
=
convert
.
Int642String
(
ctx
.
Request
().
ContentLength
)
resbytelen
=
convert
.
Int642String
(
ctx
.
Response
().
Size
)
method
=
ctx
.
Request
().
Method
proto
=
ctx
.
Request
().
Proto
status
=
convert
.
Int2String
(
ctx
.
Response
().
Status
)
userip
=
ctx
.
RemoteIP
()
}
log
:=
method
+
" "
log
+=
userip
+
" "
log
+=
proto
+
" "
log
+=
status
+
" "
log
+=
reqbytelen
+
" "
log
+=
resbytelen
+
" "
log
+=
convert
.
UInt642String
(
timetaken
)
return
log
}
type
TimeoutHookMiddleware
struct
{
BaseMiddleware
HookHandle
StandardHandle
TimeoutDuration
time.
Duration
}
func
(
m
*
TimeoutHookMiddleware
)
Handle
(
ctx
Context
)
error
{
var
begin
time.
Time
if
m
.
HookHandle
!=
nil
{
beginVal
,
exists
:=
ctx
.
Items
().
Get
(
ItemKeyHandleStartTime
)
if
!
exists
{
begin
=
time
.
Now
()
}
else
{
begin
=
beginVal
.(time.
Time
)
}
}
// Do next
err
:=
m
.
Next
(
ctx
)
if
m
.
HookHandle
!=
nil
{
realDuration
:=
time
.
Now
().
Sub
(
begin
)
ctx
.
Items
().
Set
(
ItemKeyHandleDuration
,
realDuration
)
if
realDuration
>
m
.
TimeoutDuration
{
m
.
HookHandle
(
ctx
)
}
}
return
err
}
Back
|
FazBrowse Home
|
New Git URL