@mastra/core
Version:
1 lines • 6.28 kB
Source Map (JSON)
{"version":3,"file":"inmemory-CVHnncRp.cjs","names":["MastraBase","TTLCache"],"sources":["../src/cache/base.ts","../src/cache/inmemory.ts"],"sourcesContent":["import { MastraBase } from '../base';\n\nexport abstract class MastraServerCache extends MastraBase {\n constructor({ name }: { name: string }) {\n super({\n component: 'SERVER_CACHE',\n name,\n });\n }\n\n abstract get(key: string): Promise<unknown>;\n\n abstract listLength(key: string): Promise<number>;\n\n /**\n * Store a value in the cache.\n * @param key - Cache key\n * @param value - Value to store\n * @param ttlMs - Optional per-key TTL in milliseconds. If not provided, uses\n * the implementation's default TTL.\n */\n abstract set(key: string, value: unknown, ttlMs?: number): Promise<void>;\n\n abstract listPush(key: string, value: unknown): Promise<void>;\n\n abstract listFromTo(key: string, from: number, to?: number): Promise<unknown[]>;\n\n abstract delete(key: string): Promise<void>;\n\n abstract clear(): Promise<void>;\n\n /**\n * Atomically increment a counter and return the new value.\n * Used for generating sequential indices for events.\n * Returns 1 on first call (counter starts at 0, increments to 1).\n *\n * For Redis: Uses INCR command which is atomic.\n * For in-memory: Uses a simple counter map.\n */\n abstract increment(key: string): Promise<number>;\n}\n","import { TTLCache } from '@isaacs/ttlcache';\nimport { MastraServerCache } from './base';\n\n/**\n * Options for InMemoryServerCache\n */\nexport interface InMemoryServerCacheOptions {\n /**\n * Maximum number of items to store in cache.\n * Defaults to 1000.\n */\n maxSize?: number;\n\n /**\n * Default TTL in milliseconds for cached items.\n * Defaults to 300000 (5 minutes).\n * Set to 0 to disable TTL (items persist until explicitly deleted or evicted).\n */\n ttlMs?: number;\n}\n\nexport class InMemoryServerCache extends MastraServerCache {\n private cache: TTLCache<string, unknown>;\n private ttlMs: number;\n\n constructor(options: InMemoryServerCacheOptions = {}) {\n super({ name: 'InMemoryServerCache' });\n\n this.ttlMs = options.ttlMs ?? 1000 * 60 * 5;\n // TTLCache requires positive integer or Infinity; use Infinity when TTL is disabled\n const ttl = this.ttlMs > 0 ? this.ttlMs : Infinity;\n\n this.cache = new TTLCache<string, unknown>({\n max: options.maxSize ?? 1000,\n ttl,\n });\n }\n\n async get(key: string): Promise<unknown> {\n return this.cache.get(key);\n }\n\n async set(key: string, value: unknown, ttlMs?: number): Promise<void> {\n if (ttlMs === undefined) {\n this.cache.set(key, value);\n return;\n }\n // TTLCache requires positive integer or Infinity; non-positive overrides\n // mean \"no expiry\" and must be normalized.\n this.cache.set(key, value, { ttl: ttlMs > 0 ? ttlMs : Infinity });\n }\n\n async listLength(key: string): Promise<number> {\n const value = this.cache.get(key);\n if (value === undefined) {\n return 0; // Key doesn't exist - return 0\n }\n if (!Array.isArray(value)) {\n throw new Error(`${key} exists but is not an array`);\n }\n return value.length;\n }\n\n async listPush(key: string, value: unknown): Promise<void> {\n const existing = this.cache.get(key);\n if (Array.isArray(existing)) {\n existing.push(value);\n // Refresh TTL on push by re-setting the key with the updated list\n if (this.ttlMs > 0) {\n this.cache.set(key, existing, { ttl: this.ttlMs });\n }\n } else if (existing !== undefined) {\n throw new Error(`${key} exists but is not an array`);\n } else {\n this.cache.set(key, [value]);\n }\n }\n\n async listFromTo(key: string, from: number, to: number = -1): Promise<unknown[]> {\n const list = this.cache.get(key) as unknown[];\n if (Array.isArray(list)) {\n // Make 'to' inclusive like Redis LRANGE - add 1 unless it's -1\n const endIndex = to === -1 ? undefined : to + 1;\n return list.slice(from, endIndex);\n }\n return [];\n }\n\n async delete(key: string): Promise<void> {\n this.cache.delete(key);\n }\n\n async clear(): Promise<void> {\n this.cache.clear();\n }\n\n async increment(key: string): Promise<number> {\n const value = this.cache.get(key);\n let counter: number;\n if (value === undefined) {\n counter = 1;\n } else if (typeof value === 'number') {\n counter = value + 1;\n } else {\n throw new Error(`${key} exists but is not a number`);\n }\n this.cache.set(key, counter);\n return counter;\n }\n}\n"],"mappings":";;;AAEA,IAAsB,oBAAtB,cAAgDA,aAAAA,WAAW;CACzD,YAAY,EAAE,QAA0B;EACtC,MAAM;GACJ,WAAW;GACX;EACF,CAAC;CACH;AAgCF;;;ACnBA,IAAa,sBAAb,cAAyC,kBAAkB;CACzD;CACA;CAEA,YAAY,UAAsC,CAAC,GAAG;EACpD,MAAM,EAAE,MAAM,sBAAsB,CAAC;EAErC,KAAK,QAAQ,QAAQ,SAAS,MAAO,KAAK;EAE1C,MAAM,MAAM,KAAK,QAAQ,IAAI,KAAK,QAAQ;EAE1C,KAAK,QAAQ,IAAIC,iBAAAA,SAA0B;GACzC,KAAK,QAAQ,WAAW;GACxB;EACF,CAAC;CACH;CAEA,MAAM,IAAI,KAA+B;EACvC,OAAO,KAAK,MAAM,IAAI,GAAG;CAC3B;CAEA,MAAM,IAAI,KAAa,OAAgB,OAA+B;EACpE,IAAI,UAAU,KAAA,GAAW;GACvB,KAAK,MAAM,IAAI,KAAK,KAAK;GACzB;EACF;EAGA,KAAK,MAAM,IAAI,KAAK,OAAO,EAAE,KAAK,QAAQ,IAAI,QAAQ,SAAS,CAAC;CAClE;CAEA,MAAM,WAAW,KAA8B;EAC7C,MAAM,QAAQ,KAAK,MAAM,IAAI,GAAG;EAChC,IAAI,UAAU,KAAA,GACZ,OAAO;EAET,IAAI,CAAC,MAAM,QAAQ,KAAK,GACtB,MAAM,IAAI,MAAM,GAAG,IAAI,4BAA4B;EAErD,OAAO,MAAM;CACf;CAEA,MAAM,SAAS,KAAa,OAA+B;EACzD,MAAM,WAAW,KAAK,MAAM,IAAI,GAAG;EACnC,IAAI,MAAM,QAAQ,QAAQ,GAAG;GAC3B,SAAS,KAAK,KAAK;GAEnB,IAAI,KAAK,QAAQ,GACf,KAAK,MAAM,IAAI,KAAK,UAAU,EAAE,KAAK,KAAK,MAAM,CAAC;EAErD,OAAO,IAAI,aAAa,KAAA,GACtB,MAAM,IAAI,MAAM,GAAG,IAAI,4BAA4B;OAEnD,KAAK,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC;CAE/B;CAEA,MAAM,WAAW,KAAa,MAAc,KAAa,IAAwB;EAC/E,MAAM,OAAO,KAAK,MAAM,IAAI,GAAG;EAC/B,IAAI,MAAM,QAAQ,IAAI,GAAG;GAEvB,MAAM,WAAW,OAAO,KAAK,KAAA,IAAY,KAAK;GAC9C,OAAO,KAAK,MAAM,MAAM,QAAQ;EAClC;EACA,OAAO,CAAC;CACV;CAEA,MAAM,OAAO,KAA4B;EACvC,KAAK,MAAM,OAAO,GAAG;CACvB;CAEA,MAAM,QAAuB;EAC3B,KAAK,MAAM,MAAM;CACnB;CAEA,MAAM,UAAU,KAA8B;EAC5C,MAAM,QAAQ,KAAK,MAAM,IAAI,GAAG;EAChC,IAAI;EACJ,IAAI,UAAU,KAAA,GACZ,UAAU;OACL,IAAI,OAAO,UAAU,UAC1B,UAAU,QAAQ;OAElB,MAAM,IAAI,MAAM,GAAG,IAAI,4BAA4B;EAErD,KAAK,MAAM,IAAI,KAAK,OAAO;EAC3B,OAAO;CACT;AACF"}