UNPKG

@daiso-tech/core

Version:

The library offers flexible, framework-agnostic solutions for modern web applications, built on adaptable components that integrate seamlessly with popular frameworks like Next Js.

117 lines (116 loc) 5.36 kB
/** * @module RateLimiter */ import { type EventBusInput } from "../../../../event-bus/contracts/_module.js"; import { type IExecutionContext } from "../../../../execution-context/contracts/_module.js"; import { type INamespace } from "../../../../namespace/contracts/_module.js"; import { type IRateLimiterFactoryResolver, type IRateLimiterFactory, type IRateLimiterAdapter } from "../../../../rate-limiter/contracts/_module.js"; import { type RateLimiterFactorySettingsBase } from "../../../../rate-limiter/implementations/derivables/rate-limiter-factory/_module.js"; import { type ErrorPolicy, type WaitUntil } from "../../../../utilities/_module.js"; /** * IMPORT_PATH: `"@daiso-tech/core/rate-limiter"` * @group Derivables */ export type RateLimiterAdapters<TAdapters extends string> = Partial<Record<TAdapters, IRateLimiterAdapter>>; /** * Configuration for `RateLimiterFactoryResolver`. * Registers named rate-limiter adapters and optionally designates a default. * * IMPORT_PATH: `"@daiso-tech/core/rate-limiter"` * @group Derivables */ export type RateLimiterFactoryResolverSettings<TAdapters extends string> = RateLimiterFactorySettingsBase & { /** * Named registry of rate-limiter adapters. Each key is an adapter alias and the corresponding value is the adapter instance. */ adapters: RateLimiterAdapters<TAdapters>; /** * The alias of the adapter to use when none is explicitly specified. Must be a key in the `adapters` map. */ defaultAdapter?: NoInfer<TAdapters>; }; /** * The `RateLimiterFactoryResolver` class is immutable. * * IMPORT_PATH: `"@daiso-tech/core/rate-limiter"` * @group Derivables */ export declare class RateLimiterFactoryResolver<TAdapters extends string> implements IRateLimiterFactoryResolver<TAdapters> { private readonly settings; /** * @example * ```ts * import { RateLimiterFactoryResolver } from "@daiso-tech/core/rate-limiter"; * import { MemoryRateLimiterStorageAdapter } from "@daiso-tech/core/rate-limiter/memory-rate-limiter-storate-adapter"; * import { DatabaseRateLimiterAdapter } from "@daiso-tech/core/rate-limiter/database-rate-limiter-adapter"; * import { RedisRateLimiterAdapter } from "@daiso-tech/core/rate-limiter/redis-rate-limiter-adapter"; * import { Serde } from "@daiso-tech/core/serde"; * import { SuperJsonSerdeAdapter } from "@daiso-tech/core/serde/super-json-serde-adapter"; * import Redis from "ioredis" * * const serde = new Serde(new SuperJsonSerdeAdapter()); * const rateLimiterFactoryResolver = new RateLimiterFactoryResolver({ * serde, * adapters: { * memory: new DatabaseRateLimiterAdapter({ * adapter: new MemoryRateLimiterStorageAdapter() * }), * redis: new RedisRateLimiterAdapter({ * database: new Redis("YOUR_REDIS_CONNECTION") * }), * }, * defaultAdapter: "memory", * }); * ``` */ constructor(settings: RateLimiterFactoryResolverSettings<TAdapters>); setNamespace(namespace: INamespace): RateLimiterFactoryResolver<TAdapters>; setEventBus(eventBus: EventBusInput): RateLimiterFactoryResolver<TAdapters>; setOnlyError(onlyError?: boolean): RateLimiterFactoryResolver<TAdapters>; setDefaultErrorPolicy(errorPolicy: ErrorPolicy): RateLimiterFactoryResolver<TAdapters>; setWaitUntil(waitUntil: WaitUntil): RateLimiterFactoryResolver<TAdapters>; setExecutionContext(executionContext: IExecutionContext): RateLimiterFactoryResolver<TAdapters>; /** * @example * ```ts * import { RateLimiterFactoryResolver } from "@daiso-tech/core/rate-limiter"; * import { MemoryRateLimiterStorageAdapter } from "@daiso-tech/core/rate-limiter/memory-rate-limiter-storate-adapter"; * import { DatabaseRateLimiterAdapter } from "@daiso-tech/core/rate-limiter/database-rate-limiter-adapter"; * import { RedisRateLimiterAdapter } from "@daiso-tech/core/rate-limiter/redis-rate-limiter-adapter"; * import { Serde } from "@daiso-tech/core/serde"; * import { SuperJsonSerdeAdapter } from "@daiso-tech/core/serde/super-json-serde-adapter"; * import Redis from "ioredis" * * const serde = new Serde(new SuperJsonSerdeAdapter()); * const rateLimiterFactoryResolver = new RateLimiterFactoryResolver({ * serde, * adapters: { * memory: new DatabaseRateLimiterAdapter({ * adapter: new MemoryRateLimiterStorageAdapter() * }), * redis: new RedisRateLimiterAdapter({ * database: new Redis("YOUR_REDIS_CONNECTION") * }), * }, * defaultAdapter: "memory", * }); * * // Will apply rate limiter logic the default adapter which is MemoryRateLimiterStorageAdapter * await rateLimiterFactoryResolver * .use() * .create("a") * .runOrFail(async () => { * // ... code to apply rate limiter logic * }); * * // Will apply rate limiter logic the default adapter which is RedisRateLimiterAdapter * await rateLimiterFactoryResolver * .use("redis") * .create("a") * .runOrFail(async () => { * // ... code to apply rate limiter logic * }); * ``` */ use(adapterName?: TAdapters | undefined): IRateLimiterFactory; }