Skip to content

Commit 762e9a6

Browse files
add token bucket
1 parent f433e11 commit 762e9a6

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

src/token_bucket.ts

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
export class TokenBucket {
2+
private budget: number;
3+
constructor(allowance: number) {
4+
this.budget = allowance;
5+
}
6+
deposit(tokens: number) {
7+
this.budget += tokens;
8+
}
9+
10+
consume(tokens: number): boolean {
11+
if (tokens > this.budget) return false;
12+
13+
this.budget -= tokens;
14+
return true;
15+
}
16+
}
17+
18+
export const TOKEN_REFRESH_RATE = 0.1;
19+
export const INITIAL_SIZE = 1000;
20+
export const RETRY_COST = 1;

0 commit comments

Comments
 (0)