veffect
Version:
powerful TypeScript validation library built on the robust foundation of Effect combining exceptional type safety, high performance, and developer experience. Taking inspiration from Effect's functional principles, VEffect delivers a balanced approach tha
42 lines • 2.12 kB
JavaScript
import * as Duration from "../Duration.js";
import * as Effect from "../Effect.js";
import * as FiberRef from "../FiberRef.js";
import { globalValue } from "../GlobalValue.js";
/** @internal */
export const make = ({
algorithm = "token-bucket",
interval,
limit
}) => {
switch (algorithm) {
case "fixed-window":
{
return fixedWindow(limit, interval);
}
case "token-bucket":
{
return tokenBucket(limit, interval);
}
}
};
const tokenBucket = (limit, window) => Effect.gen(function* (_) {
const millisPerToken = Math.ceil(Duration.toMillis(window) / limit);
const semaphore = yield* _(Effect.makeSemaphore(limit));
const latch = yield* _(Effect.makeSemaphore(0));
const refill = Effect.sleep(millisPerToken).pipe(Effect.zipRight(latch.releaseAll), Effect.zipRight(semaphore.release(1)), Effect.flatMap(free => free === limit ? Effect.unit : refill));
yield* _(latch.take(1), Effect.zipRight(refill), Effect.forever, Effect.forkScoped, Effect.interruptible);
const take = Effect.uninterruptibleMask(restore => Effect.flatMap(FiberRef.get(currentCost), cost => Effect.zipRight(restore(semaphore.take(cost)), latch.release(1))));
return effect => Effect.zipRight(take, effect);
});
const fixedWindow = (limit, window) => Effect.gen(function* (_) {
const semaphore = yield* _(Effect.makeSemaphore(limit));
const latch = yield* _(Effect.makeSemaphore(0));
yield* _(latch.take(1), Effect.zipRight(Effect.sleep(window)), Effect.zipRight(latch.releaseAll), Effect.zipRight(semaphore.releaseAll), Effect.forever, Effect.forkScoped, Effect.interruptible);
const take = Effect.uninterruptibleMask(restore => Effect.flatMap(FiberRef.get(currentCost), cost => Effect.zipRight(restore(semaphore.take(cost)), latch.release(1))));
return effect => Effect.zipRight(take, effect);
});
/** @internal */
const currentCost = /*#__PURE__*/globalValue( /*#__PURE__*/Symbol.for("effect/RateLimiter/currentCost"), () => FiberRef.unsafeMake(1));
/** @internal */
export const withCost = cost => Effect.locally(currentCost, cost);
//# sourceMappingURL=rateLimiter.js.map