NestJS 入门教程(四):文件上传和下载功能的实现

1. 前言

在上一篇文章中,我们实现了用户认证和授权功能。本文将介绍如何实现文件上传和下载功能,包括单文件上传、多文件上传、文件下载等。

2. 安装依赖

首先安装文件处理相关的依赖:

pnpm add @nestjs/platform-express multer @types/multer

3. 创建文件模块

使用 NestJS CLI 创建文件模块:

nest g module file
nest g controller file
nest g service file

4. 配置文件上传

src/modules/file/config/multer.config.ts 中创建 Multer 配置:

import {
   
    diskStorage } from 'multer';
import {
   
    extname } from 'path';
import {
   
    v4 as uuid } from 'uuid';

export const multerConfig = {
   
   
  storage: diskStorage({
   
   
    destination: './uploads',
    filename: (req, file, callback) => {
   
   
      const fileExtName = extname(file.originalname);
      const fileName = `${
     
     uuid()}${
     
     fileExtName}`;
      callback(null, fileName);
    },
  }),
  fileFilter: (req, file, callback) => {
   
   
    if (!file.originalname.match(/\.(jpg|jpeg|png|gif|pdf|doc|docx|xls|xlsx)$/)) {
   
   
      return callback(new Error('Only image, PDF and document files are allowed!'), false);
    }
    callback(null, true);
  },
  limits: {
   
   
    fileSize: 5 * 1024 * 1024, // 5MB
  },
};

5. 实现文件服务

src/modules/file/file.service.ts 中实现文件服务:

import {
   
    Injectable } from '@nestjs/common';
import {
   
    ConfigService } from '@nestjs/config';
import {
   
    join } from 'path';
import {
   
    existsSync, mkdirSync } from 'fs';

@Injectable()
export class FileService {
   
   
  constructor(private configService: ConfigService) {
   
   
    const uploadPath = join(process.cwd(), 'uploads');
    if (!existsSync(uploadPath)) {
   
   
      mkdirSync(uploadPath);
    }
  }

  getFileUrl(filename: string): string {
   
   
    const baseUrl = this.configService.get('BASE_URL'
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值