AIDiagnosisService.cs 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651
  1. using WingServerCommon.Service;
  2. using WingInterfaceLibrary.Interface;
  3. using WingInterfaceLibrary.Request;
  4. using WingInterfaceLibrary.Enum;
  5. using WingServerCommon.Config;
  6. using WingServerCommon.Config.Parameters;
  7. using WingAIDiagnosisService.Carotid.Utilities;
  8. using Vinno.vCloud.Common.Vid2;
  9. using WingAIDiagnosisService.Manage;
  10. using WingAIDiagnosisService.Carotid;
  11. using Newtonsoft.Json;
  12. using Emgu.CV;
  13. using Emgu.CV.CvEnum;
  14. using AI.DiagSystem;
  15. using WingServerCommon.Log;
  16. using Emgu.CV.Structure;
  17. using System.IO;
  18. using System;
  19. using System.Threading.Tasks;
  20. using System.Collections.Generic;
  21. using System.Linq;
  22. using AI.Common;
  23. using System.Threading;
  24. using AI.Common.Log;
  25. using SkiaSharp;
  26. using System.Net.Http;
  27. using System.Net;
  28. using WingInterfaceLibrary.Request.Storage;
  29. using WingInterfaceLibrary.Result.AIDiagnosis;
  30. using WingInterfaceLibrary.Request.AIDiagnosis;
  31. using WingInterfaceLibrary.DTO.Record;
  32. using WingInterfaceLibrary.DTO.Comment;
  33. namespace WingAIDiagnosisService.Service
  34. {
  35. /// <summary>
  36. /// AI诊断服务
  37. /// </summary>
  38. public partial class AIDiagnosisService : JsonRpcService, IAIDiagnosisService
  39. {
  40. private EnumPerformance _workerLevel = EnumPerformance.Medium;
  41. private int _batchImageSize = 16;
  42. private int _sleepTime = 400;
  43. private readonly string _tempFolder = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "DiagnosisTemp");
  44. private AIDiagSystem _diagSystem;
  45. private AILesionCompareUtils _lesionCompareUtil;
  46. private RecognizeCarotidType _recognizeCarotidType = new RecognizeCarotidType();
  47. //识别左右颈是否初始化成功
  48. private static bool _recognizeCarotidTypeCanBeUse = false;
  49. static HttpClient _httpClient = new HttpClient();
  50. private IAuthenticationService _authenticationService;
  51. private IStorageService _storageService;
  52. /// <summary>
  53. /// Init service
  54. /// </summary>
  55. public override void Load(JsonRpcClientPool jsonRpcClientPool)
  56. {
  57. base.Load(jsonRpcClientPool);
  58. _authenticationService = GetProxy<IAuthenticationService>();
  59. _storageService = GetProxy<IStorageService>();
  60. _workerLevel = (EnumPerformance)ConfigurationManager.GetParammeter<IntParameter>("AI", "WorkerLevel").Value;
  61. _batchImageSize = ConfigurationManager.GetParammeter<IntParameter>("AI", "BatchImageSize").Value;
  62. _httpClient.Timeout = TimeSpan.FromMinutes(15);
  63. InitAISystem();
  64. //初始化识别左右颈
  65. _recognizeCarotidTypeCanBeUse = _recognizeCarotidType.Initialization();
  66. }
  67. private void InitAISystem()
  68. {
  69. var modeFilePath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "Networks");
  70. _diagSystem = new AIDiagSystem(_workerLevel, modeFilePath, EnumInferWorkName.Default, false, true, true);
  71. if (_diagSystem != null)
  72. {
  73. _diagSystem.NotifyError += AIdiagSystem_NotifyError;
  74. _diagSystem.NotifyLog += AIdiagSystem_NotifyLog;
  75. }
  76. _lesionCompareUtil = new AILesionCompareUtils();
  77. }
  78. /// <summary>
  79. /// 查询杏聆荟支持的AI模块
  80. /// </summary>
  81. /// <param name="request">查询杏聆荟支持的AI模块请求实体</param>
  82. /// <returns></returns>
  83. /// <show>false</show>
  84. public async Task<IList<string>> FindDiagnosisModulesAsync(TokenRequest request)
  85. {
  86. try
  87. {
  88. var moduleNames = AIDiagSystem.GetValidModuleNamesForVclound() ?? new List<EnumAIModuleNames>();
  89. var resultData = moduleNames.Select(x => x.ToString()).ToList();
  90. resultData.Add("Carotid");
  91. return await Task.FromResult(resultData);
  92. }
  93. catch (Exception)
  94. {
  95. return new List<string>();
  96. }
  97. }
  98. /// <summary>
  99. /// 图像诊断
  100. /// </summary>
  101. /// <param name="request">图像诊断请求实体</param>
  102. /// <returns>图像诊断结果</returns>
  103. /// <show>false</show>
  104. public async Task<DiagnosisImageResult> DiagnosisImageAsync(DiagnosisImageRequest request)
  105. {
  106. try
  107. {
  108. var localFile = await DownloadAsync(request.FileToken);
  109. if (File.Exists(localFile))
  110. {
  111. if (_diagSystem == null)
  112. {
  113. InitAISystem();
  114. }
  115. using (var imageData = new VinnoImageData(localFile, OperationMode.Open))
  116. {
  117. if (DiagnosisHelper.IsCarotid(imageData))
  118. {
  119. if (imageData.ImageCount > DiagnosisHelper.CarotidMinImageCounts)
  120. {
  121. //颈动脉
  122. var result = CarotidDiagnosis(imageData);
  123. var resultData = new DiagnosisImageResult()
  124. {
  125. CarotidResult = result
  126. };
  127. return resultData;
  128. }
  129. }
  130. else
  131. {
  132. //乳腺、肝脏
  133. var results = NormalDiagnosis(imageData);
  134. var diagnosisResults = results.Select(x => new AIDiagnosisPerImageModel(x.Key, x.Value)).ToList();
  135. var diagnosisResult = Newtonsoft.Json.JsonConvert.DeserializeObject<List<AIDiagnosisPerImageDTO>>(Newtonsoft.Json.JsonConvert.SerializeObject(diagnosisResults));
  136. var resultData = new DiagnosisImageResult()
  137. {
  138. DiagnosisConclusion = (DiagnosisConclusionEnum)GetDiagnosisConclusion(diagnosisResults),
  139. DiagnosisResult = diagnosisResult,
  140. DiagnosisOrgans = GetDiagnosisOrgans(diagnosisResult),
  141. };
  142. return resultData;
  143. }
  144. }
  145. }
  146. }
  147. catch (Exception ex)
  148. {
  149. Logger.WriteLineWarn($"DiagnosisImageAsync err {ex}");
  150. }
  151. return new DiagnosisImageResult();
  152. }
  153. /// <summary>
  154. /// 生成AI报告
  155. /// </summary>
  156. /// <param name="request">生成AI报告请求实体</param>
  157. /// <returns></returns>
  158. /// <show>false</show>
  159. public async Task<DiagnosisReportResult> DiagnosisReportAsync(DiagnosisReportRequest request)
  160. {
  161. try
  162. {
  163. var manager = await CreateDiagnosisManagerAsync(request);
  164. var reportPerImages = manager.GetReportResults();
  165. var diagnosisConclusion = GetDiagnosisConclusion(reportPerImages);
  166. var diagnosisResult = new List<DiagnosisPerImageDTO>();
  167. foreach (var item in reportPerImages)
  168. {
  169. var aiFileToken = await UploadFileAsync(item.AILocalImagePath, Path.GetFileName(item.AILocalImagePath));
  170. var perImageResult = Newtonsoft.Json.JsonConvert.DeserializeObject<DiagnosisPerImageDTO>(Newtonsoft.Json.JsonConvert.SerializeObject(item));
  171. perImageResult.RemedicalCode = item.RemedicalCode;
  172. perImageResult.DataType = (RemedicalFileDataTypeEnum)item.DataType;
  173. perImageResult.Pixel = item.Pixel;
  174. perImageResult.AIFileToken = aiFileToken;
  175. diagnosisResult.Add(perImageResult);
  176. }
  177. return new DiagnosisReportResult
  178. {
  179. DiagnosisConclusion = (DiagnosisConclusionEnum)diagnosisConclusion,
  180. DiagnosisResult = diagnosisResult,
  181. };
  182. }
  183. catch (Exception ex)
  184. {
  185. Logger.WriteLineWarn($"AIService DiagnosisReport err, {ex}");
  186. }
  187. return new DiagnosisReportResult();
  188. }
  189. private async Task<BaseDiagnosis> CreateDiagnosisManagerAsync(DiagnosisReportRequest request)
  190. {
  191. var recordResults = new List<DiagnosisPerImageModel>();
  192. foreach (var remedical in request.RemedicalList)
  193. {
  194. foreach (var perImage in remedical.DiagnosisResult)
  195. {
  196. var localFile = await DownloadAsync(remedical.FileToken);
  197. if (File.Exists((localFile)))
  198. {
  199. var diagnosisPerImage = new DiagnosisPerImageModel();
  200. diagnosisPerImage.RemedicalCode = remedical.RemedicalCode;
  201. diagnosisPerImage.DataType = (WingAIDiagnosisService.Manage.DataType)remedical.DataType;
  202. diagnosisPerImage.LocalVidPath = localFile;
  203. diagnosisPerImage.Index = perImage.Index;
  204. diagnosisPerImage.PriorityScore = perImage.PriorityScore;
  205. var diagnosisResults = Newtonsoft.Json.JsonConvert.DeserializeObject<List<WingAIDiagnosisService.Manage.AIDiagnosisResultPerOrgan>>(Newtonsoft.Json.JsonConvert.SerializeObject(perImage.DiagResultsForEachOrgan));
  206. diagnosisPerImage.DiagResultsForEachOrgan = diagnosisResults;
  207. recordResults.Add(diagnosisPerImage);
  208. }
  209. }
  210. }
  211. switch (request.Organ)
  212. {
  213. case DiagnosisOrganEnum.Breast:
  214. return new BreastDiagnosis(recordResults, request.Organ);
  215. case DiagnosisOrganEnum.Liver:
  216. return new LiverDiagnosis(recordResults, request.Organ);
  217. default:
  218. throw new Exception($"not support organ type:{request.Organ.ToString()}");
  219. }
  220. }
  221. private CarotidResultDTO CarotidDiagnosis(VinnoImageData imageData)
  222. {
  223. var resampleInputData = new ResampleInputData();
  224. resampleInputData.SurfaceFilePath = DiagnosisHelper.GetCacheFilePath(DiagnosisHelper.SurfaceFileSuffix);
  225. resampleInputData.MdlFilePath = DiagnosisHelper.GetCacheFilePath(DiagnosisHelper.ModelFileSuffix);
  226. resampleInputData.MdlZipFilePath = DiagnosisHelper.GetCacheFilePath(DiagnosisHelper.ZipFileSuffix);
  227. resampleInputData.BaseAIImagePath = DiagnosisHelper.GetCacheFilePath(DiagnosisHelper.JPGFileSuffix);
  228. resampleInputData.PlaqueAIImagePath = DiagnosisHelper.GetCacheFilePath(DiagnosisHelper.JPGFileSuffix);
  229. resampleInputData.YShapeAIImagePath = DiagnosisHelper.GetCacheFilePath(DiagnosisHelper.JPGFileSuffix);
  230. var resampleResult = TryResample(imageData, resampleInputData);
  231. if (resampleResult.ResampleErrorCode == ResampleErrorCode.Success)
  232. {
  233. var carotidAIImageTokens = new List<CarotidAIImage>();
  234. if (resampleResult.CarotidAIMeasureResult.IsYImageSuccess)
  235. {
  236. carotidAIImageTokens.Add(new CarotidAIImage() { AIImageToken = resampleInputData.YShapeAIImagePath, AIImageType = CarotidAIImageType.YShape });
  237. }
  238. if (resampleResult.CarotidAIMeasureResult.IntimaResult.IsSuccess)
  239. {
  240. carotidAIImageTokens.Add(new CarotidAIImage() { AIImageToken = resampleInputData.BaseAIImagePath, AIImageType = CarotidAIImageType.Base });
  241. }
  242. if (resampleResult.CarotidAIMeasureResult.PlaqueResult.IsSuccess && resampleResult.CarotidAIMeasureResult.PlaqueResult.PlaqueCountType != WingAIDiagnosisService.Carotid.Utilities.DetectPlaque.PlaqueCountType.NoPlaque)
  243. {
  244. carotidAIImageTokens.Add(new CarotidAIImage() { AIImageToken = resampleInputData.PlaqueAIImagePath, AIImageType = CarotidAIImageType.Plaque });
  245. }
  246. var result = new CarotidResultDTO
  247. {
  248. CarotidScanType = (CarotidScanTypeEnum)resampleInputData.ScanType,
  249. CarotidScanDirection = (CarotidScanDirectionEnum)resampleInputData.CarotidScanDirection,
  250. SurfaceFile = resampleInputData.SurfaceFilePath,
  251. MdlFile = resampleInputData.MdlFilePath,
  252. MeasureImageFiles = carotidAIImageTokens.Select(x => new MeasureImageFileDTO { ImageType = (CarotidAIImageTypeEnum)x.AIImageType, ImageFile = x.AIImageToken }).ToList(),
  253. MeasureResult = JsonConvert.SerializeObject(resampleResult.CarotidAIMeasureResult),
  254. };
  255. return result;
  256. }
  257. return null;
  258. }
  259. private ResampleResult TryResample(VinnoImageData vinnoImageData, ResampleInputData resampleInputData)
  260. {
  261. float scanDistance = 7;
  262. ResampleResult resampleResult = new ResampleResult(ResampleErrorCode.Fail);
  263. var vinnoExtendedData = VinnoCarotidExtendedData.FromBytes(vinnoImageData.ExtendedData);
  264. CarotidScanType scanType = CarotidScanType.CarotidLeft;
  265. CarotidScanDirection direction = CarotidScanDirection.TopToBottom;
  266. if (vinnoExtendedData != null)
  267. {
  268. // Should be 6~9cm normally.
  269. //scanDistance = vinnoExtendedData.ScanDistance;
  270. scanType = vinnoExtendedData.CarotidType == CarotidType.Left ? CarotidScanType.CarotidLeft : CarotidScanType.CarotidRight;
  271. direction = vinnoExtendedData.CarotidDirection == CarotidDirection.BottomToTop ? CarotidScanDirection.BottomToTop
  272. : CarotidScanDirection.TopToBottom;
  273. }
  274. else
  275. {
  276. //This is a walkaround : vid's first frame is black image or depth coordinate is not in the right place.
  277. var middleCount = vinnoImageData.ImageCount / 2;
  278. var vinnoImage = vinnoImageData.GetImage(middleCount);
  279. var imageBuffer = vinnoImage.ImageData;
  280. if (_recognizeCarotidTypeCanBeUse)
  281. {
  282. //转成彩色图像
  283. using var img = new Mat();
  284. CvInvoke.Imdecode(imageBuffer, ImreadModes.Color, img);
  285. using var image = img.ToImage<Bgr, byte>();
  286. //识别左右颈信息
  287. scanType = _recognizeCarotidType.RecognizeType(image);
  288. }
  289. }
  290. resampleInputData.ScanDistance = scanDistance;
  291. resampleInputData.ScanType = scanType;
  292. resampleInputData.CarotidScanDirection = direction;
  293. resampleResult = ResampleModel.Instance.Resample(vinnoImageData, resampleInputData);
  294. return resampleResult;
  295. }
  296. private SKBitmap CreateBitmap(VinnoImage vinnoImage)
  297. {
  298. try
  299. {
  300. return SKBitmap.Decode(vinnoImage.ImageData);
  301. }
  302. catch (Exception ex)
  303. {
  304. Logger.WriteLineError($"Create skbitmap by VinnoImage error:{ex}");
  305. }
  306. return null;
  307. }
  308. private EnumColorType MapTo(SKColorType sKColor)
  309. {
  310. switch (sKColor)
  311. {
  312. case SKColorType.Rgba8888:
  313. return EnumColorType.Rgba;
  314. case SKColorType.Rgb888x:
  315. return EnumColorType.Rgba;
  316. case SKColorType.Bgra8888:
  317. return EnumColorType.Bgra;
  318. case SKColorType.Gray8:
  319. return EnumColorType.Gray8;
  320. }
  321. throw new Exception($"AIService not support color type:{sKColor}");
  322. }
  323. private void AIdiagSystem_NotifyError(object sender, ErrorEventArgs e)
  324. {
  325. Logger.WriteLineError("AIdiagSystem_NotifyError:" + e.GetException());
  326. }
  327. private void AIdiagSystem_NotifyLog(object sender, LogEventArgs e)
  328. {
  329. if (e != null && !string.IsNullOrEmpty(e.Msg))
  330. {
  331. switch (e.LogType)
  332. {
  333. case EnumLogType.InfoLog:
  334. Logger.WriteLineInfo($"AIdiagSystem_NotifyLog:{e.Msg}");
  335. break;
  336. case EnumLogType.ErrorLog:
  337. Logger.WriteLineError($"AIdiagSystem_NotifyLog:{e.Msg}");
  338. break;
  339. case EnumLogType.WarnLog:
  340. Logger.WriteLineWarn($"AIdiagSystem_NotifyLog:{e.Msg}");
  341. break;
  342. case EnumLogType.FatalLog:
  343. Logger.WriteLineError($"AIdiagSystem_NotifyLog:{e.Msg}");
  344. break;
  345. default:
  346. Logger.WriteLineInfo(e.Msg);
  347. break;
  348. }
  349. }
  350. }
  351. private DiagnosisConclusion GetDiagnosisConclusion<T>(List<T> results) where T : AIDiagnosisPerImageModel
  352. {
  353. var diagnosisConclusions = new List<DiagnosisConclusion>();
  354. foreach (var imageResult in results)
  355. {
  356. foreach (var diagnosisResult in imageResult.DiagResultsForEachOrgan)
  357. {
  358. var benignLabels = new List<int>();
  359. var malignantLabels = new List<int>();
  360. if (diagnosisResult.Organ == DiagnosisOrganEnum.Breast)
  361. {
  362. benignLabels = new List<int> { 1, 2, 3 };
  363. malignantLabels = new List<int> { 4, 5, 6, 7 };
  364. }
  365. else if (diagnosisResult.Organ == DiagnosisOrganEnum.Liver)
  366. {
  367. benignLabels = new List<int> { 1, 2, 3, 5, 6, 7, 8 };
  368. malignantLabels = new List<int> { 4 };
  369. }
  370. var labels = diagnosisResult.DetectedObjects.Select(x => x.Label);
  371. if (labels.Contains(0))
  372. {
  373. diagnosisConclusions.Add(DiagnosisConclusion.NoObviousLesion);
  374. }
  375. if (labels.Intersect(benignLabels).Any())
  376. {
  377. diagnosisConclusions.Add(DiagnosisConclusion.Benign);
  378. }
  379. if (labels.Intersect(malignantLabels).Any())
  380. {
  381. diagnosisConclusions.Add(DiagnosisConclusion.Malignant);
  382. }
  383. }
  384. }
  385. var containsBenign = diagnosisConclusions.Contains(DiagnosisConclusion.Benign);
  386. var containsMalignant = diagnosisConclusions.Contains(DiagnosisConclusion.Malignant);
  387. var containsNoObviousLesion = diagnosisConclusions.Contains(DiagnosisConclusion.NoObviousLesion);
  388. if (containsBenign && containsMalignant)
  389. {
  390. return DiagnosisConclusion.BenignAndMalignant;
  391. }
  392. else if (containsBenign)
  393. {
  394. return DiagnosisConclusion.Benign;
  395. }
  396. else if (containsMalignant)
  397. {
  398. return DiagnosisConclusion.Malignant;
  399. }
  400. else if (containsNoObviousLesion)
  401. {
  402. return DiagnosisConclusion.NoObviousLesion;
  403. }
  404. else
  405. {
  406. return DiagnosisConclusion.Unrecognized;
  407. }
  408. }
  409. private List<DiagnosisOrganEnum> GetDiagnosisOrgans(List<AIDiagnosisPerImageDTO> results)
  410. {
  411. var diagnosisOrgans = new List<DiagnosisOrganEnum>();
  412. foreach (var imageResult in results)
  413. {
  414. foreach (var diagnosisResult in imageResult.DiagResultsForEachOrgan)
  415. {
  416. diagnosisOrgans.Add((DiagnosisOrganEnum)diagnosisResult.Organ);
  417. }
  418. }
  419. return diagnosisOrgans.Distinct().ToList();
  420. }
  421. private Dictionary<int, AIDiagResultPerImg> NormalDiagnosis(VinnoImageData imageData)
  422. {
  423. var images = new List<RawImage>();
  424. var results = new Dictionary<int, AIDiagResultPerImg>();
  425. try
  426. {
  427. var totalCount = imageData.ImageCount;
  428. for (var i = 0; i < totalCount; i++)
  429. {
  430. var image = imageData.GetImage(i);
  431. var bitmap = CreateBitmap(image);
  432. if (bitmap != null)
  433. {
  434. images.Add(new RawImage(bitmap.Bytes, bitmap.Width, bitmap.Height, MapTo(bitmap.ColorType)));
  435. }
  436. }
  437. if (images.Count > 0)
  438. {
  439. var excuteCount = images.Count / _batchImageSize;
  440. var remainderCount = images.Count % _batchImageSize;
  441. var diagId = _diagSystem.StartEvalutationOfMultipleImageBatches(images.Count);
  442. for (var j = 0; j < excuteCount; j++)
  443. {
  444. var excuteImages = images.GetRange(j * _batchImageSize, _batchImageSize);
  445. _diagSystem.PushOneBatchOfImagesAsync(diagId, excuteImages);
  446. Thread.Sleep(_sleepTime);
  447. }
  448. if (remainderCount > 0)
  449. {
  450. var excuteImages = images.GetRange(excuteCount * _batchImageSize, remainderCount);
  451. _diagSystem.PushOneBatchOfImagesAsync(diagId, excuteImages);
  452. }
  453. results = _diagSystem.GetEvaluationsOfPushedMultipleImageBatches(diagId);
  454. }
  455. }
  456. catch (Exception ex)
  457. {
  458. Logger.WriteLineWarn($"AIService NormalDiagnosis err, {ex}");
  459. }
  460. finally
  461. {
  462. foreach (var image in images)
  463. {
  464. image?.Dispose();
  465. }
  466. }
  467. return results;
  468. }
  469. /// <summary>下载文件</summary>
  470. /// <param name="fileUrl"></param>
  471. /// <returns></returns>
  472. private async Task<string> DownloadAsync(string fileUrl)
  473. {
  474. try
  475. {
  476. if (string.IsNullOrEmpty(fileUrl))
  477. {
  478. return string.Empty;
  479. }
  480. if (!Directory.Exists(_tempFolder))
  481. {
  482. Directory.CreateDirectory(_tempFolder);
  483. }
  484. var fileName = Path.GetFileName(fileUrl);
  485. var tempFile = Path.Combine(_tempFolder, fileName);
  486. if (File.Exists(tempFile))
  487. {
  488. return tempFile;
  489. }
  490. long fileSize = 0;
  491. using (var request = new HttpRequestMessage())
  492. {
  493. request.RequestUri = new Uri(fileUrl);
  494. request.Method = HttpMethod.Get;
  495. var response = await _httpClient.SendAsync(request);
  496. if (response != null && response.StatusCode == HttpStatusCode.OK)
  497. {
  498. var contentLength = response.Content.Headers.ContentLength;
  499. fileSize = contentLength == null ? 0 : contentLength.Value;
  500. }
  501. }
  502. if (fileSize <= 0)
  503. {
  504. throw new NotSupportedException($"fileSize is {fileSize}");
  505. }
  506. byte[] bytes = await _httpClient.GetByteArrayAsync(fileUrl);
  507. File.WriteAllBytes(tempFile, bytes);
  508. return tempFile;
  509. }
  510. catch (Exception ex)
  511. {
  512. Logger.WriteLineWarn($"DiagnosisService download file err, url: {fileUrl}, {ex}");
  513. }
  514. finally
  515. {
  516. Logger.WriteLineInfo($"download file:{fileUrl}");
  517. }
  518. return string.Empty;
  519. }
  520. /// <summary>
  521. /// 上传文件
  522. /// </summary>
  523. /// <param name="filePath"></param>
  524. /// <param name="fileName"></param>
  525. /// <returns></returns>
  526. private async Task<string> UploadFileAsync(string filePath, string fileName)
  527. {
  528. var fileToken = "";
  529. using (var fileStream = new FileStream(filePath, FileMode.Open))
  530. {
  531. var size = fileStream.Length;
  532. byte[] buffer = new byte[fileStream.Length];
  533. fileStream.Read(buffer, 0, buffer.Length);
  534. fileToken = await DoUploadFile(fileName, buffer);
  535. }
  536. return fileToken;
  537. }
  538. /// <summary>
  539. /// 上传文件
  540. /// </summary>
  541. /// <param name="fileName"></param>
  542. /// <param name="fileData"></param>
  543. /// <returns></returns>
  544. async Task<string> DoUploadFile(string fileName, byte[] fileData)
  545. {
  546. var fileTypeEnum = GetUploadFileType(fileName.ToLower());
  547. var requestHeads = new Dictionary<string, string>();
  548. var defaultToken = await _authenticationService.GetServerDefaultTokenAsync();
  549. var authorizationRet = await _storageService.GetAuthorizationAsync(
  550. new FileServiceRequest
  551. {
  552. // FileTypeEnum = fileTypeEnum,
  553. // ApplicantTypeEnum = ApplicantTypeEnum.Server,
  554. Token = defaultToken
  555. });
  556. requestHeads.Add("Authorization", authorizationRet.Authorization);
  557. var fileUrl = authorizationRet.StorageUrl;
  558. Logger.WriteLineInfo($"DoUploadFile fileUrl:{fileUrl}");
  559. using (var request = new HttpRequestMessage())
  560. {
  561. request.RequestUri = new Uri(fileUrl);
  562. request.Method = HttpMethod.Put;
  563. using (MultipartFormDataContent content = new MultipartFormDataContent())
  564. {
  565. content.Add(new ByteArrayContent(fileData), "\"file\"", fileName);
  566. request.Content = content;
  567. foreach (var head in requestHeads)
  568. {
  569. request.Headers.TryAddWithoutValidation(head.Key, head.Value);
  570. }
  571. var result = await ExecuteRequest(request);
  572. if (!result)
  573. {
  574. throw new Exception("Upload file failed!");
  575. }
  576. }
  577. }
  578. return fileUrl;
  579. }
  580. private UploadFileTypeEnum GetUploadFileType(string fileName)
  581. {
  582. if (fileName.EndsWith(".vid"))
  583. {
  584. return UploadFileTypeEnum.VID;
  585. }
  586. else if (fileName.EndsWith(".jpg"))
  587. {
  588. return UploadFileTypeEnum.JPG;
  589. }
  590. else
  591. {
  592. return UploadFileTypeEnum.MP4;
  593. }
  594. }
  595. /// <summary>
  596. /// 执行请求
  597. /// </summary>
  598. /// <param name="httpRequestMessage"></param>
  599. /// <typeparam name="T"></typeparam>
  600. /// <returns></returns>
  601. public async Task<bool> ExecuteRequest(HttpRequestMessage httpRequestMessage)
  602. {
  603. try
  604. {
  605. var response = await _httpClient.SendAsync(httpRequestMessage);
  606. if (response != null && response.StatusCode == HttpStatusCode.OK)
  607. {
  608. return true;
  609. }
  610. return false;
  611. }
  612. catch (Exception ex)
  613. {
  614. throw ex;
  615. }
  616. }
  617. }
  618. }