Как установить декоратор на метод, который не является эндпоинтом?

Question

[Movsar-Khalakhoev]

Здравствуйте!

Вопрос по NestJS.
Стоит задача сделать так, чтобы при вызове метода secondMethod контроллера из метода firstMethod срабатывал кастомный декоратор. На данный момент этого не происходит. Я заметил, что срабатывают только те декораторы, которые повешаны именно на эндпоинты, т. е. на методы, которые помечены декораторами Get, Post, Patch и т.д.

class Controller {
      
      @Get('/')
      firstMethod() {
        secondMethod()
      }
      
      @CustomDecorator()
      secondMethod() {}
    }

Прикрепил минимальную версию кода. Не работает CustomDecorator.

Спасибо за помощь!

Comments

[Movsar-Khalakhoev]

Уточнение: в кастомном декораторе используется встроенный декоратор useGuards

Answer 1

[Дмитрий Свиридов]

Вот так сделайте декоратор - заработает:

const CustomDecorator = () => (target: Object, propertyKey: string, descriptor: PropertyDescriptor) => {
  let sourceMethod = descriptor.value;
  descriptor.value = function (...args: any[]) {
    console.log("before");
    let result = sourceMethod.apply(this, args);
    console.log("after");
    return result;
  }
}

P.s. вот тут можете почитать более детально

Comments

[Movsar-Khalakhoev]

Спасибо. Это действительно некоторое продвижение.

Однако я не нашёл способа заставить работать это с встроенным декоратором UseGuards :(
Придётся каждый раз создавать экземпляр моего guard. А там мне нужны некоторые сервисы.
Или я ошибаюсь?

[Дмитрий Свиридов]

Movsar-Khalakhoev, в NestJS можно объединять декораторы вот так https://docs.nestjs.com/custom-decorators#decorato.... Возможно, есть смысл таким образом объединить UseGuards с вашим декоратором и использовать объединённый.

[lssssssssssl]

Movsar-Khalakhoev, А вы вообще уверены, что вам это всё нужно?(Создавать какой-то метод в контроллере, который не имеет отношения к http ентпоинту, создавать к нему декоратор, смешивать этот декоратор с другим декоратором)? Я более чем уверен, что ваша задача может решиться по другому - более простому способу.
Да и к тому же, размешать методы в контроллере, которые не имеют отношение к http энтпоинту - неправильно.

[Movsar-Khalakhoev]

lssssssssssl, я хотел сделать единую точку входа для всех get-запросов, в котором в зависимости от набора параметров, вызывается нужная функция, которая защищена моим декоратором.

Сейчас я нашёл другое решение. В контроллере я поставил для всех отдельных фукнций декоратор Get, с помощью которого мой декоратор заработал. Для каждой отдельной фукнции назначен свой роут.
В модуль я добавил миддлвэйр, который в зависимости от параметров запроса изменяет путь на необходимый для нужной функции. В результате всё работает как нужно: у меня единая точка входа для всех гет-запросов модуля, в будущем я могу разделить функцию на более мелкие фукнции и защищать их независимо друг от друга, при этом внешнее апи никак не изменится.
Возмонжно, это не совсем корректное решение, но оно работает и, насколько я вижу, архитектурно ничего не усложняет.

Спасибо всем за отзывчивость! На стэковерфлоу всё ужасно медленно(

[Дмитрий Свиридов]

Movsar-Khalakhoev, если мой ответ вас устроил - отметьте, пожалуйста, решением. :)