FileTransferService.cs 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780
  1. using System.Collections.Concurrent;
  2. using WingServerCommon.Log;
  3. using System.Net;
  4. using System.Net.Http.Headers;
  5. using System.Security.Cryptography;
  6. using System.Text;
  7. using WingServerCommon.Config;
  8. using WingServerCommon.Config.Parameters;
  9. namespace WingServerCommon.Interfaces.FileTransfer
  10. {
  11. /// <summary>
  12. /// 文件传输服务
  13. /// </summary>
  14. public class FileTransferService
  15. {
  16. private readonly HttpClient _httpClient;
  17. private readonly UploadFile _uploadFile;
  18. private string _folderBase = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "FileTransferRecord");
  19. private int _autoSliceSizeForUpload = 0;
  20. private long _manualDivisionForUpload = 0;
  21. /// <summary>
  22. /// Init service
  23. /// </summary>
  24. public FileTransferService()
  25. {
  26. _httpClient = new HttpClient();
  27. _httpClient.Timeout = TimeSpan.FromMinutes(100);
  28. _uploadFile = new UploadFile(_httpClient);
  29. }
  30. /// <summary>
  31. /// copy文件到其他节点
  32. /// </summary>
  33. /// <param name="fileUrl">源文件地址,例如北京节点文件地址,需要源文件而不是cdn</param>
  34. /// <param name="fileName">文件名</param>
  35. /// <returns>是否存在</returns>
  36. public async Task<List<string>> CopyToOtherNode(string fileUrl, string fileName)
  37. {
  38. var res = await _uploadFile.DoCopyFileAsync(fileUrl, fileName);
  39. return res;
  40. }
  41. /// <summary>
  42. /// 请求验证文件是否存在
  43. /// </summary>
  44. /// <param name="oriFileName">FileName</param>
  45. /// <param name="fileSize">文件大小</param>
  46. /// <returns>是否存在</returns>
  47. public async Task<bool> CheckFileIsExist(string oriFileName, long fileSize, bool isNeedPartUpload = true, bool isUpgradePackage = false)
  48. {
  49. bool result = true;
  50. if (!isNeedPartUpload)
  51. {
  52. var newFirstName = Path.GetFileNameWithoutExtension(oriFileName);
  53. var newFileExtension = Path.GetExtension(oriFileName);
  54. var newFileName = newFirstName + newFileExtension;
  55. var newAuthTuple = SignatureHelper.GetAuthorizationAsync(newFileName, false, "head", null, null, false, isUpgradePackage);
  56. if (newAuthTuple == null)
  57. {
  58. Logger.WriteLineInfo("Head Create Signature Fail");
  59. return false;
  60. }
  61. var newRequestHeads = new Dictionary<string, string>();
  62. newRequestHeads.Add("Authorization", newAuthTuple.Item1);
  63. var fileUrl = newAuthTuple.Item2;
  64. using (var newRequest = new HttpRequestMessage())
  65. {
  66. newRequest.RequestUri = new Uri(fileUrl);
  67. newRequest.Method = HttpMethod.Head;
  68. foreach (var newHead in newRequestHeads)
  69. {
  70. newRequest.Headers.TryAddWithoutValidation(newHead.Key, newHead.Value);
  71. }
  72. var newResponse = await _httpClient.SendAsync(newRequest);
  73. if (newResponse != null && newResponse.StatusCode == HttpStatusCode.OK)
  74. {
  75. //read response content
  76. if (fileUrl.Contains("/SystemUpgradePackage/"))//表示vinno
  77. {
  78. var resultContent = newResponse.Content.Headers.ContentLength > 0;
  79. if (resultContent)
  80. {
  81. result = result & true;
  82. }
  83. else
  84. {
  85. result = result & false;
  86. }
  87. }
  88. else
  89. {
  90. var newETag = newResponse.Headers?.ETag?.Tag ?? string.Empty;;
  91. if (!string.IsNullOrEmpty(newETag))
  92. {
  93. result = result & true;
  94. }
  95. else
  96. {
  97. result = result & false;
  98. }
  99. }
  100. }
  101. else
  102. {
  103. result = result & false;
  104. }
  105. }
  106. return result;
  107. }
  108. if (_manualDivisionForUpload <= 0)
  109. {
  110. _manualDivisionForUpload = EnvironmentConfigManager.GetParammeter<IntParameter>("Storage", "ManualDivisionForUpload").Value;
  111. if (_manualDivisionForUpload <= 0)
  112. {
  113. _manualDivisionForUpload = 200 * 1024 * 1024;
  114. }
  115. }
  116. var partCount = (int)((fileSize / _manualDivisionForUpload) + (fileSize % _manualDivisionForUpload > 0 ? 1 : 0));
  117. if (partCount > 0)
  118. {
  119. var firstName = Path.GetFileNameWithoutExtension(oriFileName);
  120. var fileExtension = Path.GetExtension(oriFileName);
  121. for (var i = 0; i < partCount; i++)
  122. {
  123. var fileName = "";
  124. if (partCount == 1)
  125. {
  126. fileName = firstName + fileExtension;
  127. }
  128. else
  129. {
  130. fileName = firstName + "_" + i + fileExtension;
  131. }
  132. var authTuple = SignatureHelper.GetAuthorizationAsync(fileName, false, "head", null, null, false, isUpgradePackage);
  133. if (authTuple == null)
  134. {
  135. Logger.WriteLineInfo("Head Create Signature Fail");
  136. return false;
  137. }
  138. var requestHeads = new Dictionary<string, string>();
  139. requestHeads.Add("Authorization", authTuple.Item1);
  140. var fileUrl = authTuple.Item2;
  141. using (var request = new HttpRequestMessage())
  142. {
  143. request.RequestUri = new Uri(fileUrl);
  144. request.Method = HttpMethod.Head;
  145. foreach (var head in requestHeads)
  146. {
  147. request.Headers.TryAddWithoutValidation(head.Key, head.Value);
  148. }
  149. var response = await _httpClient.SendAsync(request);
  150. if (response != null && response.StatusCode == HttpStatusCode.OK)
  151. {
  152. //read response content
  153. if (fileUrl.Contains("/SystemUpgradePackage/"))//表示vinno
  154. {
  155. var resultContent = response.Content.Headers.ContentLength > 0;
  156. if (resultContent)
  157. {
  158. result = result & true;
  159. }
  160. else
  161. {
  162. result = result & false;
  163. }
  164. }
  165. else
  166. {
  167. var eTag = response.Headers?.ETag?.Tag ?? string.Empty;;
  168. if (!string.IsNullOrEmpty(eTag))
  169. {
  170. result = result & true;
  171. }
  172. else
  173. {
  174. result = result & false;
  175. break;
  176. }
  177. }
  178. }
  179. else
  180. {
  181. result = result & false;
  182. break;
  183. }
  184. }
  185. }
  186. }
  187. return result;
  188. }
  189. /// <summary>
  190. /// 分块传输上传文件
  191. /// </summary>
  192. /// <param name="sourceFilePath">源文件地址 </param>
  193. /// <param name="isRechristen">Rename</param>
  194. /// <param name="isCopy">is Copy to OtherNode</param>
  195. /// <returns>FileUrl List</returns>
  196. public async Task<List<string>> FileTransferAsync(string sourceFilePath, bool isRechristen = true, bool isNeedPartUpload = true, bool isCopy = false)
  197. {
  198. var returnFileUrlList = new List<string>();
  199. FileTransferRecorder record = new FileTransferRecorder();
  200. try
  201. {
  202. if (string.IsNullOrEmpty(sourceFilePath) || !File.Exists(sourceFilePath))
  203. {
  204. Logger.WriteLineWarn("File Absolute Path Error!");
  205. return new List<string>();
  206. }
  207. //读取缓存文件, 如果存在就是需要断点upload
  208. var recorderFilePath = Path.Combine(_folderBase, ToMD5(sourceFilePath) + ".json");
  209. if (File.Exists(recorderFilePath))
  210. {
  211. //调用断点续传
  212. return await FileTransferAgainAsync(sourceFilePath, recorderFilePath);
  213. }
  214. string oriFilename = Path.GetFileNameWithoutExtension(sourceFilePath);
  215. var factName = oriFilename;
  216. var factNameExtension = Path.GetExtension(sourceFilePath);
  217. if (string.IsNullOrEmpty(factNameExtension))
  218. {
  219. factNameExtension = ".dat";
  220. }
  221. if (isRechristen)
  222. {
  223. factName = Guid.NewGuid().ToString("N").ToUpper();
  224. }
  225. if (_manualDivisionForUpload <= 0)
  226. {
  227. _manualDivisionForUpload = EnvironmentConfigManager.GetParammeter<IntParameter>("Storage", "ManualDivisionForUpload").Value;
  228. if (_manualDivisionForUpload <= 0)
  229. {
  230. _manualDivisionForUpload = 200 * 1024 * 1024;
  231. }
  232. }
  233. if (!isNeedPartUpload)
  234. {
  235. var fileInfo = new System.IO.FileInfo(sourceFilePath);
  236. if (fileInfo.Length >= 5000000000)
  237. {
  238. Logger.WriteLineWarn($"CommonServer FileTransferAsync isNeedPartUpload Error,sourceFilePath: { sourceFilePath }, FileSize: { fileInfo.Length }!");
  239. return new List<string>();
  240. }
  241. }
  242. //读取文件 ConcurrentDictionary
  243. var dic = new Dictionary<int, byte[]>();
  244. dic = ReadFileData(sourceFilePath, _manualDivisionForUpload);
  245. if (dic == null || dic.Count <= 0)
  246. {
  247. Logger.WriteLineWarn("File Data Empty Error!");
  248. return new List<string>();
  249. }
  250. //记录文件信息
  251. record.FilePath = sourceFilePath;
  252. record.FileName = oriFilename;
  253. record.ObjectName = factName;
  254. record.FileExtension = factNameExtension;
  255. record.SliceSizeForUpload = _manualDivisionForUpload;
  256. record.PartList = new List<FileTransferRecorderDetail>();
  257. record.IsNeedPartUpload = isNeedPartUpload;
  258. var partNums = dic.Keys.OrderBy(c => c).ToList();
  259. foreach(var i in partNums)
  260. {
  261. var partDetail = new FileTransferRecorderDetail();
  262. if (partNums.Count == 1)
  263. {
  264. partDetail.PartFileName = factName + factNameExtension;
  265. }
  266. else
  267. {
  268. partDetail.PartFileName = factName + "_" + i + factNameExtension;
  269. }
  270. partDetail.PartNumber = i;
  271. partDetail.IsUploadComplete = false;
  272. partDetail.PartFileSize = dic[i].Length;
  273. record.PartList.Add(partDetail);
  274. record.FileSize += partDetail.PartFileSize;
  275. }
  276. await SaveTransferRecord(record);
  277. if (record?.PartList?.Count > 0)
  278. {
  279. try
  280. {
  281. if (isNeedPartUpload)
  282. {
  283. //todo 可尝试优化为Multi Thread
  284. foreach (var partDetail in record.PartList)
  285. {
  286. var fileUrl = await _uploadFile.DoUploadFileAsync(partDetail.PartFileName, dic[partDetail.PartNumber], false, factNameExtension);
  287. if (!string.IsNullOrEmpty(fileUrl))
  288. {
  289. record.Host = fileUrl.Replace(partDetail.PartFileName, "");
  290. partDetail.IsUploadComplete = true;
  291. if (isCopy)
  292. {
  293. //后续可以改成异步
  294. await _uploadFile.DoCopyFileAsync(fileUrl, partDetail.PartFileName);
  295. }
  296. }
  297. await SaveTransferRecord(record);
  298. }
  299. }
  300. else
  301. {
  302. var fileUrl = await _uploadFile.ManualMultiPartUploadFileAsync(record, dic, false);
  303. if (isCopy)
  304. {
  305. //后续可以改成异步
  306. string fileName = record.ObjectName + record.FileExtension;
  307. await _uploadFile.DoCopyFileAsync(fileUrl, fileName);
  308. }
  309. await SaveTransferRecord(record);
  310. returnFileUrlList = new List<string>() { fileUrl };
  311. }
  312. }
  313. catch (Exception e)
  314. {
  315. await SaveTransferRecord(record);
  316. //文件上传错误
  317. Logger.WriteLineWarn($"File Upload Error, {e}");
  318. }
  319. if (record.PartList.Count(c => c.IsUploadComplete) == record.PartList.Count)
  320. {
  321. record.IsUploadComplete = true;
  322. }
  323. else
  324. {
  325. record.IsUploadComplete = false;
  326. }
  327. }
  328. if (record.IsUploadComplete)
  329. {
  330. //删除缓存记录
  331. if (File.Exists(recorderFilePath))
  332. {
  333. //存在删除
  334. File.Delete(recorderFilePath);
  335. }
  336. //返回多个链接列表
  337. if (isNeedPartUpload)
  338. {
  339. returnFileUrlList = record.PartList.OrderBy(c => c.PartFileName).Select(c => record.Host + c.PartFileName).ToList();
  340. }
  341. }
  342. }
  343. catch (Exception ex)
  344. {
  345. Logger.WriteLineError($"File Transfer Fail, {ex}");
  346. }
  347. finally
  348. {
  349. GC.Collect();
  350. }
  351. return returnFileUrlList;
  352. }
  353. /// <summary>
  354. /// 保存上传记录文件
  355. /// </summary>
  356. /// <param name="request">请求对象</param>
  357. /// <returns>是否成功</returns>
  358. private async Task<bool> SaveTransferRecord(FileTransferRecorder record)
  359. {
  360. if (!Directory.Exists(_folderBase))
  361. {
  362. Directory.CreateDirectory(_folderBase);
  363. }
  364. var recorderFilePath = Path.Combine(_folderBase, ToMD5(record.FilePath) + ".json");
  365. if (File.Exists(recorderFilePath))
  366. {
  367. //存在删除
  368. File.Delete(recorderFilePath);
  369. }
  370. var json = Newtonsoft.Json.JsonConvert.SerializeObject(record);
  371. if (!string.IsNullOrWhiteSpace(json))
  372. {
  373. byte[] fileData = System.Text.Encoding.UTF8.GetBytes(json);
  374. //存储文件
  375. using (var fw = File.Open(recorderFilePath, FileMode.Create, FileAccess.ReadWrite, FileShare.ReadWrite))
  376. {
  377. fw.Write(fileData, 0, fileData.Length);
  378. }
  379. }
  380. return true;
  381. }
  382. /// <summary>
  383. /// 文件续传
  384. /// </summary>
  385. /// <param name="sourceFilePath">源文件地址</param>
  386. /// <returns>Part FileUrl List</returns>
  387. public async Task<List<string>> FileTransferAgainAsync(string sourceFilePath, string recorderFilePath)
  388. {
  389. var returnFileUrlList = new List<string>();
  390. FileTransferRecorder record = new FileTransferRecorder();
  391. try
  392. {
  393. var jsonContent = File.ReadAllText(recorderFilePath);
  394. if (!string.IsNullOrEmpty(jsonContent))
  395. {
  396. record = Newtonsoft.Json.JsonConvert.DeserializeObject<FileTransferRecorder>(jsonContent);
  397. }
  398. //读取文件
  399. if (record?.PartList?.Count > 0)
  400. {
  401. var waitUploadFileList = record.PartList.FindAll(c => c.IsUploadComplete == false);
  402. var dic = new Dictionary<int, byte[]>();
  403. dic = ReadFileData(sourceFilePath, record.SliceSizeForUpload);
  404. if (dic == null || dic.Count <= 0)
  405. {
  406. Logger.WriteLineWarn("File Data Empty Error!");
  407. return new List<string>();
  408. }
  409. var isNeedPartUpload = record.IsNeedPartUpload;
  410. try
  411. {
  412. if (isNeedPartUpload)
  413. {
  414. foreach (var partDetail in waitUploadFileList)
  415. {
  416. var fileUrl = await _uploadFile.DoUploadFileAsync(partDetail.PartFileName, dic[partDetail.PartNumber], false, record.FileExtension);
  417. if (!string.IsNullOrEmpty(fileUrl))
  418. {
  419. record.Host = fileUrl.Replace(partDetail.PartFileName, "");
  420. partDetail.IsUploadComplete = true;
  421. if (record.IsCopy)
  422. {
  423. //后续可以改成异步
  424. await _uploadFile.DoCopyFileAsync(fileUrl, partDetail.PartFileName);
  425. }
  426. }
  427. await SaveTransferRecord(record);
  428. }
  429. }
  430. else
  431. {
  432. var fileUrl = await _uploadFile.ManualMultiPartUploadFileAsync(record, dic, false);
  433. if (record.IsCopy)
  434. {
  435. //后续可以改成异步
  436. string fileName = record.ObjectName + record.FileExtension;
  437. await _uploadFile.DoCopyFileAsync(fileUrl, fileName);
  438. }
  439. await SaveTransferRecord(record);
  440. returnFileUrlList = new List<string>() { fileUrl };
  441. }
  442. }
  443. catch (Exception e)
  444. {
  445. await SaveTransferRecord(record);
  446. //文件上传错误
  447. Logger.WriteLineWarn($"File Upload Again Error, {e}");
  448. }
  449. if (record.PartList.Count(c => c.IsUploadComplete) == record.PartList.Count)
  450. {
  451. record.IsUploadComplete = true;
  452. }
  453. else
  454. {
  455. record.IsUploadComplete = false;
  456. }
  457. if (record.IsUploadComplete)
  458. {
  459. //删除缓存记录
  460. if (File.Exists(recorderFilePath))
  461. {
  462. //存在删除
  463. File.Delete(recorderFilePath);
  464. }
  465. //返回多个链接列表;
  466. if (isNeedPartUpload)
  467. {
  468. returnFileUrlList = record.PartList.OrderBy(c => c.PartFileName).Select(c => record.Host + c.PartFileName).ToList();
  469. }
  470. }
  471. }
  472. }
  473. catch (Exception ex)
  474. {
  475. Logger.WriteLineError($"File Transfer Again Fail, {ex}");
  476. }
  477. finally
  478. {
  479. GC.Collect();
  480. }
  481. return returnFileUrlList;
  482. }
  483. /// <summary>
  484. /// 合并文件并保存到指定目录
  485. /// </summary>
  486. /// <param name="partFileUrlList">分片地址列表</param>
  487. /// <param name="savePath">保存指定文件地址(不包含文件名)</param>
  488. /// <param name="saveFileName">保存指定文件(无需带扩展名)</param>
  489. /// <returns>保存地址文件</returns>
  490. public async Task<string> DownloadFileAsync(List<string> partFileUrlList, string savePath, string saveFileName = "")
  491. {
  492. var newSaveFileName = "";
  493. try
  494. {
  495. if (string.IsNullOrEmpty(savePath))
  496. {
  497. Logger.WriteLineWarn("Save Path Error!");
  498. return "";
  499. }
  500. if (partFileUrlList == null || partFileUrlList.Count <= 0)
  501. {
  502. Logger.WriteLineWarn("Download File List Error!");
  503. return "";
  504. }
  505. if (_autoSliceSizeForUpload <= 0)
  506. {
  507. _autoSliceSizeForUpload = EnvironmentConfigManager.GetParammeter<IntParameter>("Storage", "AutoSliceSizeForUpload").Value;
  508. }
  509. var fileExtension = "";
  510. if (partFileUrlList[0].Split('.').Length > 0)
  511. {
  512. fileExtension = partFileUrlList[0].Split('.')[(partFileUrlList[0].Split('.').Length - 1)];
  513. }
  514. else
  515. {
  516. fileExtension = "dat";
  517. }
  518. //验证文件是否存在
  519. var fileName = saveFileName;
  520. if (string.IsNullOrEmpty(fileName))
  521. {
  522. fileName = Guid.NewGuid().ToString("N").ToUpper() + "." + fileExtension;
  523. }
  524. var recorderFilePath = Path.Combine(savePath, fileName);
  525. if (File.Exists(recorderFilePath))
  526. {
  527. File.Delete(recorderFilePath);
  528. }
  529. if (partFileUrlList.Count == 1)
  530. {
  531. var url = partFileUrlList.FirstOrDefault() ?? string.Empty;
  532. var tempArray = await DownloadPartFileAsync(url);
  533. newSaveFileName = await SaveDownloadFile(savePath, fileExtension, tempArray, saveFileName);
  534. }
  535. else
  536. {
  537. var dic = new ConcurrentDictionary<string, byte[]>();
  538. long dataLength = 0;
  539. partFileUrlList.Sort();
  540. var parallelOption = new ParallelOptions() { MaxDegreeOfParallelism = 3 };
  541. Parallel.ForEach(partFileUrlList, parallelOption, item =>
  542. {
  543. var tempArray = DownloadPartFileAsync(item).GetAwaiter().GetResult();
  544. dataLength += tempArray.Length;
  545. dic.TryAdd(item, tempArray);
  546. });
  547. byte[] btArr = new byte[dataLength];
  548. int index = 0;
  549. foreach (var item in partFileUrlList)
  550. {
  551. Array.Copy(dic[item], 0, btArr, index, dic[item].Length);
  552. index += Convert.ToInt32(dic[item].Length);
  553. }
  554. newSaveFileName = await SaveDownloadFile(savePath, fileExtension, btArr, saveFileName);
  555. }
  556. }
  557. catch (Exception ex)
  558. {
  559. Logger.WriteLineError($"Download And Save File fail, {ex}");
  560. return string.Empty;
  561. }
  562. finally
  563. {
  564. GC.Collect();
  565. }
  566. return newSaveFileName;
  567. }
  568. /// <summary>
  569. /// 分块下载文件
  570. /// </summary>
  571. private async Task<byte[]> DownloadPartFileAsync(string fileUrl)
  572. {
  573. byte[] byteArr = null;
  574. using (var request = new HttpRequestMessage())
  575. {
  576. request.RequestUri = new Uri(fileUrl);
  577. request.Method = HttpMethod.Head;
  578. var response = await _httpClient.SendAsync(request);
  579. long allStreamLength = 0;
  580. if (response != null && response.StatusCode == HttpStatusCode.OK && response.Content != null && response.Content.Headers != null)
  581. {
  582. allStreamLength = response.Content.Headers.ContentLength.Value;
  583. }
  584. if (allStreamLength == 0)
  585. {
  586. Logger.WriteLineWarn($"file length is empty, token:{fileUrl}");
  587. return null;
  588. }
  589. byteArr = new byte[allStreamLength];
  590. try
  591. {
  592. for (int fileIndex = 0; true; fileIndex += _autoSliceSizeForUpload)
  593. {
  594. if (fileIndex >= allStreamLength)
  595. {
  596. Logger.WriteLineInfo("download fileIndex:"+ fileIndex);
  597. break;
  598. }
  599. var newChunk = fileIndex + _autoSliceSizeForUpload;
  600. var newLength = newChunk > allStreamLength ? allStreamLength : newChunk;
  601. var doRes = await _uploadFile.DoHttpRequest(fileUrl, byteArr, fileIndex, newLength, allStreamLength);
  602. if (!doRes)
  603. {
  604. Logger.WriteLineInfo("download doRes:"+ doRes);
  605. break;
  606. }
  607. }
  608. }
  609. catch (Exception ex)
  610. {
  611. Logger.WriteLineError($"Downloding file error {ex}");
  612. return null;
  613. }
  614. }
  615. GC.Collect();
  616. return byteArr;
  617. }
  618. /// <summary>
  619. /// 保存下载文件
  620. /// </summary>
  621. /// <param name="srcPath">保存路径</param>
  622. /// <param name="fileExtension">File Name Extension</param>
  623. /// <param name="fileData">数据</param>
  624. /// <param name="saveFileName">指定保存文件</param>
  625. /// <returns>保存文件</returns>
  626. private async Task<string> SaveDownloadFile(string srcPath, string fileExtension, byte[] fileData, string saveFileName = "")
  627. {
  628. if (!Directory.Exists(srcPath))
  629. {
  630. Directory.CreateDirectory(srcPath);
  631. }
  632. var fileName = saveFileName;
  633. if (string.IsNullOrEmpty(fileName))
  634. {
  635. fileName = Guid.NewGuid().ToString("N").ToUpper() + "." + fileExtension;
  636. }
  637. var recorderFilePath = Path.Combine(srcPath, fileName);
  638. if (File.Exists(recorderFilePath))
  639. {
  640. //存在删除
  641. File.Delete(recorderFilePath);
  642. }
  643. //存储文件
  644. using (var fw = File.Open(recorderFilePath, FileMode.Create, FileAccess.ReadWrite, FileShare.ReadWrite))
  645. {
  646. fw.Write(fileData, 0, fileData.Length);
  647. }
  648. return fileName;
  649. }
  650. /// <summary>
  651. /// 读取文件
  652. /// </summary>
  653. private Dictionary<int, byte[]> ReadFileData(string filePath, long manualDivisionForUpload)
  654. {
  655. var dic = new Dictionary<int, byte[]>();
  656. using (FileStream fs = new FileStream(filePath, FileMode.Open, FileAccess.Read))
  657. {
  658. var partCount = (int)((fs.Length / manualDivisionForUpload) + (fs.Length % manualDivisionForUpload > 0 ? 1 : 0));
  659. //大于1.7G的问题,用分块读取File
  660. if (fs.Length > 1782579200)
  661. {
  662. var byteSize = 10;
  663. byte[] buffer = new byte[byteSize * 1024 * 1024];//10M的缓冲byte
  664. int fileCursor = 0;
  665. int readerCursor = 0;
  666. var splitFileTimes = manualDivisionForUpload / buffer.Length + (manualDivisionForUpload % buffer.Length > 0 ? 1 : 0);
  667. var fileSplitFileTimes = fs.Length / buffer.Length + (fs.Length % buffer.Length > 0 ? 1 : 0);
  668. if (fileSplitFileTimes < splitFileTimes)
  669. {
  670. splitFileTimes = fileSplitFileTimes;
  671. }
  672. //超过200就不能读取了,会溢出
  673. if (splitFileTimes > 200)
  674. {
  675. return dic;
  676. }
  677. NextFileBegin:
  678. using (MemoryStream ms = new MemoryStream())
  679. {
  680. int readLength = 0;
  681. while ((readLength = fs.Read(buffer, 0, buffer.Length)) > 0)
  682. {
  683. readerCursor++;
  684. ms.Write(buffer, 0, readLength);
  685. if (readerCursor >= splitFileTimes)
  686. {
  687. var allBytes = ms.ToArray();
  688. dic.Add(fileCursor, allBytes);
  689. readerCursor = 0;
  690. fileCursor++;
  691. break;
  692. }
  693. else
  694. {
  695. if (readLength < buffer.Length && readerCursor > 0 && readerCursor < splitFileTimes)
  696. {
  697. //表示最后一个了
  698. var allBytes = ms.ToArray();
  699. dic.Add(fileCursor, allBytes);
  700. readerCursor = 0;
  701. fileCursor++;
  702. break;
  703. }
  704. }
  705. }
  706. }
  707. if (fileCursor < partCount)
  708. {
  709. goto NextFileBegin;
  710. }
  711. }
  712. else
  713. {
  714. //读取文件
  715. byte[] btArr = new byte[fs.Length];
  716. fs.Read(btArr, 0, btArr.Length);
  717. if (partCount > 1)
  718. {
  719. long offsetIndex = 0;
  720. for (var i = 0; i < partCount; i++)
  721. {
  722. var tempLength = manualDivisionForUpload;
  723. if ((partCount - 1) == i)
  724. {
  725. //最后一个分块File
  726. tempLength = btArr.Length - (partCount - 1) * manualDivisionForUpload;
  727. }
  728. var tempByte = new byte[tempLength];
  729. offsetIndex = i * manualDivisionForUpload;
  730. Array.Copy(btArr, offsetIndex, tempByte, 0, tempByte.Length);
  731. dic.Add(i, tempByte);
  732. }
  733. }
  734. else
  735. {
  736. var partDetail = new FileTransferRecorderDetail();
  737. dic.TryAdd(0, btArr);
  738. }
  739. }
  740. }
  741. return dic;
  742. }
  743. /// <summary>
  744. /// Encrypt to MD5 string
  745. /// </summary>
  746. /// <param name="str">source string</param>
  747. /// <returns></returns>
  748. private string ToMD5(string str)
  749. {
  750. if (string.IsNullOrWhiteSpace(str)) return string.Empty;
  751. System.Security.Cryptography.MD5 md5 = new MD5CryptoServiceProvider();
  752. byte[] bPwd = Encoding.UTF8.GetBytes(str);
  753. byte[] bMD5 = md5.ComputeHash(bPwd);
  754. md5.Clear();
  755. StringBuilder sbMD5Pwd = new StringBuilder();
  756. for (int i = 0; i < bMD5.Length; i++)
  757. {
  758. sbMD5Pwd.Append(bMD5[i].ToString("x2"));
  759. }
  760. return sbMD5Pwd.ToString();
  761. }
  762. }
  763. }