SysFileService.cs 8.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214
  1. using Infrastructure;
  2. using Infrastructure.Attribute;
  3. using Infrastructure.Enums;
  4. using Infrastructure.Model;
  5. using Microsoft.AspNetCore.Http;
  6. using Microsoft.Extensions.Options;
  7. using SixLabors.ImageSharp;
  8. using SixLabors.ImageSharp.Formats.Jpeg;
  9. using System.Net;
  10. using System.Security.Cryptography;
  11. using System.Text;
  12. using ZR.Common;
  13. using ZR.Model.Dto;
  14. using ZR.Model.System;
  15. namespace ZR.ServiceCore.Services
  16. {
  17. /// <summary>
  18. /// 文件管理
  19. /// </summary>
  20. [AppService(ServiceType = typeof(ISysFileService), ServiceLifetime = LifeTime.Transient)]
  21. public class SysFileService : BaseService<SysFile>, ISysFileService
  22. {
  23. private string domainUrl = AppSettings.GetConfig("ALIYUN_OSS:domainUrl");
  24. private readonly ISysConfigService SysConfigService;
  25. private OptionsSetting OptionsSetting;
  26. public SysFileService(ISysConfigService sysConfigService, IOptions<OptionsSetting> options)
  27. {
  28. SysConfigService = sysConfigService;
  29. OptionsSetting = options.Value;
  30. }
  31. /// <summary>
  32. /// 存储本地
  33. /// </summary>
  34. /// <param name="rootPath">存储根目录</param>
  35. /// <param name="formFile">上传的文件流</param>
  36. /// <param name="clasifyType">分类类型</param>
  37. /// <param name="userName"></param>
  38. /// <param name="dto"></param>
  39. /// <returns></returns>
  40. public async Task<SysFile> SaveFileToLocal(string rootPath, UploadDto dto, string userName, string clasifyType, IFormFile formFile)
  41. {
  42. var fileName = dto.FileName;
  43. var fileDir = dto.FileDir;
  44. string fileExt = Path.GetExtension(formFile.FileName);
  45. fileName = (fileName.IsEmpty() ? HashFileName() : fileName) + fileExt;
  46. string filePath = GetdirPath(fileDir);
  47. string finalFilePath = Path.Combine(rootPath, filePath, fileName);
  48. double fileSize = Math.Round(formFile.Length / 1024.0, 2);
  49. if (!Directory.Exists(Path.GetDirectoryName(finalFilePath)))
  50. {
  51. Directory.CreateDirectory(Path.GetDirectoryName(finalFilePath));
  52. }
  53. // 常见的图片扩展名
  54. var imageExtensions = new[] { ".jpg", ".jpeg", ".png", ".gif", ".bmp" };
  55. // 检查扩展名是否在图片扩展名列表中
  56. bool isImageByExtension = imageExtensions.Contains(fileExt);
  57. if (dto.Quality > 0 && isImageByExtension)
  58. {
  59. await SaveCompressedImageAsync(formFile, finalFilePath);
  60. }
  61. else
  62. {
  63. using (var stream = new FileStream(finalFilePath, FileMode.Create))
  64. {
  65. await formFile.CopyToAsync(stream);
  66. }
  67. }
  68. string uploadUrl = OptionsSetting.Upload.UploadUrl;
  69. string accessPath = string.Concat(filePath.Replace("\\", "/"), "/", fileName);
  70. Uri baseUri = new(uploadUrl);
  71. Uri fullUrl = new(baseUri, accessPath);
  72. SysFile file = new(formFile.FileName, fileName, fileExt, fileSize + "kb", filePath, userName)
  73. {
  74. StoreType = (int)StoreType.LOCAL,
  75. FileType = formFile.ContentType,
  76. FileUrl = finalFilePath.Replace("\\", "/"),
  77. AccessUrl = fullUrl.AbsoluteUri,
  78. ClassifyType = clasifyType,
  79. CategoryId = dto.CategoryId,
  80. };
  81. file.Id = await InsertFile(file);
  82. return file;
  83. }
  84. //public async Task<SysFile> SaveFileToLocal(string rootPath, string fileName, string fileDir, string userName, IFormFile formFile)
  85. //{
  86. // return await SaveFileToLocal(rootPath, fileName, fileDir, userName, string.Empty, formFile);
  87. //}
  88. public async Task<SysFile> SaveFileToLocal(string rootPath, UploadDto dto, string userName, IFormFile formFile)
  89. {
  90. return await SaveFileToLocal(rootPath, dto, userName, dto.ClassifyType, formFile);
  91. }
  92. /// <summary>
  93. /// 上传文件到阿里云
  94. /// </summary>
  95. /// <param name="file"></param>
  96. /// <param name="dto"></param>
  97. /// <param name="formFile"></param>
  98. /// <returns></returns>
  99. public async Task<SysFile> SaveFileToAliyun(SysFile file, UploadDto dto, IFormFile formFile)
  100. {
  101. file.FileName = (file.FileName.IsEmpty() ? HashFileName() : file.FileName) + file.FileExt;
  102. file.StorePath = GetdirPath(file.StorePath);
  103. string finalPath = Path.Combine(file.StorePath, file.FileName);
  104. HttpStatusCode statusCode;
  105. if (dto.Quality > 0)
  106. {
  107. // 压缩图片
  108. using var stream = new MemoryStream();
  109. await CompressImageAsync(formFile, stream, dto.Quality);
  110. stream.Position = 0;
  111. statusCode = AliyunOssHelper.PutObjectFromFile(stream, finalPath, "");
  112. }
  113. else
  114. {
  115. statusCode = AliyunOssHelper.PutObjectFromFile(formFile.OpenReadStream(), finalPath, "");
  116. }
  117. if (statusCode != HttpStatusCode.OK) return file;
  118. file.StorePath = file.StorePath;
  119. file.FileUrl = finalPath;
  120. file.AccessUrl = string.Concat(domainUrl, "/", file.StorePath.Replace("\\", "/"), "/", file.FileName);
  121. file.Id = await InsertFile(file);
  122. return file;
  123. }
  124. /// <summary>
  125. /// 获取文件存储目录
  126. /// </summary>
  127. /// <param name="storePath"></param>
  128. /// <param name="byTimeStore">是否按年月日存储</param>
  129. /// <returns></returns>
  130. public string GetdirPath(string storePath = "", bool byTimeStore = true)
  131. {
  132. DateTime date = DateTime.Now;
  133. string timeDir = date.ToString("yyyy/MMdd");
  134. if (!string.IsNullOrEmpty(storePath))
  135. {
  136. timeDir = Path.Combine(storePath, timeDir);
  137. }
  138. Console.WriteLine("文件存储目录" + timeDir);
  139. return timeDir.Replace("\\", "/");
  140. }
  141. public string HashFileName(string str = null)
  142. {
  143. if (string.IsNullOrEmpty(str))
  144. {
  145. str = Guid.NewGuid().ToString().ToLower();
  146. }
  147. return BitConverter.ToString(MD5.HashData(Encoding.Default.GetBytes(str)), 4, 8).Replace("-", "").ToLower();
  148. }
  149. public Task<long> InsertFile(SysFile file)
  150. {
  151. return Insertable(file).ExecuteReturnSnowflakeIdAsync();//单条插入返回雪花ID;
  152. }
  153. /// <summary>
  154. /// 修改文件存储表
  155. /// </summary>
  156. /// <param name="model"></param>
  157. /// <returns></returns>
  158. public int UpdateFile(SysFile model)
  159. {
  160. return Update(model, t => new { t.ClassifyType, }, true);
  161. }
  162. /// <summary>
  163. /// 压缩图片
  164. /// </summary>
  165. /// <param name="formFile"></param>
  166. /// <param name="finalFilePath"></param>
  167. /// <param name="quality"></param>
  168. /// <returns></returns>
  169. public static async Task SaveCompressedImageAsync(IFormFile formFile, string finalFilePath, int quality = 75)
  170. {
  171. using (var image = await Image.LoadAsync(formFile.OpenReadStream()))
  172. {
  173. // 进行压缩和调整大小(可选)
  174. //image.Mutate(x => x.Resize(new ResizeOptions
  175. //{
  176. // Mode = ResizeMode.Max,
  177. // Size = new Size(1920, 1080) // 限制最大尺寸,避免超大图片
  178. //}));
  179. // 保存为压缩的 JPEG
  180. var encoder = new JpegEncoder { Quality = quality }; // 质量参数控制压缩程度
  181. await using (var stream = new FileStream(finalFilePath, FileMode.Create))
  182. {
  183. await image.SaveAsync(stream, encoder);
  184. }
  185. }
  186. }
  187. private async Task CompressImageAsync(IFormFile file, Stream outputStream, int quality)
  188. {
  189. // 加载图片
  190. using var image = await Image.LoadAsync(file.OpenReadStream());
  191. // 设置 JPEG 编码器并指定质量
  192. var encoder = new JpegEncoder { Quality = quality };
  193. // 保存压缩后的图片到输出流
  194. await image.SaveAsync(outputStream, encoder);
  195. }
  196. }
  197. }