nestjs学习day5

文件上传

npm i multer -S

npm i @types/multer -D

nest g res upload

npm i compressing -S

upload模块代码

import { Module } from '@nestjs/common';
import { UploadService } from './upload.service';
import { UploadController } from './upload.controller';
import {MulterModule} from '@nestjs/platform-express'
import {diskStorage} from 'multer'
import  {extname,join} from 'path'

@Module({
  imports:[MulterModule.register({
    storage:diskStorage({
      destination:join(__dirname,'../image'),
      filename:(_,file,callback)=>{
          const fileName=`${new Date().getTime()+extname(file.originalname)}`
          return callback(null,fileName)
      }
    })
  })],
  controllers: [UploadController],
  providers: [UploadService]
})
export class UploadModule {}

import {Controller, Get, Post, Body, Patch, Param, Delete, UseInterceptors, UploadedFile, Res} from '@nestjs/common';
import { UploadService } from './upload.service';
import {FileInterceptor,FilesInterceptor} from "@nestjs/platform-express";
import {Response} from "express";
import {join} from "path";
import {zip} from 'compressing'

@Controller('upload')
export class UploadController {
  constructor(private readonly uploadService: UploadService) {}

  @Post('album')
  @UseInterceptors(FileInterceptor('file'))
  upload(@UploadedFile() file) {
    console.log('file',file)
    return 'yes'
  }

  @Get('export')
  download(@Res() res:Response){
    const url=join(__dirname,'../image/1699680481692.png')
    res.download(url)
  }

  @Get('stream')
  async down(@Res() res:Response){
    const url=join(__dirname,'../image/1699680481692.png')
    const tarStream = new zip.Stream()
    await tarStream.addEntry(url)
    res.setHeader('Content-Type','application/octet-stream')
    res.setHeader(
        'Content-Disposition',
        `attachment;filename=xiaoman`,
    )
    tarStream.pipe(res)
  }
}

main.js配置

import {NestFactory} from '@nestjs/core';
import {AppModule} from './app.module';
import {VersioningType} from "@nestjs/common";
import * as session from 'express-session'
import {NestExpressApplication} from '@nestjs/platform-express'
import {Request,Response,NextFunction} from 'express'
import {join} from "path";
async function bootstrap() {
  const app = await NestFactory.create<NestExpressApplication>(AppModule);
  app.useStaticAssets(join(__dirname,'image'),{
    prefix:'/xiaoman'
  })
  app.enableVersioning({
    type:VersioningType.URI
  })
  app.use(session({secret:"xiaoman",rolling:true,
    name:"xiaoman.sid",cookie:{maxAge:99999}}))
  await app.listen(3000);
}
bootstrap();

前端流下载接口配置

const useFetch=async (urls:string)=>{
  const res= await fetch(urls).then(res=>res.arrayBuffer())
  const blob=new Blob([res])
  const url=URL.createObjectURL(blob)
  const a= document.createElement('a')
  a.href=url
  a.download='xiaomi.zip'
  a.click()

}
const down=()=>{
  useFetch('/api/upload/steam')
}