nestjs - 04

Nestjs 自定义装饰器

用命令自动生成装饰器文件，亦可手动创建

nest g d decorator

// decorator.ts
import { SetMetadata, createParamDecorator, ExecutionContext, applyDecorators } from '@nestjs/common';
import type { Request } from 'express'
// 权限控制
export const Role = (...args: string[]) => SetMetadata('role', args);

// 获取URL
// @param data => 使用decorator时候传递的参数  context => 上下文
export const ReqUrl = createParamDecorator((data: string, context: ExecutionContext) => {
  const req = context.switchToHttp().getRequest<Request>()
  console.log('data:', data);
  return req.url
  // applyDecorators  组合装饰器，可以组合多种装饰器
  // eg return applyDecorators(Role, xx, xxxx)
})

Decorator 使用方法

import { Controller,  } from '@nestjs/common'
import { Role, ReqUrl } from './decorator'
@Controller()
export class DemoController {
  @Get()
  @Role('admin')
  findAll(@ReqUrl('data') url: string) {
    console.log('url:', url)
    return this.guardService.findAll();
  }
}