BlackHole.php 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178
  1. <?php
  2. namespace App\Libraries\BIM\BlackHole;
  3. use App\BO\BimFileBO;
  4. use App\Libraries\BIM\Abstracts\BIMAbstract;
  5. use App\Libraries\BIM\BIMDriverEnum;
  6. use App\Models\Enums\BimFileConvertStatus;
  7. use GuzzleHttp\Psr7;
  8. use Illuminate\Http\UploadedFile;
  9. class BlackHole extends BIMAbstract
  10. {
  11. public function uploadFile(?UploadedFile $file, BimFileBO $bimFileBO)
  12. {
  13. $projectId = config("bim.black_hole.project_id");
  14. $treeInfo = $this->getTreeById($projectId);
  15. $chunkSize = 1024 * 1024 * 10; // 10M
  16. $chunkCount = (int) ceil($file->getSize() / $chunkSize);
  17. $modelFile = $this->modelFileCreate($projectId, $treeInfo['subNodes'][0]['nodeId'], $file, $chunkSize, $chunkCount);
  18. $dataSetId = $modelFile['dataSetId'];
  19. $fileResource = fopen($file, 'r');
  20. for ($i = 0; $i < $chunkCount; $i++) {
  21. $chunk = fread($fileResource, $chunkSize);
  22. $stream = Psr7\Utils::streamFor($chunk);
  23. $uploadFormData = [
  24. ['name' => 'FileId', 'contents' => $modelFile['fileId']],
  25. ['name' => 'FileInfo', 'contents' => $stream, 'filename' => $file->getClientOriginalName()],
  26. ['name' => 'FileDataId', 'contents' => $modelFile['fileDataId']],
  27. ['name' => 'Key', 'contents' => $modelFile['extraProperties']['ResumableKey']],
  28. ['name' => 'BlobIndex', 'contents' => $i],
  29. ];
  30. Client::getInstance()->post("/blackHole3D/project/modelFile/append", [
  31. 'multipart' => $uploadFormData,
  32. 'headers' => [
  33. 'fileUploadScheme' => 'Bim',
  34. 'fileUploadMode' => 'ResumableAppend',
  35. ],
  36. ]);
  37. }
  38. return [
  39. 'bim_data_set_id' => $dataSetId,
  40. 'bim_file_id' => $modelFile['fileId'],
  41. 'bim_convert_status' => 0,
  42. 'bim_driver' => BIMDriverEnum::BLACK_HOLE->value,
  43. ];
  44. }
  45. public function downloadSourceFile()
  46. {
  47. }
  48. public function viewDataSetModel(array $dataSetIDS)
  49. {
  50. $result = Client::getInstance()->post("/blackHole3D/project/dataSet/viewDataSetModel", [
  51. 'json' => [
  52. 'dataSetIds' => $dataSetIDS
  53. ]
  54. ]);
  55. return $result['data'] ?? [];
  56. }
  57. public function getModelDetail(\App\Models\File $file)
  58. {
  59. }
  60. public function findConvertStatus(string $dataSetId): array
  61. {
  62. $result = Client::getInstance()->post("/blackHole3D/project/dataSet/getConvertStatus", [
  63. 'json' => [
  64. 'dataSetId' => $dataSetId
  65. ]
  66. ]);
  67. $status = match ($result['data']['status']) {
  68. 0, 1 => BimFileConvertStatus::IN_QUEUE->value,
  69. 2 => BimFileConvertStatus::CONVERTING->value,
  70. 5 => BimFileConvertStatus::PAUSE->value,
  71. 10 => BimFileConvertStatus::DONE->value,
  72. 20 => BimFileConvertStatus::FAILED_TO_ADD_QUEUE->value,
  73. };
  74. return $this->convertStatusFormat($status, $result['data']['status']);
  75. }
  76. public function addToConvertQueue(string $dataSetId): array
  77. {
  78. $formData = [
  79. 'dataSetId' => $dataSetId,
  80. 'execConvertingCallback' => '',
  81. 'isBuildDbIndex' => true,
  82. 'isResetConverting' => true,
  83. 'paramFilter' => new \stdClass(),
  84. 'fileNames' => [],
  85. 'useDefaultConfig' => false,
  86. ];
  87. $result = Client::getInstance()->post("/blackHole3D/project/modelFile/addToConvertQueue", [
  88. 'json' => $formData,
  89. ]);
  90. return $result['data'] ?? [];
  91. }
  92. /**
  93. * 模型文件上传准备
  94. *
  95. * @param string $projectId
  96. * @param string $nodeId
  97. * @param UploadedFile $file
  98. * @param int $chunkSize
  99. * @param int $chunkCount
  100. * @return array|mixed
  101. */
  102. protected function modelFileCreate(string $projectId, string $nodeId, UploadedFile $file, int $chunkSize, int $chunkCount)
  103. {
  104. $formData = [
  105. 'ProjId' => $projectId,
  106. 'DataSetId' => $nodeId,
  107. 'FileName' => uniqid() . "_" . $file->getClientOriginalName(),
  108. 'FileSize' => $file->getSize(),
  109. 'FileType' => $file->getExtension() ? $file->getExtension() : pathinfo($file->getClientOriginalName(), PATHINFO_EXTENSION),
  110. 'Scheme' => 'Bim',
  111. 'BlobCount' => $chunkCount,
  112. 'BlobSize' => $chunkSize,
  113. 'UploadWay' => 0
  114. ];
  115. $result = Client::getInstance()->post("/blackHole3D/project/modelFile/create", [
  116. 'form_params' => $formData,
  117. 'headers' => [
  118. 'fileUploadScheme' => 'Bim',
  119. 'fileUploadMode' => 'ResumableCreate'
  120. ]
  121. ]);
  122. return $result['data'] ?? [];
  123. }
  124. protected function getTreeById(string $projectId)
  125. {
  126. $result = Client::getInstance()->post("/blackHole3D/project/modelTree/getTreeById", [
  127. 'json' => [
  128. 'projId' => $projectId
  129. ]
  130. ]);
  131. return $result['data'] ?? [];
  132. }
  133. protected function addLeafNodes(array $nodes, string $parentId = "")
  134. {
  135. $items = [];
  136. foreach ($nodes as $node) {
  137. $items[]['dataSetName'] = $node;
  138. }
  139. $result = Client::getInstance()->post("/blackHole3D/project/modelTree/addLevelNodes", [
  140. 'json' => [
  141. 'parentId' => $parentId,
  142. 'dataSets' => $items
  143. ]
  144. ]);
  145. return $result['data'] ?? [];
  146. }
  147. }