@zeitar/throttle 1.0.0

package/dist/policy/SlidingWindow.js

@@ -0,0 +1,130 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SlidingWindow = void 0;
+const TimeUtil_1 = require("../util/TimeUtil");
+/**
+ * Serializable state for SlidingWindowLimiter.
+ *
+ * Tracks hit counts across current and previous windows to implement
+ * a sliding window algorithm.
+ */
+class SlidingWindow {
+    constructor(id, intervalInSeconds, hitCount, hitCountForLastWindow, windowEndAt) {
+        this.id = id;
+        this.intervalInSeconds = intervalInSeconds;
+        this.hitCount = hitCount ?? 0;
+        this.hitCountForLastWindow = hitCountForLastWindow ?? 0;
+        this.windowEndAt = windowEndAt ?? TimeUtil_1.TimeUtil.now() + intervalInSeconds;
+    }
+    getId() {
+        return this.id;
+    }
+    /**
+     * Get the expiration time in seconds since epoch.
+     */
+    getExpirationTime() {
+        // Expires after current window ends plus one more interval
+        return Math.ceil(this.windowEndAt + this.intervalInSeconds);
+    }
+    /**
+     * Create a new window from a previous window (window transition).
+     */
+    static createFromPreviousWindow(id, intervalInSeconds, previousWindow, now) {
+        // The previous window's current hits become the "last window" hits
+        const newWindowEndAt = now + intervalInSeconds;
+        return new SlidingWindow(id, intervalInSeconds, 0, // New window starts with 0 hits
+        previousWindow.hitCount, // Previous current becomes last
+        newWindowEndAt);
+    }
+    /**
+     * Check if the window has expired.
+     */
+    isExpired(now = TimeUtil_1.TimeUtil.now()) {
+        return now >= this.windowEndAt + this.intervalInSeconds;
+    }
+    /**
+     * Add hits to the current window.
+     */
+    add(hits) {
+        this.hitCount += hits;
+    }
+    /**
+     * Get the sliding window hit count at a given time.
+     *
+     * Uses the formula:
+     * (previousWindowHits * (1 - percentIntoCurrentWindow)) + currentWindowHits
+     */
+    getHitCount(now = TimeUtil_1.TimeUtil.now()) {
+        // If we're past the current window end, we need a transition
+        if (now >= this.windowEndAt) {
+            return this.hitCount;
+        }
+        // Calculate how far we are into the current window
+        const windowStartAt = this.windowEndAt - this.intervalInSeconds;
+        const timeIntoWindow = now - windowStartAt;
+        const percentIntoWindow = timeIntoWindow / this.intervalInSeconds;
+        // Sliding window formula
+        const previousContribution = this.hitCountForLastWindow * (1 - percentIntoWindow);
+        const currentContribution = this.hitCount;
+        return Math.floor(previousContribution + currentContribution);
+    }
+    /**
+     * Get the current window's hit count (not sliding).
+     */
+    getCurrentWindowHitCount() {
+        return this.hitCount;
+    }
+    /**
+     * Get the previous window's hit count.
+     */
+    getPreviousWindowHitCount() {
+        return this.hitCountForLastWindow;
+    }
+    /**
+     * Get when the current window ends.
+     */
+    getWindowEndAt() {
+        return this.windowEndAt;
+    }
+    /**
+     * Get the interval in seconds.
+     */
+    getInterval() {
+        return this.intervalInSeconds;
+    }
+    /**
+     * Calculate time needed for N tokens to become available.
+     */
+    calculateTimeForTokens(maxSize, tokens, now = TimeUtil_1.TimeUtil.now()) {
+        const currentCount = this.getHitCount(now);
+        const available = Math.max(0, maxSize - currentCount);
+        if (available >= tokens) {
+            return 0; // Already available
+        }
+        // Need to wait - in sliding window, we need to wait until the window slides
+        // enough that we have space
+        const windowStartAt = this.windowEndAt - this.intervalInSeconds;
+        // Simplified: wait for next window
+        return this.windowEndAt - now;
+    }
+    /**
+     * Convert to JSON for serialization.
+     */
+    toJSON() {
+        return {
+            id: this.id,
+            hitCount: this.hitCount,
+            hitCountForLastWindow: this.hitCountForLastWindow,
+            windowEndAt: this.windowEndAt,
+            intervalInSeconds: this.intervalInSeconds,
+        };
+    }
+    /**
+     * Create from JSON.
+     */
+    static fromJSON(data) {
+        return new SlidingWindow(data.id, data.intervalInSeconds, data.hitCount, data.hitCountForLastWindow, data.windowEndAt);
+    }
+}
+exports.SlidingWindow = SlidingWindow;
+//# sourceMappingURL=SlidingWindow.js.map

package/dist/policy/SlidingWindow.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SlidingWindow.js","sourceRoot":"","sources":["../../src/policy/SlidingWindow.ts"],"names":[],"mappings":";;;AACA,+CAA4C;AAE5C;;;;;GAKG;AACH,MAAa,aAAa;IAOxB,YACE,EAAU,EACV,iBAAyB,EACzB,QAAiB,EACjB,qBAA8B,EAC9B,WAAoB;QAEpB,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,QAAQ,GAAG,QAAQ,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,qBAAqB,GAAG,qBAAqB,IAAI,CAAC,CAAC;QACxD,IAAI,CAAC,WAAW,GAAG,WAAW,IAAI,mBAAQ,CAAC,GAAG,EAAE,GAAG,iBAAiB,CAAC;IACvE,CAAC;IAED,KAAK;QACH,OAAO,IAAI,CAAC,EAAE,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,2DAA2D;QAC3D,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9D,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,wBAAwB,CAC7B,EAAU,EACV,iBAAyB,EACzB,cAA6B,EAC7B,GAAW;QAEX,mEAAmE;QACnE,MAAM,cAAc,GAAG,GAAG,GAAG,iBAAiB,CAAC;QAC/C,OAAO,IAAI,aAAa,CACtB,EAAE,EACF,iBAAiB,EACjB,CAAC,EAAE,gCAAgC;QACnC,cAAc,CAAC,QAAQ,EAAE,gCAAgC;QACzD,cAAc,CACf,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,MAAc,mBAAQ,CAAC,GAAG,EAAE;QACpC,OAAO,GAAG,IAAI,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC;IAC1D,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,IAAY;QACd,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC;IACxB,CAAC;IAED;;;;;OAKG;IACH,WAAW,CAAC,MAAc,mBAAQ,CAAC,GAAG,EAAE;QACtC,6DAA6D;QAC7D,IAAI,GAAG,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC,QAAQ,CAAC;QACvB,CAAC;QAED,mDAAmD;QACnD,MAAM,aAAa,GAAG,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAChE,MAAM,cAAc,GAAG,GAAG,GAAG,aAAa,CAAC;QAC3C,MAAM,iBAAiB,GAAG,cAAc,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAElE,yBAAyB;QACzB,MAAM,oBAAoB,GACxB,IAAI,CAAC,qBAAqB,GAAG,CAAC,CAAC,GAAG,iBAAiB,CAAC,CAAC;QACvD,MAAM,mBAAmB,GAAG,IAAI,CAAC,QAAQ,CAAC;QAE1C,OAAO,IAAI,CAAC,KAAK,CAAC,oBAAoB,GAAG,mBAAmB,CAAC,CAAC;IAChE,CAAC;IAED;;OAEG;IACH,wBAAwB;QACtB,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,yBAAyB;QACvB,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,sBAAsB,CAAC,OAAe,EAAE,MAAc,EAAE,MAAc,mBAAQ,CAAC,GAAG,EAAE;QAClF,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAC3C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,GAAG,YAAY,CAAC,CAAC;QAEtD,IAAI,SAAS,IAAI,MAAM,EAAE,CAAC;YACxB,OAAO,CAAC,CAAC,CAAC,oBAAoB;QAChC,CAAC;QAED,4EAA4E;QAC5E,4BAA4B;QAC5B,MAAM,aAAa,GAAG,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAEhE,mCAAmC;QACnC,OAAO,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;YACjD,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;SAC1C,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,QAAQ,CAAC,IAMf;QACC,OAAO,IAAI,aAAa,CACtB,IAAI,CAAC,EAAE,EACP,IAAI,CAAC,iBAAiB,EACtB,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,qBAAqB,EAC1B,IAAI,CAAC,WAAW,CACjB,CAAC;IACJ,CAAC;CACF;AA1KD,sCA0KC"}

package/dist/policy/SlidingWindowLimiter.d.ts

@@ -0,0 +1,41 @@
+import type { LimiterInterface } from '../LimiterInterface';
+import type { StorageInterface } from '../storage/StorageInterface';
+import type { LockInterface } from '../storage/LockInterface';
+import { RateLimit } from '../RateLimit';
+import { Reservation } from '../Reservation';
+/**
+ * Sliding window rate limiter implementation.
+ *
+ * Uses a weighted combination of the previous and current windows to smooth
+ * out burst behavior at window boundaries. More sophisticated than fixed window
+ * but slightly more expensive computationally.
+ */
+export declare class SlidingWindowLimiter implements LimiterInterface {
+    private readonly id;
+    private readonly limit;
+    private readonly intervalInSeconds;
+    private readonly storage;
+    private readonly lock;
+    constructor(id: string, limit: number, intervalInSeconds: number, storage: StorageInterface, lock?: LockInterface);
+    /**
+     * Reserve tokens with optional maximum wait time.
+     */
+    reserve(tokens?: number, maxTime?: number | null): Promise<Reservation>;
+    /**
+     * Try to consume tokens immediately.
+     */
+    consume(tokens?: number): Promise<RateLimit>;
+    /**
+     * Reset the rate limiter state.
+     */
+    reset(): Promise<void>;
+    /**
+     * Get existing window or create a new one.
+     */
+    private getOrCreateWindow;
+    /**
+     * Transition to a new window if the current one has ended.
+     */
+    private maybeTransitionWindow;
+}
+//# sourceMappingURL=SlidingWindowLimiter.d.ts.map

package/dist/policy/SlidingWindowLimiter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"SlidingWindowLimiter.d.ts","sourceRoot":"","sources":["../../src/policy/SlidingWindowLimiter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAM7C;;;;;;GAMG;AACH,qBAAa,oBAAqB,YAAW,gBAAgB;IAC3D,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAS;IAC5B,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAS;IAC3C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmB;IAC3C,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAgB;gBAGnC,EAAE,EAAE,MAAM,EACV,KAAK,EAAE,MAAM,EACb,iBAAiB,EAAE,MAAM,EACzB,OAAO,EAAE,gBAAgB,EACzB,IAAI,CAAC,EAAE,aAAa;IAStB;;OAEG;IACG,OAAO,CAAC,MAAM,GAAE,MAAU,EAAE,OAAO,GAAE,MAAM,GAAG,IAAW,GAAG,OAAO,CAAC,WAAW,CAAC;IAuDtF;;OAEG;IACG,OAAO,CAAC,MAAM,GAAE,MAAU,GAAG,OAAO,CAAC,SAAS,CAAC;IAgCrD;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAM5B;;OAEG;YACW,iBAAiB;IAW/B;;OAEG;IACH,OAAO,CAAC,qBAAqB;CAsB9B"}

package/dist/policy/SlidingWindowLimiter.js

@@ -0,0 +1,127 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SlidingWindowLimiter = void 0;
+const RateLimit_1 = require("../RateLimit");
+const Reservation_1 = require("../Reservation");
+const MaxWaitDurationExceededError_1 = require("../errors/MaxWaitDurationExceededError");
+const SlidingWindow_1 = require("./SlidingWindow");
+const TimeUtil_1 = require("../util/TimeUtil");
+const LockInterface_1 = require("../storage/LockInterface");
+/**
+ * Sliding window rate limiter implementation.
+ *
+ * Uses a weighted combination of the previous and current windows to smooth
+ * out burst behavior at window boundaries. More sophisticated than fixed window
+ * but slightly more expensive computationally.
+ */
+class SlidingWindowLimiter {
+    constructor(id, limit, intervalInSeconds, storage, lock) {
+        this.id = id;
+        this.limit = limit;
+        this.intervalInSeconds = intervalInSeconds;
+        this.storage = storage;
+        this.lock = lock ?? new LockInterface_1.NoLock();
+    }
+    /**
+     * Reserve tokens with optional maximum wait time.
+     */
+    async reserve(tokens = 1, maxTime = null) {
+        return this.lock.withLock(this.id, async () => {
+            const now = TimeUtil_1.TimeUtil.now();
+            const window = await this.getOrCreateWindow(now);
+            // Transition to new window if needed
+            const updatedWindow = this.maybeTransitionWindow(window, now);
+            const currentHitCount = updatedWindow.getHitCount(now);
+            const availableTokens = Math.max(0, this.limit - currentHitCount);
+            if (tokens > this.limit) {
+                throw new Error(`Cannot reserve ${tokens} tokens, limit is ${this.limit}`);
+            }
+            let timeToAct;
+            let waitTime;
+            if (availableTokens >= tokens) {
+                // Tokens available now
+                timeToAct = now;
+                waitTime = 0;
+                updatedWindow.add(tokens);
+            }
+            else {
+                // Need to wait for window to slide
+                waitTime = updatedWindow.calculateTimeForTokens(this.limit, tokens, now);
+                timeToAct = now + waitTime;
+            }
+            // Check max wait time
+            if (maxTime !== null && waitTime > maxTime) {
+                const retryAfter = new Date((now + waitTime) * 1000);
+                const rateLimit = new RateLimit_1.RateLimit(availableTokens, retryAfter, false, this.limit);
+                throw new MaxWaitDurationExceededError_1.MaxWaitDurationExceededError(`Cannot reserve ${tokens} tokens within ${maxTime} seconds`, rateLimit);
+            }
+            await this.storage.save(updatedWindow);
+            const retryAfter = new Date(timeToAct * 1000);
+            const rateLimit = new RateLimit_1.RateLimit(Math.max(0, availableTokens - tokens), retryAfter, true, this.limit);
+            return new Reservation_1.Reservation(timeToAct * 1000, rateLimit); // Convert to ms
+        });
+    }
+    /**
+     * Try to consume tokens immediately.
+     */
+    async consume(tokens = 1) {
+        return this.lock.withLock(this.id, async () => {
+            const now = TimeUtil_1.TimeUtil.now();
+            const window = await this.getOrCreateWindow(now);
+            // Transition to new window if needed
+            const updatedWindow = this.maybeTransitionWindow(window, now);
+            const currentHitCount = updatedWindow.getHitCount(now);
+            const availableTokens = Math.max(0, this.limit - currentHitCount);
+            if (availableTokens >= tokens) {
+                // Success
+                updatedWindow.add(tokens);
+                await this.storage.save(updatedWindow);
+                return new RateLimit_1.RateLimit(availableTokens - tokens, new Date(now * 1000), true, this.limit);
+            }
+            else {
+                // Rate limit exceeded
+                const waitTime = updatedWindow.calculateTimeForTokens(this.limit, tokens, now);
+                const retryAfter = new Date((now + waitTime) * 1000);
+                return new RateLimit_1.RateLimit(availableTokens, retryAfter, false, this.limit);
+            }
+        });
+    }
+    /**
+     * Reset the rate limiter state.
+     */
+    async reset() {
+        await this.lock.withLock(this.id, async () => {
+            await this.storage.delete(this.id);
+        });
+    }
+    /**
+     * Get existing window or create a new one.
+     */
+    async getOrCreateWindow(now) {
+        const state = await this.storage.fetch(this.id);
+        if (state instanceof SlidingWindow_1.SlidingWindow) {
+            return state;
+        }
+        // Create new window
+        return new SlidingWindow_1.SlidingWindow(this.id, this.intervalInSeconds);
+    }
+    /**
+     * Transition to a new window if the current one has ended.
+     */
+    maybeTransitionWindow(window, now) {
+        const windowEndAt = window.getWindowEndAt();
+        if (now < windowEndAt) {
+            // Still in current window
+            return window;
+        }
+        // Check if we're completely past the window (expired)
+        if (window.isExpired(now)) {
+            // Create fresh window
+            return new SlidingWindow_1.SlidingWindow(this.id, this.intervalInSeconds);
+        }
+        // Transition to new window, carrying over current hits as "last window"
+        return SlidingWindow_1.SlidingWindow.createFromPreviousWindow(this.id, this.intervalInSeconds, window, now);
+    }
+}
+exports.SlidingWindowLimiter = SlidingWindowLimiter;
+//# sourceMappingURL=SlidingWindowLimiter.js.map

package/dist/policy/SlidingWindowLimiter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SlidingWindowLimiter.js","sourceRoot":"","sources":["../../src/policy/SlidingWindowLimiter.ts"],"names":[],"mappings":";;;AAGA,4CAAyC;AACzC,gDAA6C;AAC7C,yFAAsF;AACtF,mDAAgD;AAChD,+CAA4C;AAC5C,4DAAkD;AAElD;;;;;;GAMG;AACH,MAAa,oBAAoB;IAO/B,YACE,EAAU,EACV,KAAa,EACb,iBAAyB,EACzB,OAAyB,EACzB,IAAoB;QAEpB,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,IAAI,sBAAM,EAAE,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,SAAiB,CAAC,EAAE,UAAyB,IAAI;QAC7D,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,GAAG,GAAG,mBAAQ,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAEjD,qCAAqC;YACrC,MAAM,aAAa,GAAG,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAE9D,MAAM,eAAe,GAAG,aAAa,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YACvD,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,GAAG,eAAe,CAAC,CAAC;YAElE,IAAI,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CACb,kBAAkB,MAAM,qBAAqB,IAAI,CAAC,KAAK,EAAE,CAC1D,CAAC;YACJ,CAAC;YAED,IAAI,SAAiB,CAAC;YACtB,IAAI,QAAgB,CAAC;YAErB,IAAI,eAAe,IAAI,MAAM,EAAE,CAAC;gBAC9B,uBAAuB;gBACvB,SAAS,GAAG,GAAG,CAAC;gBAChB,QAAQ,GAAG,CAAC,CAAC;gBACb,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC;iBAAM,CAAC;gBACN,mCAAmC;gBACnC,QAAQ,GAAG,aAAa,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;gBACzE,SAAS,GAAG,GAAG,GAAG,QAAQ,CAAC;YAC7B,CAAC;YAED,sBAAsB;YACtB,IAAI,OAAO,KAAK,IAAI,IAAI,QAAQ,GAAG,OAAO,EAAE,CAAC;gBAC3C,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,CAAC,GAAG,GAAG,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;gBACrD,MAAM,SAAS,GAAG,IAAI,qBAAS,CAAC,eAAe,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;gBAChF,MAAM,IAAI,2DAA4B,CACpC,kBAAkB,MAAM,kBAAkB,OAAO,UAAU,EAC3D,SAAS,CACV,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAEvC,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC;YAC9C,MAAM,SAAS,GAAG,IAAI,qBAAS,CAC7B,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,eAAe,GAAG,MAAM,CAAC,EACrC,UAAU,EACV,IAAI,EACJ,IAAI,CAAC,KAAK,CACX,CAAC;YAEF,OAAO,IAAI,yBAAW,CAAC,SAAS,GAAG,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,gBAAgB;QACvE,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,SAAiB,CAAC;QAC9B,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,GAAG,GAAG,mBAAQ,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAEjD,qCAAqC;YACrC,MAAM,aAAa,GAAG,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAE9D,MAAM,eAAe,GAAG,aAAa,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YACvD,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,GAAG,eAAe,CAAC,CAAC;YAElE,IAAI,eAAe,IAAI,MAAM,EAAE,CAAC;gBAC9B,UAAU;gBACV,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC1B,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBAEvC,OAAO,IAAI,qBAAS,CAClB,eAAe,GAAG,MAAM,EACxB,IAAI,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,EACpB,IAAI,EACJ,IAAI,CAAC,KAAK,CACX,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,sBAAsB;gBACtB,MAAM,QAAQ,GAAG,aAAa,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;gBAC/E,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,CAAC,GAAG,GAAG,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;gBAErD,OAAO,IAAI,qBAAS,CAAC,eAAe,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;YACvE,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB,CAAC,GAAW;QACzC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEhD,IAAI,KAAK,YAAY,6BAAa,EAAE,CAAC;YACnC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,oBAAoB;QACpB,OAAO,IAAI,6BAAa,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC5D,CAAC;IAED;;OAEG;IACK,qBAAqB,CAAC,MAAqB,EAAE,GAAW;QAC9D,MAAM,WAAW,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;QAE5C,IAAI,GAAG,GAAG,WAAW,EAAE,CAAC;YACtB,0BAA0B;YAC1B,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,sDAAsD;QACtD,IAAI,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1B,sBAAsB;YACtB,OAAO,IAAI,6BAAa,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC5D,CAAC;QAED,wEAAwE;QACxE,OAAO,6BAAa,CAAC,wBAAwB,CAC3C,IAAI,CAAC,EAAE,EACP,IAAI,CAAC,iBAAiB,EACtB,MAAM,EACN,GAAG,CACJ,CAAC;IACJ,CAAC;CACF;AAlKD,oDAkKC"}

package/dist/policy/TokenBucket.d.ts

@@ -0,0 +1,63 @@
+import type { LimiterStateInterface } from '../LimiterStateInterface';
+import { Rate } from './Rate';
+/**
+ * Serializable state for TokenBucketLimiter.
+ *
+ * Tracks the number of available tokens and when they were last updated.
+ */
+export declare class TokenBucket implements LimiterStateInterface {
+    private id;
+    private tokens;
+    private readonly burstSize;
+    private timer;
+    private readonly rate;
+    constructor(id: string, burstSize: number, rate: Rate, tokens?: number, timer?: number);
+    getId(): string;
+    /**
+     * Get the expiration time in seconds since epoch.
+     * Token bucket expires after enough time has passed to refill completely.
+     */
+    getExpirationTime(): number;
+    /**
+     * Get the available tokens at a given time, accounting for refills.
+     */
+    getAvailableTokens(now?: number): number;
+    /**
+     * Set the number of tokens.
+     */
+    setTokens(tokens: number): void;
+    /**
+     * Get the timer (last update time).
+     */
+    getTimer(): number;
+    /**
+     * Set the timer (last update time).
+     */
+    setTimer(timer: number): void;
+    /**
+     * Get the burst size (maximum tokens).
+     */
+    getBurstSize(): number;
+    /**
+     * Get the refill rate.
+     */
+    getRate(): Rate;
+    /**
+     * Convert to JSON for serialization.
+     */
+    toJSON(): Record<string, unknown>;
+    /**
+     * Create from JSON.
+     */
+    static fromJSON(data: {
+        id: string;
+        tokens: number;
+        burstSize: number;
+        timer: number;
+        rate: {
+            interval: number;
+            amount: number;
+        };
+    }): TokenBucket;
+}
+//# sourceMappingURL=TokenBucket.d.ts.map

package/dist/policy/TokenBucket.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TokenBucket.d.ts","sourceRoot":"","sources":["../../src/policy/TokenBucket.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAG9B;;;;GAIG;AACH,qBAAa,WAAY,YAAW,qBAAqB;IACvD,OAAO,CAAC,EAAE,CAAS;IACnB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAO;gBAG1B,EAAE,EAAE,MAAM,EACV,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,IAAI,EACV,MAAM,CAAC,EAAE,MAAM,EACf,KAAK,CAAC,EAAE,MAAM;IAShB,KAAK,IAAI,MAAM;IAIf;;;OAGG;IACH,iBAAiB,IAAI,MAAM;IAK3B;;OAEG;IACH,kBAAkB,CAAC,GAAG,GAAE,MAAuB,GAAG,MAAM;IAMxD;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAI/B;;OAEG;IACH,QAAQ,IAAI,MAAM;IAIlB;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAI7B;;OAEG;IACH,YAAY,IAAI,MAAM;IAItB;;OAEG;IACH,OAAO,IAAI,IAAI;IAIf;;OAEG;IACH,MAAM,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAajC;;OAEG;IACH,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE;QACpB,EAAE,EAAE,MAAM,CAAC;QACX,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE;YAAE,QAAQ,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,CAAC;KAC5C,GAAG,WAAW;CAUhB"}

package/dist/policy/TokenBucket.js

@@ -0,0 +1,92 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TokenBucket = void 0;
+const Rate_1 = require("./Rate");
+const TimeUtil_1 = require("../util/TimeUtil");
+/**
+ * Serializable state for TokenBucketLimiter.
+ *
+ * Tracks the number of available tokens and when they were last updated.
+ */
+class TokenBucket {
+    constructor(id, burstSize, rate, tokens, timer) {
+        this.id = id;
+        this.burstSize = burstSize;
+        this.rate = rate;
+        this.tokens = tokens ?? burstSize;
+        this.timer = timer ?? TimeUtil_1.TimeUtil.now();
+    }
+    getId() {
+        return this.id;
+    }
+    /**
+     * Get the expiration time in seconds since epoch.
+     * Token bucket expires after enough time has passed to refill completely.
+     */
+    getExpirationTime() {
+        const refillTime = this.rate.calculateTimeForTokens(this.burstSize);
+        return Math.ceil(this.timer + refillTime);
+    }
+    /**
+     * Get the available tokens at a given time, accounting for refills.
+     */
+    getAvailableTokens(now = TimeUtil_1.TimeUtil.now()) {
+        const elapsed = now - this.timer;
+        const newTokens = this.rate.calculateNewTokensDuringInterval(elapsed);
+        return Math.min(this.burstSize, this.tokens + newTokens);
+    }
+    /**
+     * Set the number of tokens.
+     */
+    setTokens(tokens) {
+        this.tokens = tokens;
+    }
+    /**
+     * Get the timer (last update time).
+     */
+    getTimer() {
+        return this.timer;
+    }
+    /**
+     * Set the timer (last update time).
+     */
+    setTimer(timer) {
+        this.timer = timer;
+    }
+    /**
+     * Get the burst size (maximum tokens).
+     */
+    getBurstSize() {
+        return this.burstSize;
+    }
+    /**
+     * Get the refill rate.
+     */
+    getRate() {
+        return this.rate;
+    }
+    /**
+     * Convert to JSON for serialization.
+     */
+    toJSON() {
+        return {
+            id: this.id,
+            tokens: this.tokens,
+            burstSize: this.burstSize,
+            timer: this.timer,
+            rate: {
+                interval: this.rate.getInterval(),
+                amount: this.rate.getAmount(),
+            },
+        };
+    }
+    /**
+     * Create from JSON.
+     */
+    static fromJSON(data) {
+        const rate = new Rate_1.Rate(data.rate.interval, data.rate.amount);
+        return new TokenBucket(data.id, data.burstSize, rate, data.tokens, data.timer);
+    }
+}
+exports.TokenBucket = TokenBucket;
+//# sourceMappingURL=TokenBucket.js.map

package/dist/policy/TokenBucket.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TokenBucket.js","sourceRoot":"","sources":["../../src/policy/TokenBucket.ts"],"names":[],"mappings":";;;AACA,iCAA8B;AAC9B,+CAA4C;AAE5C;;;;GAIG;AACH,MAAa,WAAW;IAOtB,YACE,EAAU,EACV,SAAiB,EACjB,IAAU,EACV,MAAe,EACf,KAAc;QAEd,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,SAAS,CAAC;QAClC,IAAI,CAAC,KAAK,GAAG,KAAK,IAAI,mBAAQ,CAAC,GAAG,EAAE,CAAC;IACvC,CAAC;IAED,KAAK;QACH,OAAO,IAAI,CAAC,EAAE,CAAC;IACjB,CAAC;IAED;;;OAGG;IACH,iBAAiB;QACf,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACpE,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,MAAc,mBAAQ,CAAC,GAAG,EAAE;QAC7C,MAAM,OAAO,GAAG,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC;QACjC,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,gCAAgC,CAAC,OAAO,CAAC,CAAC;QACtE,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;IAC3D,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,MAAc;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,KAAa;QACpB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE;gBACJ,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;gBACjC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;aAC9B;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,QAAQ,CAAC,IAMf;QACC,MAAM,IAAI,GAAG,IAAI,WAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5D,OAAO,IAAI,WAAW,CACpB,IAAI,CAAC,EAAE,EACP,IAAI,CAAC,SAAS,EACd,IAAI,EACJ,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,KAAK,CACX,CAAC;IACJ,CAAC;CACF;AAjHD,kCAiHC"}

package/dist/policy/TokenBucketLimiter.d.ts

@@ -0,0 +1,38 @@
+import type { LimiterInterface } from '../LimiterInterface';
+import type { StorageInterface } from '../storage/StorageInterface';
+import type { LockInterface } from '../storage/LockInterface';
+import { RateLimit } from '../RateLimit';
+import { Reservation } from '../Reservation';
+import { Rate } from './Rate';
+/**
+ * Token bucket rate limiter implementation.
+ *
+ * The token bucket algorithm allows bursts up to the bucket size while
+ * maintaining a steady refill rate. This is the most flexible algorithm
+ * and supports both reserve() and consume() patterns.
+ */
+export declare class TokenBucketLimiter implements LimiterInterface {
+    private readonly id;
+    private readonly burstSize;
+    private readonly rate;
+    private readonly storage;
+    private readonly lock;
+    constructor(id: string, burstSize: number, rate: Rate, storage: StorageInterface, lock?: LockInterface);
+    /**
+     * Reserve tokens with optional maximum wait time.
+     */
+    reserve(tokens?: number, maxTime?: number | null): Promise<Reservation>;
+    /**
+     * Try to consume tokens immediately.
+     */
+    consume(tokens?: number): Promise<RateLimit>;
+    /**
+     * Reset the rate limiter state.
+     */
+    reset(): Promise<void>;
+    /**
+     * Get existing bucket or create a new one.
+     */
+    private getOrCreateBucket;
+}
+//# sourceMappingURL=TokenBucketLimiter.d.ts.map

package/dist/policy/TokenBucketLimiter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TokenBucketLimiter.d.ts","sourceRoot":"","sources":["../../src/policy/TokenBucketLimiter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAG7C,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAI9B;;;;;;GAMG;AACH,qBAAa,kBAAmB,YAAW,gBAAgB;IACzD,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAS;IAC5B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAO;IAC5B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmB;IAC3C,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAgB;gBAGnC,EAAE,EAAE,MAAM,EACV,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,IAAI,EACV,OAAO,EAAE,gBAAgB,EACzB,IAAI,CAAC,EAAE,aAAa;IAStB;;OAEG;IACG,OAAO,CAAC,MAAM,GAAE,MAAU,EAAE,OAAO,GAAE,MAAM,GAAG,IAAW,GAAG,OAAO,CAAC,WAAW,CAAC;IA0DtF;;OAEG;IACG,OAAO,CAAC,MAAM,GAAE,MAAU,GAAG,OAAO,CAAC,SAAS,CAAC;IAiCrD;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAM5B;;OAEG;YACW,iBAAiB;CAUhC"}

package/dist/policy/TokenBucketLimiter.js

@@ -0,0 +1,114 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TokenBucketLimiter = void 0;
+const RateLimit_1 = require("../RateLimit");
+const Reservation_1 = require("../Reservation");
+const MaxWaitDurationExceededError_1 = require("../errors/MaxWaitDurationExceededError");
+const TokenBucket_1 = require("./TokenBucket");
+const TimeUtil_1 = require("../util/TimeUtil");
+const LockInterface_1 = require("../storage/LockInterface");
+/**
+ * Token bucket rate limiter implementation.
+ *
+ * The token bucket algorithm allows bursts up to the bucket size while
+ * maintaining a steady refill rate. This is the most flexible algorithm
+ * and supports both reserve() and consume() patterns.
+ */
+class TokenBucketLimiter {
+    constructor(id, burstSize, rate, storage, lock) {
+        this.id = id;
+        this.burstSize = burstSize;
+        this.rate = rate;
+        this.storage = storage;
+        this.lock = lock ?? new LockInterface_1.NoLock();
+    }
+    /**
+     * Reserve tokens with optional maximum wait time.
+     */
+    async reserve(tokens = 1, maxTime = null) {
+        return this.lock.withLock(this.id, async () => {
+            const now = TimeUtil_1.TimeUtil.now();
+            const bucket = await this.getOrCreateBucket(now);
+            const availableTokens = bucket.getAvailableTokens(now);
+            // Update bucket state to account for elapsed time
+            bucket.setTokens(availableTokens);
+            bucket.setTimer(now);
+            if (tokens > this.burstSize) {
+                throw new Error(`Cannot reserve ${tokens} tokens, burst size is ${this.burstSize}`);
+            }
+            let timeToAct;
+            let waitTime;
+            if (availableTokens >= tokens) {
+                // Tokens available now
+                timeToAct = now;
+                waitTime = 0;
+                bucket.setTokens(availableTokens - tokens);
+            }
+            else {
+                // Need to wait for tokens
+                const tokensNeeded = tokens - availableTokens;
+                waitTime = this.rate.calculateTimeForTokens(tokensNeeded);
+                timeToAct = now + waitTime;
+                bucket.setTokens(0); // All current tokens consumed
+                bucket.setTimer(timeToAct); // Next refill starts after timeToAct
+            }
+            // Check max wait time
+            if (maxTime !== null && waitTime > maxTime) {
+                const retryAfter = new Date((now + waitTime) * 1000);
+                const rateLimit = new RateLimit_1.RateLimit(availableTokens, retryAfter, false, this.burstSize);
+                throw new MaxWaitDurationExceededError_1.MaxWaitDurationExceededError(`Cannot reserve ${tokens} tokens within ${maxTime} seconds`, rateLimit);
+            }
+            await this.storage.save(bucket);
+            const retryAfter = new Date(timeToAct * 1000);
+            const rateLimit = new RateLimit_1.RateLimit(Math.max(0, availableTokens - tokens), retryAfter, true, this.burstSize);
+            return new Reservation_1.Reservation(timeToAct * 1000, rateLimit); // Convert to ms
+        });
+    }
+    /**
+     * Try to consume tokens immediately.
+     */
+    async consume(tokens = 1) {
+        return this.lock.withLock(this.id, async () => {
+            const now = TimeUtil_1.TimeUtil.now();
+            const bucket = await this.getOrCreateBucket(now);
+            const availableTokens = bucket.getAvailableTokens(now);
+            // Update bucket state
+            bucket.setTokens(availableTokens);
+            bucket.setTimer(now);
+            if (availableTokens >= tokens) {
+                // Success
+                bucket.setTokens(availableTokens - tokens);
+                await this.storage.save(bucket);
+                return new RateLimit_1.RateLimit(availableTokens - tokens, new Date(now * 1000), true, this.burstSize);
+            }
+            else {
+                // Rate limit exceeded
+                const tokensNeeded = tokens - availableTokens;
+                const waitTime = this.rate.calculateTimeForTokens(tokensNeeded);
+                const retryAfter = new Date((now + waitTime) * 1000);
+                return new RateLimit_1.RateLimit(availableTokens, retryAfter, false, this.burstSize);
+            }
+        });
+    }
+    /**
+     * Reset the rate limiter state.
+     */
+    async reset() {
+        await this.lock.withLock(this.id, async () => {
+            await this.storage.delete(this.id);
+        });
+    }
+    /**
+     * Get existing bucket or create a new one.
+     */
+    async getOrCreateBucket(now) {
+        const state = await this.storage.fetch(this.id);
+        if (state instanceof TokenBucket_1.TokenBucket) {
+            return state;
+        }
+        // Create new bucket
+        return new TokenBucket_1.TokenBucket(this.id, this.burstSize, this.rate, undefined, now);
+    }
+}
+exports.TokenBucketLimiter = TokenBucketLimiter;
+//# sourceMappingURL=TokenBucketLimiter.js.map

package/dist/policy/TokenBucketLimiter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TokenBucketLimiter.js","sourceRoot":"","sources":["../../src/policy/TokenBucketLimiter.ts"],"names":[],"mappings":";;;AAGA,4CAAyC;AACzC,gDAA6C;AAC7C,yFAAsF;AACtF,+CAA4C;AAE5C,+CAA4C;AAC5C,4DAAkD;AAElD;;;;;;GAMG;AACH,MAAa,kBAAkB;IAO7B,YACE,EAAU,EACV,SAAiB,EACjB,IAAU,EACV,OAAyB,EACzB,IAAoB;QAEpB,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,IAAI,sBAAM,EAAE,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,SAAiB,CAAC,EAAE,UAAyB,IAAI;QAC7D,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,GAAG,GAAG,mBAAQ,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAEjD,MAAM,eAAe,GAAG,MAAM,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC;YAEvD,kDAAkD;YAClD,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YAClC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAErB,IAAI,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC5B,MAAM,IAAI,KAAK,CACb,kBAAkB,MAAM,0BAA0B,IAAI,CAAC,SAAS,EAAE,CACnE,CAAC;YACJ,CAAC;YAED,IAAI,SAAiB,CAAC;YACtB,IAAI,QAAgB,CAAC;YAErB,IAAI,eAAe,IAAI,MAAM,EAAE,CAAC;gBAC9B,uBAAuB;gBACvB,SAAS,GAAG,GAAG,CAAC;gBAChB,QAAQ,GAAG,CAAC,CAAC;gBACb,MAAM,CAAC,SAAS,CAAC,eAAe,GAAG,MAAM,CAAC,CAAC;YAC7C,CAAC;iBAAM,CAAC;gBACN,0BAA0B;gBAC1B,MAAM,YAAY,GAAG,MAAM,GAAG,eAAe,CAAC;gBAC9C,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,YAAY,CAAC,CAAC;gBAC1D,SAAS,GAAG,GAAG,GAAG,QAAQ,CAAC;gBAC3B,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,8BAA8B;gBACnD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,qCAAqC;YACnE,CAAC;YAED,sBAAsB;YACtB,IAAI,OAAO,KAAK,IAAI,IAAI,QAAQ,GAAG,OAAO,EAAE,CAAC;gBAC3C,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,CAAC,GAAG,GAAG,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;gBACrD,MAAM,SAAS,GAAG,IAAI,qBAAS,CAAC,eAAe,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;gBACpF,MAAM,IAAI,2DAA4B,CACpC,kBAAkB,MAAM,kBAAkB,OAAO,UAAU,EAC3D,SAAS,CACV,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAEhC,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC;YAC9C,MAAM,SAAS,GAAG,IAAI,qBAAS,CAC7B,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,eAAe,GAAG,MAAM,CAAC,EACrC,UAAU,EACV,IAAI,EACJ,IAAI,CAAC,SAAS,CACf,CAAC;YAEF,OAAO,IAAI,yBAAW,CAAC,SAAS,GAAG,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,gBAAgB;QACvE,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,SAAiB,CAAC;QAC9B,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,GAAG,GAAG,mBAAQ,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAEjD,MAAM,eAAe,GAAG,MAAM,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC;YAEvD,sBAAsB;YACtB,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YAClC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAErB,IAAI,eAAe,IAAI,MAAM,EAAE,CAAC;gBAC9B,UAAU;gBACV,MAAM,CAAC,SAAS,CAAC,eAAe,GAAG,MAAM,CAAC,CAAC;gBAC3C,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAEhC,OAAO,IAAI,qBAAS,CAClB,eAAe,GAAG,MAAM,EACxB,IAAI,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,EACpB,IAAI,EACJ,IAAI,CAAC,SAAS,CACf,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,sBAAsB;gBACtB,MAAM,YAAY,GAAG,MAAM,GAAG,eAAe,CAAC;gBAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,YAAY,CAAC,CAAC;gBAChE,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,CAAC,GAAG,GAAG,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;gBAErD,OAAO,IAAI,qBAAS,CAAC,eAAe,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YAC3E,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB,CAAC,GAAY;QAC1C,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEhD,IAAI,KAAK,YAAY,yBAAW,EAAE,CAAC;YACjC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,oBAAoB;QACpB,OAAO,IAAI,yBAAW,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,GAAG,CAAC,CAAC;IAC7E,CAAC;CACF;AA5ID,gDA4IC"}