Merge pull request #112 from quangtran88/feat/controller-cache-ttl

feat(cache-manager): allow @CacheTTL at the controller level

Author: kamilmysliwiec

lib/interceptors/cache.interceptor.ts

@@ -54,7 +54,9 @@ export class CacheInterceptor implements NestInterceptor {
   ): Promise<Observable<any>> {
     const key = this.trackBy(context);
     const ttlValueOrFactory =
-      this.reflector.get(CACHE_TTL_METADATA, context.getHandler()) ?? null;
+      this.reflector.get(CACHE_TTL_METADATA, context.getHandler()) ??
+      this.reflector.get(CACHE_TTL_METADATA, context.getClass()) ??
+      null;
 
     if (!key) {
       return next.handle();

tests/e2e/custom-ttl.spec.ts

@@ -29,6 +29,18 @@ describe('Caching Custom TTL', () => {
     await request(server).get('/').expect(200, '0');
   });
 
+  it('should return a different value after the TTL of the controller is elapsed', async () => {
+    await request(server).get('/controller').expect(200, '0');
+    await new Promise(resolve => setTimeout(resolve, 600));
+    await request(server).get('/controller').expect(200, '1');
+  });
+
+  it('should return the cached value within the TTL of the controller', async () => {
+    await request(server).get('/controller').expect(200, '0');
+    await new Promise(resolve => setTimeout(resolve, 300));
+    await request(server).get('/controller').expect(200, '0');
+  });
+
   afterEach(async () => {
     await app.close();
   });

tests/src/custom-ttl/custom-ttl.controller.ts

@@ -2,6 +2,7 @@ import { Controller, Get, UseInterceptors } from '@nestjs/common';
 import { CacheInterceptor, CacheTTL } from '../../../lib';
 
 @Controller()
+@CacheTTL(600)
 export class CustomTtlController {
   counter = 0;
   constructor() {}
@@ -12,4 +13,10 @@ export class CustomTtlController {
   getNumber() {
     return this.counter++;
   }
+
+  @Get('/controller')
+  @UseInterceptors(CacheInterceptor)
+  getNumberWithControllerTTL() {
+    return this.counter++;
+  }
 }