noselasd/libucore
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small optimization change when tickets < period

Browse Source
This commit is contained in:
Nils O. Selåsdal
2013-10-15 20:14:15 +02:00
parent 28514c0a57
commit 9c15b73774
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/rate_limit.c
+10 -2
View File
@@ -4,7 +4,13 @@
void uc_ratelimit_init(struct RateLimit *r, long tickets, long period, long current_ts) void uc_ratelimit_init(struct RateLimit *r, long tickets, long period, long current_ts)
{ {
r->tickets = tickets; r->tickets = tickets;
if (tickets > period) {
r->ticket_cost = tickets * period; r->ticket_cost = tickets * period;
} else {
r->ticket_cost = period;
}
r->period = period; r->period = period;
r->funds = r->ticket_cost * tickets; r->funds = r->ticket_cost * tickets;
r->last_ts = current_ts; r->last_ts = current_ts;
@@ -19,8 +25,9 @@ void uc_ratelimit_reset(struct RateLimit *r, long current_ts)
{ {
r->funds = r->ticket_cost * r->tickets; r->funds = r->ticket_cost * r->tickets;
if (current_ts) if (current_ts) {
r->last_ts = current_ts; r->last_ts = current_ts;
}
} }
int uc_ratelimit_allow(struct RateLimit *r, long current_ts) int uc_ratelimit_allow(struct RateLimit *r, long current_ts)
@@ -45,8 +52,9 @@ int uc_ratelimit_allow(struct RateLimit *r, long current_ts)
((r->tickets * r->ticket_cost) / r->period); ((r->tickets * r->ticket_cost) / r->period);
//throttle handing out tickets //throttle handing out tickets
if (r->funds > r->ticket_cost * r->tickets) if (r->funds > r->ticket_cost * r->tickets) {
r->funds = r->ticket_cost * r->tickets; r->funds = r->ticket_cost * r->tickets;
}
//If we have enough to buy atleast one ticket, we can allow //If we have enough to buy atleast one ticket, we can allow
if (r->funds >= r->ticket_cost) { if (r->funds >= r->ticket_cost) {