BlackHole.php 5.3 KB

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