2 * Ratelimiting calculations
4 * Copyright IBM, Corp. 2011
7 * Stefan Hajnoczi <stefanha@linux.vnet.ibm.com>
9 * This work is licensed under the terms of the GNU LGPL, version 2 or later.
10 * See the COPYING.LIB file in the top-level directory.
14 #ifndef QEMU_RATELIMIT_H
15 #define QEMU_RATELIMIT_H
18 int64_t slice_start_time
;
19 int64_t slice_end_time
;
25 /** Calculate and return delay for next request in ns
27 * Record that we sent @p n data units. If we may send more data units
28 * in the current time slice, return 0 (i.e. no delay). Otherwise
29 * return the amount of time (in ns) until the start of the next time
30 * slice that will permit sending the next chunk of data.
32 * Recording sent data units even after exceeding the quota is
33 * permitted; the time slice will be extended accordingly.
35 static inline int64_t ratelimit_calculate_delay(RateLimit
*limit
, uint64_t n
)
37 int64_t now
= qemu_clock_get_ns(QEMU_CLOCK_REALTIME
);
38 uint64_t delay_slices
;
40 assert(limit
->slice_quota
&& limit
->slice_ns
);
42 if (limit
->slice_end_time
< now
) {
43 /* Previous, possibly extended, time slice finished; reset the
45 limit
->slice_start_time
= now
;
46 limit
->slice_end_time
= now
+ limit
->slice_ns
;
47 limit
->dispatched
= 0;
50 limit
->dispatched
+= n
;
51 if (limit
->dispatched
< limit
->slice_quota
) {
52 /* We may send further data within the current time slice, no
53 * need to delay the next request. */
57 /* Quota exceeded. Calculate the next time slice we may start
58 * sending data again. */
59 delay_slices
= (limit
->dispatched
+ limit
->slice_quota
- 1) /
61 limit
->slice_end_time
= limit
->slice_start_time
+
62 delay_slices
* limit
->slice_ns
;
63 return limit
->slice_end_time
- now
;
66 static inline void ratelimit_set_speed(RateLimit
*limit
, uint64_t speed
,
69 limit
->slice_ns
= slice_ns
;
70 limit
->slice_quota
= MAX(((double)speed
* slice_ns
) / 1000000000ULL, 1);