]>
Commit | Line | Data |
---|---|---|
1 | import * as express from 'express' | |
2 | import { parallel } from 'async' | |
3 | ||
4 | import { | |
5 | BaseRequestScheduler, | |
6 | getRequestScheduler, | |
7 | getRequestVideoQaduScheduler, | |
8 | getRequestVideoEventScheduler | |
9 | } from '../../lib' | |
10 | import { authenticate, ensureIsAdmin } from '../../middlewares' | |
11 | ||
12 | const requestsRouter = express.Router() | |
13 | ||
14 | requestsRouter.get('/stats', | |
15 | authenticate, | |
16 | ensureIsAdmin, | |
17 | getStatsRequests | |
18 | ) | |
19 | ||
20 | // --------------------------------------------------------------------------- | |
21 | ||
22 | export { | |
23 | requestsRouter | |
24 | } | |
25 | ||
26 | // --------------------------------------------------------------------------- | |
27 | ||
28 | function getStatsRequests (req: express.Request, res: express.Response, next: express.NextFunction) { | |
29 | parallel({ | |
30 | requestScheduler: buildRequestSchedulerFunction(getRequestScheduler()), | |
31 | requestVideoQaduScheduler: buildRequestSchedulerFunction(getRequestVideoQaduScheduler()), | |
32 | requestVideoEventScheduler: buildRequestSchedulerFunction(getRequestVideoEventScheduler()) | |
33 | }, function (err, result) { | |
34 | if (err) return next(err) | |
35 | ||
36 | return res.json(result) | |
37 | }) | |
38 | } | |
39 | ||
40 | // --------------------------------------------------------------------------- | |
41 | ||
42 | function buildRequestSchedulerFunction (requestScheduler: BaseRequestScheduler) { | |
43 | return function (callback) { | |
44 | requestScheduler.remainingRequestsCount(function (err, count) { | |
45 | if (err) return callback(err) | |
46 | ||
47 | const result = { | |
48 | totalRequests: count, | |
49 | requestsLimitPods: requestScheduler.limitPods, | |
50 | requestsLimitPerPod: requestScheduler.limitPerPod, | |
51 | remainingMilliSeconds: requestScheduler.remainingMilliSeconds(), | |
52 | milliSecondsInterval: requestScheduler.requestInterval | |
53 | } | |
54 | ||
55 | return callback(null, result) | |
56 | }) | |
57 | } | |
58 | } |