@mastra/redis 1.0.2 → 1.1.0

package/dist/index.d.ts

@@ -1,2 +1,3 @@
 export * from './storage/index.js';
+export { RedisServerCache, type RedisClient, type RedisServerCacheOptions, upstashPreset, nodeRedisPreset, } from './cache.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,cAAc,WAAW,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,cAAc,WAAW,CAAC;AAG1B,OAAO,EACL,gBAAgB,EAChB,KAAK,WAAW,EAChB,KAAK,uBAAuB,EAC5B,aAAa,EACb,eAAe,GAChB,MAAM,SAAS,CAAC"}

package/dist/index.js

@@ -4,6 +4,7 @@ import { MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ensureDa
 import crypto2 from 'crypto';
 import { saveScorePayloadSchema } from '@mastra/core/evals';
 import { createClient } from 'redis';
+import { MastraServerCache } from '@mastra/core/cache';
 
 // src/storage/domains/memory/index.ts
 function getKey(tableName, keys) {
@@ -1789,7 +1790,105 @@ var RedisStore = class extends MastraStorage {
     return `redis://${config.host}:${config.port || 6379}/${config.db || 0}`;
   }
 };
+var defaultSetWithExpiry = (client, key, value, seconds) => {
+  return client.set(key, value, "EX", seconds);
+};
+var defaultScanKeys = (client, cursor, pattern, count) => {
+  return client.scan(cursor, "MATCH", pattern, "COUNT", count);
+};
+var RedisServerCache = class extends MastraServerCache {
+  client;
+  keyPrefix;
+  ttlSeconds;
+  setWithExpiry;
+  scanKeys;
+  constructor(config, options = {}) {
+    super({ name: "RedisServerCache" });
+    this.client = config.client;
+    this.keyPrefix = options.keyPrefix ?? "mastra:cache:";
+    this.ttlSeconds = options.ttlSeconds ?? 300;
+    this.setWithExpiry = options.setWithExpiry ?? defaultSetWithExpiry;
+    this.scanKeys = options.scanKeys ?? defaultScanKeys;
+  }
+  getKey(key) {
+    return `${this.keyPrefix}${key}`;
+  }
+  serialize(value) {
+    return JSON.stringify(value);
+  }
+  deserialize(value) {
+    if (typeof value === "string") {
+      try {
+        return JSON.parse(value);
+      } catch {
+        return value;
+      }
+    }
+    return value;
+  }
+  async get(key) {
+    const fullKey = this.getKey(key);
+    const value = await this.client.get(fullKey);
+    if (value === null) {
+      return null;
+    }
+    return this.deserialize(value);
+  }
+  async set(key, value) {
+    const fullKey = this.getKey(key);
+    const serialized = this.serialize(value);
+    if (this.ttlSeconds > 0) {
+      await this.setWithExpiry(this.client, fullKey, serialized, this.ttlSeconds);
+    } else {
+      await this.client.set(fullKey, serialized);
+    }
+  }
+  async listLength(key) {
+    const fullKey = this.getKey(key);
+    return this.client.llen(fullKey);
+  }
+  async listPush(key, value) {
+    const fullKey = this.getKey(key);
+    const serialized = this.serialize(value);
+    await this.client.rpush(fullKey, serialized);
+    if (this.ttlSeconds > 0) {
+      await this.client.expire(fullKey, this.ttlSeconds);
+    }
+  }
+  async listFromTo(key, from, to = -1) {
+    const fullKey = this.getKey(key);
+    const values = await this.client.lrange(fullKey, from, to);
+    return values.map((v) => this.deserialize(v));
+  }
+  async delete(key) {
+    const fullKey = this.getKey(key);
+    await this.client.del(fullKey);
+  }
+  async clear() {
+    const pattern = `${this.keyPrefix}*`;
+    let cursor = "0";
+    do {
+      const [nextCursor, keys] = await this.scanKeys(this.client, cursor, pattern, 100);
+      if (keys.length > 0) {
+        await this.client.del(...keys);
+      }
+      cursor = nextCursor;
+    } while (cursor !== "0" && cursor !== 0);
+  }
+  async increment(key) {
+    const fullKey = this.getKey(key);
+    return this.client.incr(fullKey);
+  }
+};
+var upstashPreset = {
+  setWithExpiry: (client, key, value, seconds) => client.set(key, value, { ex: seconds }),
+  scanKeys: (client, cursor, pattern, count) => client.scan(cursor, { match: pattern, count })
+};
+var nodeRedisPreset = {
+  setWithExpiry: (client, key, value, seconds) => client.set(key, value, { EX: seconds }),
+  scanKeys: (client, cursor, pattern, count) => client.scan(cursor, { MATCH: pattern, COUNT: count })
+};
 
-export { RedisStore, ScoresRedis, StoreMemoryRedis, WorkflowsRedis };
+export { RedisServerCache, RedisStore, ScoresRedis, StoreMemoryRedis, WorkflowsRedis, nodeRedisPreset, upstashPreset };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map