feat(cache): cacheTTL

jiawei397 · Jan 18, 2022 · 5132b8c · 5132b8c
1 parent f32c116
commit 5132b8c
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 4 deletions.
diff --git a/modules/cache/example/app.controller.ts b/modules/cache/example/app.controller.ts
@@ -1,5 +1,11 @@
-import { Controller, Get, Query, UseInterceptors } from "../../../mod.ts";
-import { CacheInterceptor } from "../mod.ts";
+import {
+  Controller,
+  Get,
+  Params,
+  Query,
+  UseInterceptors,
+} from "../../../mod.ts";
+import { CacheInterceptor, CacheTTL } from "../mod.ts";
 
 @Controller("")
 @UseInterceptors(CacheInterceptor)
@@ -14,4 +20,14 @@ export class AppController {
       }, 500);
     });
   }
+
+  @Get("/delay/:id")
+  @CacheTTL(3000)
+  delay2(@Params("id") id: string) {
+    return new Promise((resolve) => {
+      setTimeout(() => {
+        resolve("delay " + id);
+      }, 500);
+    });
+  }
 }
diff --git a/modules/cache/src/cache.constant.ts b/modules/cache/src/cache.constant.ts
@@ -1 +1,3 @@
 export const optionKey = Symbol("options");
+
+export const META_CACHE_TTL_KEY = Symbol("meta:cache:ttl");
diff --git a/modules/cache/src/cache.interceptor.ts b/modules/cache/src/cache.interceptor.ts
@@ -6,11 +6,18 @@ import {
   NestInterceptor,
   NestInterceptorOptions,
   Next,
+  Reflect,
 } from "../../../mod.ts";
 import { isDebug } from "../../../src/utils.ts";
-import { optionKey } from "./cache.constant.ts";
+import { META_CACHE_TTL_KEY, optionKey } from "./cache.constant.ts";
 import { CacheModuleOptions } from "./cache.interface.ts";
 
+export function CacheTTL(ttl: number) {
+  return (_target: any, _methodName: string, descriptor: any) => {
+    Reflect.defineMetadata(META_CACHE_TTL_KEY, ttl, descriptor.value);
+  };
+}
+
 @Injectable()
 export class CacheInterceptor implements NestInterceptor {
   ttl: number;
@@ -70,9 +77,14 @@ export class CacheInterceptor implements NestInterceptor {
     }
     const result = next();
     cache.set(key, result);
+
+    const ttl = Reflect.getOwnMetadata(
+      META_CACHE_TTL_KEY,
+      options.target[options.methodName],
+    ) || this.ttl;
     const st = setTimeout(() => {
       cache.delete(key);
-    }, this.ttl * 1000);
+    }, ttl * 1000);
     Promise.resolve(result)
       .then((val) => {
         if (this.cacheModuleOptions?.isCacheableValue) {