53 lines
1.2 KiB
TypeScript
53 lines
1.2 KiB
TypeScript
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): Promise<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);
|
|
|
|
await this.storageRepository.save({
|
|
name,
|
|
fileName,
|
|
extName,
|
|
path,
|
|
type,
|
|
size,
|
|
userId,
|
|
});
|
|
|
|
return path;
|
|
}
|
|
}
|