Yi.Admin/Yi.Abp.Net8/module/rbac/Yi.Framework.Rbac.Domain/Managers/FileManager.cs

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)
    {
        var formFiles = files as IFormFile[] ?? files.ToArray();
        if (!formFiles.Any())
        {
            throw new ArgumentException("文件上传为空！");
        }

        //批量插入
        List<FileAggregateRoot> entities = new();
        foreach (var file in formFiles)
        {
            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)
            {

                this.LoggerFactory.CreateLogger<FileManager>().LogError(exception, exception.Message);
            }
            finally
            {
                //如果失败了，直接复制一份到缩略图上即可
                compressImageStream = fileStream;
            }

          
            using (var stream = new FileStream(thumbnailSavePath, FileMode.CreateNew, FileAccess.ReadWrite))
            {
                await compressImageStream.CopyToAsync(stream);
                compressImageStream.Position = 0;
            }
        }
    }
}