34 lines
814 B
TypeScript
34 lines
814 B
TypeScript
|
import {
|
|||
|
|
CallHandler,
|
|||
|
|
ExecutionContext,
|
|||
|
|
Injectable,
|
|||
|
|
NestInterceptor,
|
|||
|
|
} from '@nestjs/common';
|
|||
|
|
import { Observable } from 'rxjs';
|
|||
|
|
import { map } from 'rxjs/operators';
|
|||
|
|
import { ApiResponse } from '../dto/api-response.dto';
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 全局统一响应拦截器
|
|||
|
|
* 把所有 controller 返回值包装成 { code, msg, data } 结构
|
|||
|
|
*/
|
|||
|
|
@Injectable()
|
|||
|
|
export class TransformInterceptor<T>
|
|||
|
|
implements NestInterceptor<T, ApiResponse<T>>
|
|||
|
|
{
|
|||
|
|
intercept(
|
|||
|
|
context: ExecutionContext,
|
|||
|
|
next: CallHandler<T>,
|
|||
|
|
): Observable<ApiResponse<T>> {
|
|||
|
|
return next.handle().pipe(
|
|||
|
|
map((data) => {
|
|||
|
|
// 如果已经是 ApiResponse，则原样返回
|
|||
|
|
if (data instanceof ApiResponse) {
|
|||
|
|
return data;
|
|||
|
|
}
|
|||
|
|
return ApiResponse.success(data) as ApiResponse<T>;
|
|||
|
|
}),
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
}
|