oa_based/src/modules/tools/upload/upload.service.ts

import { MultipartFile } from '@fastify/multipart';
import { Injectable, NotFoundException } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { isNil } from 'lodash';
import { Repository } from 'typeorm';

import { Storage } from '~/modules/tools/storage/storage.entity';

import {
  fileRename,
  getExtname,
  getFilePath,
  getFileType,
  getSize,
  saveLocalFile
} from '~/utils/file.util';

@Injectable()
export class UploadService {
  constructor(
    @InjectRepository(Storage)
    private storageRepository: Repository<Storage>
  ) {}

  /**
   * 保存文件上传记录
   */
  async saveFile(
    file: MultipartFile,
    userId: number,
    bussinessModule?: string,
    bussinessRecordId?: number
  ): Promise<{ id: number; path: string }> {
    if (isNil(file)) throw new NotFoundException('Have not any file to upload!');

    const fileName = file.filename;
    const size = getSize(file.file.bytesRead);
    const extName = getExtname(fileName);
    const type = getFileType(extName);
    const name = fileRename(fileName);
    const path = getFilePath(name);

    saveLocalFile(await file.toBuffer(), name);

    const storage = await this.storageRepository.save({
      name,
      fileName,
      extName,
      path,
      type,
      size,
      userId,
      bussinessModule,
      bussinessRecordId
    });

    return { path, id: storage.id };
  }
}