]>
Commit | Line | Data |
---|---|---|
4d4e5cd4 | 1 | import * as express from 'express' |
eb080476 | 2 | import * as Bluebird from 'bluebird' |
d3cd34be | 3 | |
65fcc311 | 4 | import { |
15a30294 | 5 | AbstractRequestScheduler, |
65fcc311 C |
6 | getRequestScheduler, |
7 | getRequestVideoQaduScheduler, | |
8 | getRequestVideoEventScheduler | |
9 | } from '../../lib' | |
eb080476 | 10 | import { authenticate, ensureIsAdmin, asyncMiddleware } from '../../middlewares' |
e636eb88 | 11 | import { RequestSchedulerStatsAttributes } from '../../../shared' |
d3cd34be | 12 | |
e636eb88 | 13 | const requestSchedulerRouter = express.Router() |
d3cd34be | 14 | |
e636eb88 | 15 | requestSchedulerRouter.get('/stats', |
65fcc311 C |
16 | authenticate, |
17 | ensureIsAdmin, | |
eb080476 | 18 | asyncMiddleware(getRequestSchedulersStats) |
d3cd34be C |
19 | ) |
20 | ||
21 | // --------------------------------------------------------------------------- | |
22 | ||
65fcc311 | 23 | export { |
e636eb88 | 24 | requestSchedulerRouter |
65fcc311 | 25 | } |
d3cd34be C |
26 | |
27 | // --------------------------------------------------------------------------- | |
28 | ||
eb080476 C |
29 | async function getRequestSchedulersStats (req: express.Request, res: express.Response, next: express.NextFunction) { |
30 | const result = await Bluebird.props({ | |
e636eb88 C |
31 | requestScheduler: buildRequestSchedulerStats(getRequestScheduler()), |
32 | requestVideoQaduScheduler: buildRequestSchedulerStats(getRequestVideoQaduScheduler()), | |
33 | requestVideoEventScheduler: buildRequestSchedulerStats(getRequestVideoEventScheduler()) | |
d3cd34be | 34 | }) |
eb080476 C |
35 | |
36 | return res.json(result) | |
d3cd34be | 37 | } |
99fdec46 C |
38 | |
39 | // --------------------------------------------------------------------------- | |
40 | ||
eb080476 C |
41 | async function buildRequestSchedulerStats (requestScheduler: AbstractRequestScheduler<any>) { |
42 | const count = await requestScheduler.remainingRequestsCount() | |
43 | ||
44 | const result: RequestSchedulerStatsAttributes = { | |
45 | totalRequests: count, | |
46 | requestsLimitPods: requestScheduler.limitPods, | |
47 | requestsLimitPerPod: requestScheduler.limitPerPod, | |
48 | remainingMilliSeconds: requestScheduler.remainingMilliSeconds(), | |
49 | milliSecondsInterval: requestScheduler.requestInterval | |
50 | } | |
51 | ||
52 | return result | |
99fdec46 | 53 | } |