squash

Author: baileympearson

src/token_bucket.ts

@@ -1,4 +1,4 @@
-class TokenBucket {
+export class TokenBucket {
   private budget: number;
   constructor(allowance: number) {
     this.budget = allowance;
@@ -15,22 +15,6 @@ class TokenBucket {
   }
 }
 
-const TOKEN_REFRESH_RATE = 0.1;
-const INITIAL_SIZE = 1000;
-const RETRY_COST = 1;
-
-export class RetryTokenBucket {
-  private bucket = new TokenBucket(INITIAL_SIZE);
-
-  onCommandSuccess(retry: boolean) {
-    this.bucket.deposit(TOKEN_REFRESH_RATE);
-
-    if (retry) {
-      this.bucket.deposit(RETRY_COST);
-    }
-  }
-
-  canRetry() {
-    return this.bucket.consume(RETRY_COST);
-  }
-}
+export const TOKEN_REFRESH_RATE = 0.1;
+export const INITIAL_SIZE = 1000;
+export const RETRY_COST = 1;