116 lines
4.1 KiB
C#
116 lines
4.1 KiB
C#
using Mapster;
|
||
using Microsoft.AspNetCore.Http;
|
||
using Microsoft.Extensions.Logging;
|
||
using Volo.Abp.Domain.Repositories;
|
||
using Volo.Abp.Domain.Services;
|
||
using Volo.Abp.Guids;
|
||
using Volo.Abp.Imaging;
|
||
using Yi.Framework.Core.Enums;
|
||
using Yi.Framework.Core.Helper;
|
||
using Yi.Framework.Rbac.Domain.Entities;
|
||
|
||
namespace Yi.Framework.Rbac.Domain.Managers;
|
||
|
||
public class FileManager : DomainService, IFileManager
|
||
{
|
||
private IGuidGenerator _guidGenerator;
|
||
private readonly IRepository<FileAggregateRoot> _repository;
|
||
private readonly IImageCompressor _imageCompressor;
|
||
public FileManager(IGuidGenerator guidGenerator, IRepository<FileAggregateRoot> repository,
|
||
|
||
IImageCompressor imageCompressor)
|
||
{
|
||
_guidGenerator = guidGenerator;
|
||
_repository = repository;
|
||
_imageCompressor = imageCompressor;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 批量插入数据库
|
||
/// </summary>
|
||
/// <param name="files"></param>
|
||
/// <exception cref="ArgumentException"></exception>
|
||
public async Task<List<FileAggregateRoot>> CreateAsync(IEnumerable<IFormFile> files)
|
||
{
|
||
if (files.Count() == 0)
|
||
{
|
||
throw new ArgumentException("文件上传为空！");
|
||
}
|
||
|
||
//批量插入
|
||
List<FileAggregateRoot> entities = new();
|
||
foreach (var file in files)
|
||
{
|
||
FileAggregateRoot data = new(_guidGenerator.Create(), file.FileName, (decimal)file.Length / 1024);
|
||
data.CheckDirectoryOrCreate();
|
||
entities.Add(data);
|
||
}
|
||
|
||
await _repository.InsertManyAsync(entities);
|
||
return entities;
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// 保存文件
|
||
/// </summary>
|
||
/// <param name="file"></param>
|
||
/// <param name="fileStream"></param>
|
||
public async Task SaveFileAsync(FileAggregateRoot file, Stream fileStream)
|
||
{
|
||
var filePath = file.GetSaveFilePath();
|
||
|
||
//生成文件
|
||
using (var stream = new FileStream(filePath, FileMode.CreateNew, FileAccess.ReadWrite))
|
||
{
|
||
await fileStream.CopyToAsync(stream);
|
||
fileStream.Position = 0;
|
||
}
|
||
|
||
|
||
//如果是图片类型，还需要生成缩略图
|
||
//这里根据自己需求变更，我们的需求是：原始文件与缩略图文件，都要一份
|
||
var fileType = file.GetFileType();
|
||
//如果文件类型是图片，尝试进行压缩
|
||
if (FileTypeEnum.image==fileType)
|
||
{
|
||
var thumbnailSavePath = file.GetAndCheakThumbnailSavePath(true);
|
||
Stream compressImageStream=null;
|
||
try
|
||
{
|
||
//压缩图片
|
||
var compressResult = await _imageCompressor.CompressAsync(fileStream, file.GetMimeMapping());
|
||
if (compressResult.State == ImageProcessState.Done)
|
||
{
|
||
compressImageStream =
|
||
(await _imageCompressor.CompressAsync(fileStream, file.GetMimeMapping())).Result;
|
||
}
|
||
else if (compressResult.State == ImageProcessState.Canceled)
|
||
{
|
||
throw new NotSupportedException($"当前图片无法再进行压缩,文件id：{file.Id}");
|
||
}
|
||
else
|
||
{
|
||
throw new NotSupportedException($"当前图片不支持压缩,文件id：{file.Id}");
|
||
}
|
||
}
|
||
catch (Exception exception) when (exception is NotSupportedException)
|
||
{
|
||
this.LoggerFactory.CreateLogger<FileManager>().LogInformation(exception, exception.Message);
|
||
}
|
||
catch (Exception exception)
|
||
{
|
||
//如果失败了，直接复制一份到缩略图上即可
|
||
compressImageStream = fileStream;
|
||
this.LoggerFactory.CreateLogger<FileManager>().LogError(exception, exception.Message);
|
||
}
|
||
|
||
|
||
using (var stream = new FileStream(thumbnailSavePath, FileMode.CreateNew, FileAccess.ReadWrite))
|
||
{
|
||
await compressImageStream.CopyToAsync(stream);
|
||
compressImageStream.Position = 0;
|
||
}
|
||
}
|
||
}
|
||
} |