-
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
10 changed files
with
4,434 additions
and
5,078 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,5 +1,6 @@ | ||
| export type { Store, StoreConfig, Stores } from 'cache-manager' | ||
| export type { CacheManagerStore } from 'cache-manager' | ||
| export type { Store } from './types' | ||
| export { redisStore } from './redis' | ||
| export * from './manager' | ||
| export { memoryStore } from './memory' | ||
| export type { MemoryStore, MemoryCache, MemoryConfig } from './memory' | ||
| export type { MemoryConfig } from './memory' |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,113 +1,13 @@ | ||
| import { LRUCache } from 'lru-cache' | ||
| import cloneDeep from 'lodash.clonedeep' | ||
| import type { Cache, Config, Store } from 'cache-manager' | ||
| import { Keyv } from 'keyv' | ||
| import type { CacheableMemoryOptions } from 'cacheable' | ||
| import { KeyvCacheableMemory } from 'cacheable' | ||
|
|
||
| function clone<T>(object: T): T { | ||
| if (typeof object === 'object' && object !== null) | ||
| return cloneDeep(object) | ||
|
|
||
| return object | ||
| export type MemoryConfig = CacheableMemoryOptions & { | ||
| namespace?: string | ||
| } | ||
|
|
||
| type LRU = LRUCache<string, any> | ||
|
|
||
| type Pre = LRUCache.OptionsTTLLimit<string, any, unknown> | ||
| type Options = Omit<Pre, 'ttlAutopurge'> & Partial<Pick<Pre, 'ttlAutopurge'>> | ||
| export type MemoryConfig = { | ||
| max?: number | ||
| sizeCalculation?: (value: unknown, key: string) => number | ||
| shouldCloneBeforeSet?: boolean | ||
| } & Options & | ||
| Config | ||
|
|
||
| export type MemoryStore = Store & { | ||
| dump: LRU['dump'] | ||
| load: LRU['load'] | ||
| calculatedSize: LRU['calculatedSize'] | ||
| get size(): number | ||
| } | ||
| export type MemoryCache = Cache<MemoryStore> | ||
|
|
||
| /** | ||
| * Wrapper for lru-cache. | ||
| */ | ||
| export function memoryStore(arguments_?: MemoryConfig): MemoryStore { | ||
| const shouldCloneBeforeSet = arguments_?.shouldCloneBeforeSet !== false // Clone by default | ||
| const isCacheable = arguments_?.isCacheable ?? (value => value !== undefined) | ||
|
|
||
| const lruOptions = { | ||
| ttlAutopurge: true, | ||
| ...arguments_, | ||
| max: arguments_?.max ?? 500, | ||
| ttl: arguments_?.ttl === undefined ? 0 : arguments_.ttl, | ||
| } | ||
|
|
||
| const lruCache = new LRUCache(lruOptions) | ||
|
|
||
| return { | ||
| async del(key) { | ||
| lruCache.delete(key) | ||
| }, | ||
| get: async <T>(key: string) => lruCache.get(key) as T, | ||
| keys: async (pattern?: string) => { | ||
| const keys = [...lruCache.keys()] | ||
| if (!pattern) | ||
| return keys | ||
|
|
||
| const regex = new RegExp(`(?<!.)${pattern}`) | ||
| return keys.filter(key => regex.test(key)) | ||
| }, | ||
|
|
||
| mget: async (...arguments_) => arguments_.map(x => lruCache.get(x)), | ||
| async mset(arguments_, ttl?) { | ||
| const opt = { ttl: ttl ?? lruOptions.ttl } as const | ||
| for (const [key, value] of arguments_) { | ||
| if (!isCacheable(value)) | ||
| throw new Error(`no cacheable value ${JSON.stringify(value)}`) | ||
|
|
||
| if (shouldCloneBeforeSet) | ||
| lruCache.set(key, clone(value), opt) | ||
| else | ||
| lruCache.set(key, value, opt) | ||
| } | ||
| }, | ||
| async mdel(...arguments_) { | ||
| for (const key of arguments_) | ||
| lruCache.delete(key) | ||
| }, | ||
| async reset() { | ||
| lruCache.clear() | ||
| }, | ||
| ttl: async key => lruCache.getRemainingTTL(key), | ||
| async set(key, value, opt) { | ||
| if (!isCacheable(value)) | ||
| throw new Error(`no cacheable value ${JSON.stringify(value)}`) | ||
|
|
||
| if (shouldCloneBeforeSet) | ||
| value = clone(value) | ||
|
|
||
| const ttl = opt ?? lruOptions.ttl | ||
|
|
||
| lruCache.set(key, value, { ttl }) | ||
| }, | ||
| get calculatedSize() { | ||
| return lruCache.calculatedSize | ||
| }, | ||
| /** | ||
| * This method is not available in the caching modules. | ||
| */ | ||
| get size() { | ||
| return lruCache.size | ||
| }, | ||
| /** | ||
| * This method is not available in the caching modules. | ||
| */ | ||
| dump: () => lruCache.dump(), | ||
| /** | ||
| * This method is not available in the caching modules. | ||
| */ | ||
| load(...arguments_: Parameters<LRU['load']>) { | ||
| lruCache.load(...arguments_) | ||
| }, | ||
| } | ||
| export function memoryStore(options?: MemoryConfig) { | ||
| const store = new KeyvCacheableMemory(options) | ||
| const keyv = new Keyv({ store }, { namespace: '' }) | ||
| return keyv | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,24 +1,33 @@ | ||
| import type { RedisOptions } from 'ioredis' | ||
| import { Redis } from 'ioredis' | ||
| import type { Config } from 'cache-manager' | ||
| import { RedisClusterConfig, redisInsStore } from 'cache-manager-ioredis-yet' | ||
| import type { KeyvRedisOptions, RedisClientConnectionType, RedisClientOptions, RedisClusterOptions } from '@keyv/redis' | ||
| import KeyvRedis, { Keyv } from '@keyv/redis' | ||
|
|
||
| export interface RedisConnect { | ||
| host?: string | ||
| port?: number | ||
| username?: string | ||
| password?: string | ||
| db?: number | ||
| } | ||
|
|
||
| export function redisStore( | ||
| options?: (RedisOptions | { clusterConfig: RedisClusterConfig }) & Config, | ||
| connect?: string | RedisClientOptions | RedisClusterOptions | RedisClientConnectionType | RedisConnect, | ||
| options?: KeyvRedisOptions, | ||
| ) { | ||
| options ||= {} | ||
| const redisCache | ||
| = 'clusterConfig' in options | ||
| ? new Redis.Cluster( | ||
| options.clusterConfig.nodes, | ||
| options.clusterConfig.options, | ||
| ) | ||
| : new Redis(options) | ||
|
|
||
| return redisInsStore(redisCache, options) | ||
| } | ||
| if (typeof connect === 'object' && connect) { | ||
| if (Reflect.get(connect, 'host')) { | ||
| const c = connect as RedisConnect | ||
| let connectStr = 'redis://' | ||
| if (c.username) | ||
| connectStr += `${c.username}:${c.password}@` | ||
| connectStr += `${c.host}:${c.port}` | ||
| if (c.db) | ||
| connectStr += `/${c.db}` | ||
| connect = connectStr | ||
| } | ||
| } | ||
|
|
||
| export { | ||
| RedisClusterConfig, | ||
| redisInsStore, | ||
| const keyv = new Keyv(new KeyvRedis(connect, options), { | ||
| namespace: '', | ||
| }) | ||
| return keyv | ||
| } |
Oops, something went wrong.