@scallop-io/sui-scallop-sdk 1.4.15-rc.3 → 1.4.16

package/src/utils/tokenBucket.ts

@@ -1,68 +0,0 @@
-import { DEFAULT_INTERVAL_IN_MS } from 'src/constants/tokenBucket';
-
-class TokenBucket {
-  private tokensPerInterval: number;
-  private interval: number;
-  private tokens: number;
-  private lastRefill: number;
-
-  constructor(tokensPerInterval: number, intervalInMs: number) {
-    this.tokensPerInterval = tokensPerInterval;
-    this.interval = intervalInMs;
-    this.tokens = tokensPerInterval;
-    this.lastRefill = Date.now();
-  }
-
-  refill() {
-    const now = Date.now();
-    const elapsed = now - this.lastRefill;
-
-    if (elapsed >= this.interval) {
-      const tokensToAdd =
-        Math.floor(elapsed / this.interval) * this.tokensPerInterval;
-      this.tokens = Math.min(this.tokens + tokensToAdd, this.tokensPerInterval);
-
-      // Update lastRefill to reflect the exact time of the last "refill"
-      this.lastRefill += Math.floor(elapsed / this.interval) * this.interval;
-    }
-  }
-
-  removeTokens(count: number) {
-    this.refill();
-    if (this.tokens >= count) {
-      this.tokens -= count;
-      return true;
-    }
-    return false;
-  }
-}
-
-const callWithRateLimit = async <T>(
-  tokenBucket: TokenBucket,
-  fn: () => Promise<T>,
-  retryDelayInMs = DEFAULT_INTERVAL_IN_MS,
-  maxRetries = 15,
-  backoffFactor = 1.25 // The factor by which to increase the delay
-): Promise<T | null> => {
-  let retries = 0;
-
-  const tryRequest = async (): Promise<T | null> => {
-    if (tokenBucket.removeTokens(1)) {
-      const result = await fn();
-      return result;
-    } else if (retries < maxRetries) {
-      retries++;
-      const delay = retryDelayInMs * Math.pow(backoffFactor, retries);
-      // console.error(`Rate limit exceeded, retrying in ${delay} ms`);
-      await new Promise((resolve) => setTimeout(resolve, delay));
-      return tryRequest();
-    } else {
-      console.error('Maximum retries reached');
-      return null;
-    }
-  };
-
-  return tryRequest();
-};
-
-export { TokenBucket, callWithRateLimit };