Return key instead of value

wunderwerkio · Sep 1, 2022 · 5301932 · 5301932
1 parent 040bd3d
commit 5301932
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/map/base.ts b/src/map/base.ts
@@ -39,7 +39,7 @@ export abstract class LRUMap<
     }
   }
 
-  public set(key: K, item: I): I[] {
+  public set(key: K, item: I): K[] {
     // Check if entry exists.
     const existingEntry = this.keymap.get(key);
     if (existingEntry) {
@@ -70,7 +70,7 @@ export abstract class LRUMap<
     return this.evict();
   }
 
-  protected shift(): I | null {
+  protected shift(): K | null {
     const entry = this.oldest;
 
     if (entry) {
@@ -91,7 +91,7 @@ export abstract class LRUMap<
 
       this._length--;
 
-      return entry.item;
+      return entry.key;
     }
 
     return null;
@@ -129,7 +129,7 @@ export abstract class LRUMap<
     this.newest = entry;
   }
 
-  protected abstract evict(): I[];
+  protected abstract evict(): K[];
 
   public [Symbol.iterator]() {
     return new EntryIterator(this.oldest);

diff --git a/src/map/limit.ts b/src/map/limit.ts
@@ -18,7 +18,7 @@ export class LimitBasedLRUMap<K extends Key, V extends Value> extends LRUMap<
     }
   }
 
-  protected evict(): LRUItem<V>[] {
+  protected evict() {
     if (this._length > this.limit) {
       return [this.shift()];
     }