BlackHole.php 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173
  1. <?php
  2. namespace App\Services\File\BIM\BlackHole;
  3. use App\Models\Enums\BimFileConvertStatus;
  4. use App\Services\File\BIM\Abstracts\BIMAbstract;
  5. use App\Services\File\BIM\BIMDriverEnum;
  6. use App\Services\File\BIM\Contacts\BIMContact;
  7. use GuzzleHttp\Psr7;
  8. use Illuminate\Http\UploadedFile;
  9. class BlackHole extends BIMAbstract
  10. {
  11. public function uploadFile(UploadedFile $file, array $params = [])
  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 findConvertStatus(string $dataSetId): array
  58. {
  59. $result = Client::getInstance()->post("/blackHole3D/project/dataSet/getConvertStatus", [
  60. 'json' => [
  61. 'dataSetId' => $dataSetId
  62. ]
  63. ]);
  64. $status = match ($result['data']['status']) {
  65. 0, 1 => BimFileConvertStatus::IN_QUEUE->value,
  66. 2 => BimFileConvertStatus::CONVERTING->value,
  67. 5 => BimFileConvertStatus::PAUSE->value,
  68. 10 => BimFileConvertStatus::DONE->value,
  69. 20 => BimFileConvertStatus::FAILED_TO_ADD_QUEUE->value,
  70. };
  71. return $this->convertStatusFormat($status, $result['data']['status']);
  72. }
  73. public function addToConvertQueue(string $dataSetId): array
  74. {
  75. $formData = [
  76. 'dataSetId' => $dataSetId,
  77. 'execConvertingCallback' => '',
  78. 'isBuildDbIndex' => true,
  79. 'isResetConverting' => true,
  80. 'paramFilter' => new \stdClass(),
  81. 'fileNames' => [],
  82. 'useDefaultConfig' => false,
  83. ];
  84. $result = Client::getInstance()->post("/blackHole3D/project/modelFile/addToConvertQueue", [
  85. 'json' => $formData,
  86. ]);
  87. return $result['data'] ?? [];
  88. }
  89. /**
  90. * 模型文件上传准备
  91. *
  92. * @param string $projectId
  93. * @param string $nodeId
  94. * @param UploadedFile $file
  95. * @param int $chunkSize
  96. * @param int $chunkCount
  97. * @return array|mixed
  98. */
  99. protected function modelFileCreate(string $projectId, string $nodeId, UploadedFile $file, int $chunkSize, int $chunkCount)
  100. {
  101. $formData = [
  102. 'ProjId' => $projectId,
  103. 'DataSetId' => $nodeId,
  104. 'FileName' => uniqid() . "_" . $file->getClientOriginalName(),
  105. 'FileSize' => $file->getSize(),
  106. 'FileType' => $file->getExtension() ? $file->getExtension() : pathinfo($file->getClientOriginalName(), PATHINFO_EXTENSION),
  107. 'Scheme' => 'Bim',
  108. 'BlobCount' => $chunkCount,
  109. 'BlobSize' => $chunkSize,
  110. 'UploadWay' => 0
  111. ];
  112. $result = Client::getInstance()->post("/blackHole3D/project/modelFile/create", [
  113. 'form_params' => $formData,
  114. 'headers' => [
  115. 'fileUploadScheme' => 'Bim',
  116. 'fileUploadMode' => 'ResumableCreate'
  117. ]
  118. ]);
  119. return $result['data'] ?? [];
  120. }
  121. protected function getTreeById(string $projectId)
  122. {
  123. $result = Client::getInstance()->post("/blackHole3D/project/modelTree/getTreeById", [
  124. 'json' => [
  125. 'projId' => $projectId
  126. ]
  127. ]);
  128. return $result['data'] ?? [];
  129. }
  130. protected function addLeafNodes(array $nodes, string $parentId = "")
  131. {
  132. $items = [];
  133. foreach ($nodes as $node) {
  134. $items[]['dataSetName'] = $node;
  135. }
  136. $result = Client::getInstance()->post("/blackHole3D/project/modelTree/addLevelNodes", [
  137. 'json' => [
  138. 'parentId' => $parentId,
  139. 'dataSets' => $items
  140. ]
  141. ]);
  142. return $result['data'] ?? [];
  143. }
  144. }