feat: implementing cache (not tested yet)

yUnreal · May 22, 2024 · 27e5e3e · 27e5e3e
1 parent 5188d72
commit 27e5e3e
Show file tree

Hide file tree

Showing 3 changed files with 59 additions and 5 deletions.
diff --git a/src/managers/BaseCache.ts b/src/managers/BaseCache.ts
@@ -0,0 +1,10 @@
+import { Cache } from '../structs/Cache';
+import { CacheOptions } from '../typings/model';
+
+export abstract class BaseCache<K, V> {
+    public cache: Cache<K, V>;
+
+    public constructor(options: CacheOptions) {
+        this.cache = new Cache<K, V>(options);
+    }
+}
diff --git a/src/structs/Cache.ts b/src/structs/Cache.ts
@@ -0,0 +1,15 @@
+import { CacheOptions } from '../typings/model';
+
+export class Cache<K, V> extends Map<K, V> {
+    public constructor(public options: CacheOptions) {
+        super();
+    }
+
+    public set(key: K, value: V) {
+        if (this.size === this.options.limit) return this;
+
+        setTimeout(() => this.delete(key), this.options.lifetime);
+
+        return super.set(key, value);
+    }
+}
diff --git a/src/structs/Model.ts b/src/structs/Model.ts
@@ -15,9 +15,19 @@ import uuid from 'uuid-random';
 import { inspect } from 'util';
 import { QueryManager } from '../managers/QueryManager';
 import { ModelOptions } from '../typings/model';
+import { BaseCache } from '../managers/BaseCache';
+
+export class Model<S extends Schema<AnyObject>> extends BaseCache<
+    string,
+    Doc<Infer<S>>
+> {
+    public constructor(public options: ModelOptions<S>) {
+        super(options.cache ?? { lifetime: Infinity });
+    }
 
-export class Model<S extends Schema<AnyObject>> {
-    public constructor(public options: ModelOptions<S>) {}
+    protected isUsingCache() {
+        return Boolean(this.options.cache);
+    }
 
     public get name() {
         return this.options.name;
@@ -62,9 +72,19 @@ export class Model<S extends Schema<AnyObject>> {
         });
 
         if (!doc) return null;
-        if (projection) return new Doc(project(doc, projection), this);
+        if (projection) {
+            const createdDoc = new Doc(project(doc, projection), this);
+
+            if (this.isUsingCache()) this.cache.set(createdDoc.id, createdDoc);
+
+            return createdDoc;
+        }
+
+        const createdDoc = new Doc(doc, this);
 
-        return new Doc(doc, this);
+        if (this.isUsingCache()) this.cache.set(createdDoc.id, createdDoc);
+
+        return createdDoc;
     }
 
     public deleteOne(options: DeleteOneOptions<Infer<S>> | string) {
@@ -78,6 +98,8 @@ export class Model<S extends Schema<AnyObject>> {
             return null;
         }
 
+        this.cache.delete(doc.id);
+
         return doc;
     }
 
@@ -95,6 +117,8 @@ export class Model<S extends Schema<AnyObject>> {
             return data;
         });
 
+        // TODO: if (this.isUsingCache()) this.cache.set;
+
         return;
     }
 
@@ -111,7 +135,12 @@ export class Model<S extends Schema<AnyObject>> {
             return crrData;
         });
 
-        return new Doc(data, this);
+        // @ts-expect-error Fix this soon
+        const createdDoc = new Doc<Infer<S>>(data, this);
+
+        this.cache.set(createdDoc.id, createdDoc);
+
+        return createdDoc;
     }
 
     public createMany(...data: Infer<S>[]) {