54 lines
1.2 KiB
TypeScript
54 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
|
||
|
|
}
|
||
|
|
}
|