@mastra/redis 1.2.0 → 1.2.1

package/dist/index.js

@@ -1799,12 +1799,24 @@ var defaultSetWithExpiry = (client, key, value, seconds) => {
 var defaultScanKeys = (client, cursor, pattern, count) => {
   return client.scan(cursor, "MATCH", pattern, "COUNT", count);
 };
+var defaultGetListLength = (client, key) => {
+  return client.llen(key);
+};
+var defaultPushToList = (client, key, value) => {
+  return client.rpush(key, value);
+};
+var defaultGetListRange = (client, key, start, stop) => {
+  return client.lrange(key, start, stop);
+};
 var RedisServerCache = class extends MastraServerCache {
   client;
   keyPrefix;
   ttlSeconds;
   setWithExpiry;
   scanKeys;
+  getListLength;
+  pushToList;
+  getListRange;
   constructor(config, options = {}) {
     super({ name: "RedisServerCache" });
     this.client = config.client;
@@ -1812,6 +1824,9 @@ var RedisServerCache = class extends MastraServerCache {
     this.ttlSeconds = options.ttlSeconds ?? 300;
     this.setWithExpiry = options.setWithExpiry ?? defaultSetWithExpiry;
     this.scanKeys = options.scanKeys ?? defaultScanKeys;
+    this.getListLength = options.getListLength ?? defaultGetListLength;
+    this.pushToList = options.pushToList ?? defaultPushToList;
+    this.getListRange = options.getListRange ?? defaultGetListRange;
   }
   getKey(key) {
     return `${this.keyPrefix}${key}`;
@@ -1850,19 +1865,19 @@ var RedisServerCache = class extends MastraServerCache {
   }
   async listLength(key) {
     const fullKey = this.getKey(key);
-    return this.client.llen(fullKey);
+    return this.getListLength(this.client, fullKey);
   }
   async listPush(key, value) {
     const fullKey = this.getKey(key);
     const serialized = this.serialize(value);
-    await this.client.rpush(fullKey, serialized);
+    await this.pushToList(this.client, 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);
+    const values = await this.getListRange(this.client, fullKey, from, to);
     return values.map((v) => this.deserialize(v));
   }
   async delete(key) {
@@ -1891,7 +1906,10 @@ var upstashPreset = {
 };
 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 })
+  scanKeys: (client, cursor, pattern, count) => client.scan(cursor, { MATCH: pattern, COUNT: count }),
+  getListLength: (client, key) => client.lLen(key),
+  pushToList: (client, key, value) => client.rPush(key, value),
+  getListRange: (client, key, start, stop) => client.lRange(key, start, stop)
 };
 
 export { RedisServerCache, RedisStore, ScoresRedis, StoreMemoryRedis, WorkflowsRedis, nodeRedisPreset, upstashPreset };