nestjs-aop/src/__test__/fixture/cache/cache.decorator.ts

<!-- PROJECT LOGO --> <br /> <div align="center"> <a href="https://github.com/toss/nestjs-aop"> <img src="https://toss.tech/wp-content/uploads/2022/11/tech-article-nest-js-02.png" alt="Logo" height="200"> </a>

import { Aspect } from '../../../aspect';
import { createDecorator } from '../../../create-decorator';
import { LazyDecorator, WrapParams } from '../../../lazy-decorator';
import { CacheService } from './cache.service';

export const CACHE = Symbol('CACHE');

export const Cache = () => createDecorator(CACHE);

@Aspect(CACHE)
export class CacheDecorator implements LazyDecorator<any, void> {
  constructor(private readonly cacheService: CacheService) {}

  wrap({ method }: WrapParams<any, void>) {
    return (...args: any[]) => {
      const cacheKey = JSON.stringify(args);
      const cache = this.cacheService.get(cacheKey);
      if (cache) {
        return cache;
      }

      const result = method(...args);
      this.cacheService.set(cacheKey, result);
      return result;
    };
  }
}