BlackHole.php 5.5 KB

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