chore: add decompression for mget operation

momentohq · Apr 19, 2024 · 0ccd643 · 0ccd643
1 parent a85ce5c
commit 0ccd643
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/momento-redis-adapter.ts b/src/momento-redis-adapter.ts
@@ -694,8 +694,17 @@ export class MomentoRedisAdapter
     const resp = await this.momentoClient.getBatch(this.cacheName, args);
 
     if (resp instanceof CacheGetBatch.Success) {
-      const keyValueRecord = resp.values();
-      return Object.values(keyValueRecord);
+      const keyValueRecord = resp.valuesRecordStringUint8Array();
+      const maybeCompressedValues = Object.values(keyValueRecord);
+      if (this.useCompression) {
+        return await Promise.all(
+          maybeCompressedValues.map(async value => {
+            return await decompress(value);
+          })
+        );
+      } else {
+        return Object.values(resp.values());
+      }
     } else if (resp instanceof CacheGetBatch.Error) {
       this.emitError('mget', resp.message(), resp.errorCode());
     }