only do multipart upload for large files

Signed-off-by: Robin Appelman <robin@icewind.nl>

1 files changed, 24 insertions, 0 deletions

diff --git a/lib/private/Files/ObjectStore/S3ObjectTrait.php b/lib/private/Files/ObjectStore/S3ObjectTrait.php
index f7d9fb305c2..6527614e8d5 100644
--- a/lib/private/Files/ObjectStore/S3ObjectTrait.php
+++ b/lib/private/Files/ObjectStore/S3ObjectTrait.php
@@ -26,6 +26,8 @@ use Aws\S3\MultipartUploader;
 use Aws\S3\S3Client;
 use Psr\Http\Message\StreamInterface;
 
+const S3_UPLOAD_PART_SIZE = 5368709120;
+
 trait S3ObjectTrait {
 	/**
 	 * Returns the connection
@@ -62,17 +64,39 @@ trait S3ObjectTrait {
 	 * @since 7.0.0
 	 */
 	function writeObject($urn, $stream) {
+		$stat = fstat($stream);
+
+		if ($stat['size'] && $stat['size'] < S3_UPLOAD_PART_SIZE) {
+			$this->singlePartUpload($urn, $stream);
+		} else {
+			$this->multiPartUpload($urn, $stream);
+		}
+
+	}
+
+	private function singlePartUpload($urn, $stream) {
+		$this->getConnection()->putObject([
+			'Bucket' => $this->bucket,
+			'Key' => $urn,
+			'Body' => $stream
+		]);
+	}
+
+	private function multiPartUpload($urn, $stream) {
 		$uploader = new MultipartUploader($this->getConnection(), $stream, [
 			'bucket' => $this->bucket,
 			'key' => $urn,
 		]);
+
 		$tries = 0;
+
 		do {
 			try {
 				$result = $uploader->upload();
 			} catch (MultipartUploadException $e) {
 				rewind($stream);
 				$tries++;
+
 				if ($tries < 5) {
 					$uploader = new MultipartUploader($this->getConnection(), $stream, [
 						'state' => $e->getState()