使用nestjs拦截器修改响应

Question

我有拦截器来修改图标路径，如下所示：

@Injectable()
export class GetProgramIconInterceptor implements NestInterceptor {
    intercept(context: ExecutionContext, next: CallHandler<any>): Observable<any> | Promise<Observable<any>> {
        return next.handle().pipe(
            map(data => {
                return {
                    ...data,
                    icon: generalFUnctions.getApplicationIcon(data.icon)
                };
            }),
        );
    }
}

当我在我的 find 方法中使用它时，它可以正常工作并修改图标路径。

@UseInterceptors(GetProgramIconInterceptor)
async find(id: string): Promise<Application> {
    const application = await this.repository.findById(id);
     if (!application) throw new ApplicationNotFoundException();
     return application;
}

当我将此拦截器用于 findAll 方法时，它无法正常工作并且不会修改路径。

@UseInterceptors(GetProgramIconInterceptor)
async findAll(): Promise<Application[]> {
     return this.repository.find();
}

我知道 findAll 方法的结果是一个数组，我可以为它创建另一个拦截器。 有没有办法用一个拦截器来处理它

Answer 1

由于您正在处理来自存储库模型函数（ find和findById ）的承诺（并且还不是可观察的）结果，当管道进入可观察对象时，这意味着您正在观察repository或repository[]

附上一个示例的代码框，我已经创建了可能是单个对象或对象数组的资源的映射。

这里的神奇之处在于确保我们有一个条件表达式来判断资源是否为数组

map((repositoryOrRespositories) => {
  if (Array.isArray(repositoryOrRespositories)) {
    return item.map((repository) => ({
      ...repository,
      icon: generalFUnctions.getApplicationIcon(repository.icon)
    }));
  } else {
    return { ...repositoryOrRespositories, icon: generalFUnctions.getApplicationIcon(repositoryOrRespositories.icon) };
  }
})

如果存储库模型的结果是可观察的，则必须使用 rxjs 中的iif表达式。