跳到主要内容

文件流式传输

备注

本章介绍如何从您的 HTTP 应用程序中流式传输文件。下面的示例不适用于 GraphQL 或微服务应用程序。

在某些情况下,您可能希望从您的 REST API 将文件发送回客户端。为了在 Nest 中实现这一点,通常会执行以下操作:

@Controller('file')
export class FileController {
@Get()
getFile(@Res() res: Response) {
const file = createReadStream(join(process.cwd(), 'package.json'));
file.pipe(res);
}
}

但是通过这样做,您最终会失去对控制器后拦截器逻辑的访问。 为了处理这个问题,您可以返回一个 StreamableFile 实例,在底层,框架将负责管道化响应。

StreamableFile 类

StreamableFile 是一个保存即将返回的流的类。要创建一个新的 StreamableFile,您可以将 Buffer 或 Stream 传递给 StreamableFile 构造函数。

提示

可以从 @nestjs/common 导入 StreamableFile 类。

跨平台支持

Fastify 默认情况下可以支持在不需要调用 stream.pipe(res) 的情况下发送文件,因此根本不需要使用 StreamableFile 类。 然而,Nest 在两种平台类型中都支持 StreamableFile 的使用,因此如果在 Express 和 Fastify 之间切换, 无需担心两个引擎之间的兼容性问题。

示例

下面是一个简单的示例,将 package.json 作为文件返回而不是 JSON, 但这个想法自然扩展到图像、文档和任何其他文件类型。

import { Controller, Get, StreamableFile } from '@nestjs/common';
import { createReadStream } from 'fs';
import { join } from 'path';

@Controller('file')
export class FileController {
@Get()
getFile(): StreamableFile {
const file = createReadStream(join(process.cwd(), 'package.json'));
return new StreamableFile(file);
}
}

默认的内容类型是 application/octet-stream,如果您需要自定义响应, 可以使用 res.set 方法或 @Header() 装饰器,如下所示:

import { Controller, Get, StreamableFile, Res, Header } from '@nestjs/common';
import { createReadStream } from 'fs';
import { join } from 'path';
import type { Response } from 'express';

@Controller('file')
export class FileController {
@Get()
getFile(@Res({ passthrough: true }) res: Response): StreamableFile {
const file = createReadStream(join(process.cwd(), 'package.json'));
res.set({
'Content-Type': 'application/json',
'Content-Disposition': 'attachment; filename="package.json"',
});
return new StreamableFile(file);
}

// 或者甚至:
@Get()
@Header('Content-Type', 'application/json')
@Header('Content-Disposition', 'attachment; filename="package.json"')
getStaticFile(): StreamableFile {
const file = createReadStream(join(process.cwd(), 'package.json'));
return new StreamableFile(file);
}
}