-
Notifications
You must be signed in to change notification settings - Fork 0
/
middleware.js
52 lines (44 loc) · 989 Bytes
/
middleware.js
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
const { v4: uuid } = require('uuid')
const perfy = require('perfy')
const defaults = {}
const handleEnd = (req) => {
const trackingId =
req.context && req.context.performance
? req.context.performance.trackingId
: null
if (trackingId) {
req.context.performance.timing = perfy.end(trackingId)
req.context.performance.trackingId = null
}
}
module.exports = options => {
options = { ...defaults, ...options }
const middleware = {
start: (req, _res, next) => {
req.context = {
...req.context,
performance: {
trackingId: uuid()
}
}
perfy.start(req.context.performance.trackingId)
if (next) {
next()
}
},
end: (req, _res, next) => {
handleEnd(req)
if (next) {
next()
}
},
error: (err, req, _res, next) => {
handleEnd(req)
if (next) {
next(err)
}
}
}
middleware.options = options
return middleware
}