nest拦截器

213 阅读1分钟

响应拦截器

新建一个Response.ts文件
import { CallHandler, ExecutionContext, Injectable, NestInterceptor } from '@nestjs/common';
import { Observable } from 'rxjs';
import { map } from 'rxjs/operators';

interface Data<T>  {
    data:T
}

@Injectable()
export class ResponseInterceptor<T> implements NestInterceptor {
  intercept(
    context: ExecutionContext,
    next: CallHandler<any>,
  ): Observable<Data<T>> | Promise<Observable<any>> {
    return next.handle().pipe(map(data => {
        return {
            data,
            status:0,
            msg:'成功',
            success:true
        }
    }));
  }
}

异常处理拦截器

 新建一个Filter.ts文件
 
import { ArgumentsHost, Catch, ExceptionFilter, HttpException } from "@nestjs/common";
import { Request, Response } from "express";


@Catch(HttpException)
export class HttpFilter implements ExceptionFilter {
    catch(exception: HttpException, host: ArgumentsHost) {
        const ctx = host.switchToHttp()
        
        const request = ctx.getRequest<Request>()

        const response = ctx.getResponse<Response>()

        const status = exception.getStatus()

        response.status(status).json({
            success:false,
            time:new Date(),
            data:exception.message,
            status,
            path:request.url
        })

    }
}

在mian.ts 中使用全局定义的拦截器

import { AppModule } from './app.module';

import * as session from 'express-session';
import * as cors from 'cors'
import { VersioningType } from '@nestjs/common';
import {Request,Response,NextFunction} from 'express'
import { NestExpressApplication } from '@nestjs/platform-express';

// 响应拦截器
import { ResponseInterceptor } from './common/Response'
// 异常拦截器
import { HttpFilter } from './common/Filter'

async function bootstrap() {
  const app = await NestFactory.create<NestExpressApplication>(AppModule);
    // 使用响应拦截器
    app.useGlobalInterceptors(new ResponseInterceptor())
    // 使用异常拦截器
    app.useGlobalFilters(new HttpFilter())
  await app.listen(3000);
}
bootstrap();