@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.
167 lines • 6.27 kB
JavaScript
/**
* @module RateLimiter
*/
import {} from "../../../../backoff-policies/contracts/_module.js";
import {} from "../../../../event-bus/contracts/_module.js";
import {} from "../../../../execution-context/contracts/_module.js";
import {} from "../../../../namespace/contracts/_module.js";
import {} from "../../../../rate-limiter/contracts/_module.js";
import { DatabaseRateLimiterAdapter } from "../../../../rate-limiter/implementations/adapters/database-rate-limiter-adapter/_module.js";
import { RateLimiterFactory, } from "../../../../rate-limiter/implementations/derivables/rate-limiter-factory/_module.js";
import {
// eslint-disable-next-line @typescript-eslint/no-unused-vars
UnregisteredAdapterError,
// eslint-disable-next-line @typescript-eslint/no-unused-vars
DefaultAdapterNotDefinedError, } from "../../../../utilities/_module.js";
/**
* The `DatabaseRateLimiterFactoryResolver` class is immutable.
*
* IMPORT_PATH: `"@daiso-tech/core/rate-limiter"`
* @group Derivables
*/
export class DatabaseRateLimiterFactoryResolver {
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 { KyselyRateLimiterStorageAdapter } from "@daiso-tech/core/rate-limiter/kysely-rate-limiter-storate-adapter";
* import { DatabaseRateLimiterAdapter } from "@daiso-tech/core/rate-limiter/database-rate-limiter-adapter";
* import { Serde } from "@daiso-tech/core/serde";
* import { SuperJsonSerdeAdapter } from "@daiso-tech/core/serde/super-json-serde-adapter";
* import Sqlite from "better-sqlite3";
* import { Kysely, SqliteDialect } from "kysely";
*
* const serde = new Serde(new SuperJsonSerdeAdapter());
* const rateLimiterFactoryResolver = new RateLimiterFactoryResolver({
* serde,
* adapters: {
* memory: new MemoryRateLimiterStorageAdapter(),
* sqlite: new KyselyRateLimiterStorageAdapter({
* kysely: new Kysely({
* dialect: new SqliteDialect({
* database: new Sqlite("local.db"),
* }),
* }),
* serde,
* }),
* },
* defaultAdapter: "memory",
* });
* ```
*/
constructor(settings) {
this.settings = settings;
}
setNamespace(namespace) {
return new DatabaseRateLimiterFactoryResolver({
...this.settings,
namespace,
});
}
setEventBus(eventBus) {
return new DatabaseRateLimiterFactoryResolver({
...this.settings,
eventBus,
});
}
setOnlyError(onlyError) {
return new DatabaseRateLimiterFactoryResolver({
...this.settings,
onlyError,
});
}
setDefaultErrorPolicy(errorPolicy) {
return new DatabaseRateLimiterFactoryResolver({
...this.settings,
defaultErrorPolicy: errorPolicy,
});
}
setBackoffPolicy(backoffPolicy) {
return new DatabaseRateLimiterFactoryResolver({
...this.settings,
backoffPolicy,
});
}
setRateLimiterPolicy(rateLimiterPolicy) {
return new DatabaseRateLimiterFactoryResolver({
...this.settings,
rateLimiterPolicy,
});
}
setWaitUntil(waitUntil) {
return new DatabaseRateLimiterFactoryResolver({
...this.settings,
waitUntil,
});
}
setExecutionContext(executionContext) {
return new DatabaseRateLimiterFactoryResolver({
...this.settings,
executionContext,
});
}
/**
* @example
* ```ts
* import { RateLimiterFactoryResolver } from "@daiso-tech/core/rate-limiter";
* import { MemoryRateLimiterStorageAdapter } from "@daiso-tech/core/rate-limiter/memory-rate-limiter-storate-adapter";
* import { KyselyRateLimiterStorageAdapter } from "@daiso-tech/core/rate-limiter/kysely-rate-limiter-storate-adapter";
* import { DatabaseRateLimiterAdapter } from "@daiso-tech/core/rate-limiter/database-rate-limiter-adapter";
* import { Serde } from "@daiso-tech/core/serde";
* import { SuperJsonSerdeAdapter } from "@daiso-tech/core/serde/super-json-serde-adapter";
* import Sqlite from "better-sqlite3";
* import { Kysely, SqliteDialect } from "kysely";
*
* const serde = new Serde(new SuperJsonSerdeAdapter());
* const rateLimiterFactoryResolver = new RateLimiterFactoryResolver({
* serde,
* adapters: {
* memory: new MemoryRateLimiterStorageAdapter(),
* sqlite: new KyselyRateLimiterStorageAdapter({
* kysely: new Kysely({
* dialect: new SqliteDialect({
* database: new Sqlite("local.db"),
* }),
* }),
* serde,
* }),
* },
* 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 KyselyRateLimiterStorageAdapter
* await rateLimiterFactoryResolver
* .use("sqlite")
* .create("a")
* .runOrFail(async () => {
* // ... code to apply rate limiter logic
* });
* ```
*/
use(adapterName = this.settings.defaultAdapter) {
if (adapterName === undefined) {
throw new DefaultAdapterNotDefinedError(DatabaseRateLimiterFactoryResolver.name);
}
const adapter = this.settings.adapters[adapterName];
if (adapter === undefined) {
throw new UnregisteredAdapterError(adapterName);
}
return new RateLimiterFactory({
...this.settings,
adapter: new DatabaseRateLimiterAdapter({
adapter,
}),
});
}
}
//# sourceMappingURL=database-rate-limiter-factory-resolver.js.map