mirror of
https://we.phorge.it/source/phorge.git
synced 2024-12-20 04:20:55 +01:00
Stream chunks when sending chunked files
Summary: Ref T7149. Return a real iterator from the Chunk engine, which processes chunks sequentially. Test Plan: This is a bit hard to read, but shows the underlying chunks being accessed one at a time and only some being accessed when requesting a range of a file: ``` $ ./bin/files cat F878 --trace --begin 100 --end 256 ... >>> [10] <query> SELECT * FROM `file_storageblob` WHERE `id` = 85 <<< [10] <query> 240 us better software. Phabricat>>> [11] <query> SELECT * FROM `file_storageblob` WHERE `id` = 84 <<< [11] <query> 205 us or includes applications for: >>> [12] <query> SELECT * FROM `file_storageblob` WHERE `id` = 83 <<< [12] <query> 226 us - reviewing and auditing source>>> [13] <query> SELECT * FROM `file_storageblob` WHERE `id` = 82 <<< [13] <query> 203 us code; - hosting and browsing >>> [14] <query> SELECT * FROM `file_storageblob` WHERE `id` = 81 <<< [14] <query> 231 us repositories; - tracking bugs; ``` Reviewers: btrahan Reviewed By: btrahan Subscribers: joshuaspence, epriestley Maniphest Tasks: T7149 Differential Revision: https://secure.phabricator.com/D12073
This commit is contained in:
parent
81d88985a0
commit
c19bb57730
5 changed files with 105 additions and 13 deletions
|
@ -1795,6 +1795,7 @@ phutil_register_library_map(array(
|
|||
'PhabricatorFile' => 'applications/files/storage/PhabricatorFile.php',
|
||||
'PhabricatorFileBundleLoader' => 'applications/files/query/PhabricatorFileBundleLoader.php',
|
||||
'PhabricatorFileChunk' => 'applications/files/storage/PhabricatorFileChunk.php',
|
||||
'PhabricatorFileChunkIterator' => 'applications/files/engine/PhabricatorFileChunkIterator.php',
|
||||
'PhabricatorFileChunkQuery' => 'applications/files/query/PhabricatorFileChunkQuery.php',
|
||||
'PhabricatorFileCommentController' => 'applications/files/controller/PhabricatorFileCommentController.php',
|
||||
'PhabricatorFileComposeController' => 'applications/files/controller/PhabricatorFileComposeController.php',
|
||||
|
@ -5096,6 +5097,10 @@ phutil_register_library_map(array(
|
|||
'PhabricatorPolicyInterface',
|
||||
'PhabricatorDestructibleInterface',
|
||||
),
|
||||
'PhabricatorFileChunkIterator' => array(
|
||||
'Phobject',
|
||||
'Iterator',
|
||||
),
|
||||
'PhabricatorFileChunkQuery' => 'PhabricatorCursorPagedPolicyAwareQuery',
|
||||
'PhabricatorFileCommentController' => 'PhabricatorFileController',
|
||||
'PhabricatorFileComposeController' => 'PhabricatorFileController',
|
||||
|
|
|
@ -177,4 +177,15 @@ final class PhabricatorChunkedFileStorageEngine
|
|||
return 32;
|
||||
}
|
||||
|
||||
public function getFileDataIterator(PhabricatorFile $file, $begin, $end) {
|
||||
$chunks = id(new PhabricatorFileChunkQuery())
|
||||
->setViewer(PhabricatorUser::getOmnipotentUser())
|
||||
->withChunkHandles(array($file->getStorageHandle()))
|
||||
->withByteRange($begin, $end)
|
||||
->needDataFiles(true)
|
||||
->execute();
|
||||
|
||||
return new PhabricatorFileChunkIterator($chunks, $begin, $end);
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
@ -0,0 +1,72 @@
|
|||
<?php
|
||||
|
||||
final class PhabricatorFileChunkIterator
|
||||
extends Phobject
|
||||
implements Iterator {
|
||||
|
||||
private $chunks;
|
||||
private $cursor;
|
||||
private $begin;
|
||||
private $end;
|
||||
private $data;
|
||||
|
||||
public function __construct(array $chunks, $begin = null, $end = null) {
|
||||
$chunks = msort($chunks, 'getByteStart');
|
||||
$this->chunks = $chunks;
|
||||
|
||||
if ($begin !== null) {
|
||||
foreach ($chunks as $key => $chunk) {
|
||||
if ($chunk->getByteEnd() >= $begin) {
|
||||
unset($chunks[$key]);
|
||||
}
|
||||
break;
|
||||
}
|
||||
$this->begin = $begin;
|
||||
}
|
||||
|
||||
if ($end !== null) {
|
||||
foreach ($chunks as $key => $chunk) {
|
||||
if ($chunk->getByteStart() <= $end) {
|
||||
unset($chunks[$key]);
|
||||
}
|
||||
}
|
||||
$this->end = $end;
|
||||
}
|
||||
}
|
||||
|
||||
public function current() {
|
||||
$chunk = head($this->chunks);
|
||||
$data = $chunk->getDataFile()->loadFileData();
|
||||
|
||||
if ($this->end !== null) {
|
||||
if ($chunk->getByteEnd() > $this->end) {
|
||||
$data = substr($data, 0, ($this->end - $chunk->getByteStart()));
|
||||
}
|
||||
}
|
||||
|
||||
if ($this->begin !== null) {
|
||||
if ($chunk->getByteStart() < $this->begin) {
|
||||
$data = substr($data, ($this->begin - $chunk->getByteStart()));
|
||||
}
|
||||
}
|
||||
|
||||
return $data;
|
||||
}
|
||||
|
||||
public function key() {
|
||||
return head_key($this->chunks);
|
||||
}
|
||||
|
||||
public function next() {
|
||||
unset($this->chunks[$this->key()]);
|
||||
}
|
||||
|
||||
public function rewind() {
|
||||
return;
|
||||
}
|
||||
|
||||
public function valid() {
|
||||
return (count($this->chunks) > 0);
|
||||
}
|
||||
|
||||
}
|
|
@ -290,5 +290,20 @@ abstract class PhabricatorFileStorageEngine {
|
|||
return $engine->getChunkSize();
|
||||
}
|
||||
|
||||
public function getFileDataIterator(PhabricatorFile $file, $begin, $end) {
|
||||
// The default implementation is trivial and just loads the entire file
|
||||
// upfront.
|
||||
$data = $file->loadFileData();
|
||||
|
||||
if ($begin !== null && $end !== null) {
|
||||
$data = substr($data, $begin, ($end - $begin));
|
||||
} else if ($begin !== null) {
|
||||
$data = substr($data, $begin);
|
||||
} else if ($end !== null) {
|
||||
$data = substr($data, 0, $end);
|
||||
}
|
||||
|
||||
return array($data);
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
@ -612,19 +612,8 @@ final class PhabricatorFile extends PhabricatorFileDAO
|
|||
* @return Iterable Iterable object which emits requested data.
|
||||
*/
|
||||
public function getFileDataIterator($begin = null, $end = null) {
|
||||
// The default implementation is trivial and just loads the entire file
|
||||
// upfront.
|
||||
$data = $this->loadFileData();
|
||||
|
||||
if ($begin !== null && $end !== null) {
|
||||
$data = substr($data, $begin, ($end - $begin));
|
||||
} else if ($begin !== null) {
|
||||
$data = substr($data, $begin);
|
||||
} else if ($end !== null) {
|
||||
$data = substr($data, 0, $end);
|
||||
}
|
||||
|
||||
return array($data);
|
||||
$engine = $this->instantiateStorageEngine();
|
||||
return $engine->getFileDataIterator($this, $begin, $end);
|
||||
}
|
||||
|
||||
|
||||
|
|
Loading…
Reference in a new issue