Merge branch 'refs/heads/abp' into digital-collectibles

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

@@ -0,0 +1,116 @@
+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;
+            }
+        }
+    }
+}

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

@@ -0,0 +1,6 @@
+namespace Yi.Framework.Rbac.Domain.Managers;
+
+public interface IFileManager
+{
+    
+}