feat: add a specialized writeStream implementation for s3 external storage

Signed-off-by: Robin Appelman <robin@icewind.nl>
This commit is contained in:
Robin Appelman 2024-07-22 21:51:34 +02:00 committed by skjnldsv
parent a6db0d33a8
commit e21ddcfeb0

View file

@ -43,6 +43,7 @@ use Aws\Result;
use Aws\S3\Exception\S3Exception;
use Aws\S3\S3Client;
use Icewind\Streams\CallbackWrapper;
use Icewind\Streams\CountWrapper;
use Icewind\Streams\IteratorDirectory;
use OCP\Cache\CappedMemoryCache;
use OC\Files\Cache\CacheEntry;
@ -790,4 +791,24 @@ class AmazonS3 extends \OC\Files\Storage\Common {
return true;
}
}
public function writeStream(string $path, $stream, ?int $size = null): int {
if ($size === null) {
$size = 0;
// track the number of bytes read from the input stream to return as the number of written bytes.
$stream = CountWrapper::wrap($stream, function (int $writtenSize) use (&$size) {
$size = $writtenSize;
});
}
if (!is_resource($stream)) {
throw new \InvalidArgumentException("Invalid stream provided");
}
$path = $this->normalizePath($path);
$this->writeObject($path, $stream, $this->mimeDetector->detectPath($path));
$this->invalidateCache($path);
return $size;
}
}