如何使用NestJS将网络链接图片保存至磁盘？

问题根源 #

你当前的代码存在核心错误：

• @UseInterceptors、FileInterceptor这类装饰器仅能用于控制器方法，不能在服务内部定义，Nest无法识别这类用法，直接引发报错。
• FileInterceptor是处理客户端上传文件的工具，完全不适用于下载网络图片的场景。

正确实现方案 #

我们需要通过HTTP请求下载网络图片，再将文件写入本地磁盘，具体步骤如下：

1. 安装依赖 #

先安装HTTP客户端和增强型文件处理工具：

npm install axios fs-extra

2. 修改服务代码 #

替换原有avatar方法，实现图片下载与本地存储逻辑：

import { Injectable } from '@nestjs/common';
import axios from 'axios';
import * as fs from 'fs-extra';
import { join } from 'path';

@Injectable()
export class ApiService {
  constructor(private readonly imageModel: any) {}

  async avatar(doc) {
    if (!doc?.avatar) {
      throw new Error('用户头像链接不存在');
    }

    try {
      // 下载图片流
      const response = await axios({
        url: doc.avatar,
        method: 'GET',
        responseType: 'stream',
      });

      // 定义可靠的保存路径（基于项目根目录生成绝对路径）
      const saveDir = join(process.cwd(), 'files');
      // 确保目标文件夹存在，不存在则自动创建
      await fs.ensureDir(saveDir);
      // 用用户ID作为文件名，可根据实际需求调整
      const fileName = `${doc.id}.jpg`;
      const savePath = join(saveDir, fileName);

      // 将图片流写入磁盘
      const writer = fs.createWriteStream(savePath);
      response.data.pipe(writer);

      // 等待文件写入完成
      await new Promise((resolve, reject) => {
        writer.on('finish', resolve);
        writer.on('error', reject);
      });

      // 保存记录到数据库，可新增字段存储本地路径
      return await new this.imageModel({
        userId: doc.id,
        hash: doc.avatar,
        localPath: savePath,
      }).save();
    } catch (error) {
      throw new Error(`图片保存失败: ${error.message}`);
    }
  }

  // 你的getUser方法保持原有逻辑
  async getUser(id: string) {
    // 获取用户数据的业务逻辑
  }
}

3. 控制器代码无需修改 #

原有控制器逻辑合理，保持不变即可：

import { Controller, Get, Param } from '@nestjs/common';
import { ApiService } from './api.service';

@Controller()
export class AppController {
  constructor(private readonly apiService: ApiService) {}

  @Get('user/:id/avatar')
  async setAvatar(@Param('id') id: string) {
    const userData = await this.apiService.getUser(id);    
    return this.apiService.avatar(userData);
  }
}

注意事项 #

• 路径可靠性：使用process.cwd()获取项目根目录，结合join生成绝对路径，避免相对路径引发的层级混乱。
• 图片格式适配：若头像链接后缀不固定，可从响应头content-type提取格式，或用哈希值作为文件名。
• 异常处理：需捕获网络请求失败、磁盘写入失败等异常，避免程序崩溃。

内容的提问来源于stack exchange，提问作者José Luis