BlackHole.php 5.4 KB

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