From cc9dd4280f22c7cecc5d25336c0c36c4451fb0be Mon Sep 17 00:00:00 2001
From: Robin Appelman <robin@icewind.nl>
Date: Wed, 7 Jun 2017 17:34:17 +0200
Subject: add stat cache for s3 external storage

Signed-off-by: Robin Appelman <robin@icewind.nl>
---
 apps/files_external/lib/Lib/Storage/AmazonS3.php | 63 +++++++++++++++++++++---
 1 file changed, 56 insertions(+), 7 deletions(-)

(limited to 'apps/files_external')

diff --git a/apps/files_external/lib/Lib/Storage/AmazonS3.php b/apps/files_external/lib/Lib/Storage/AmazonS3.php
index 59d8fad5ac8..6d300312112 100644
--- a/apps/files_external/lib/Lib/Storage/AmazonS3.php
+++ b/apps/files_external/lib/Lib/Storage/AmazonS3.php
@@ -36,10 +36,12 @@
 
 namespace OCA\Files_External\Lib\Storage;
 
+use Aws\Result;
 use Aws\S3\S3Client;
 use Aws\S3\Exception\S3Exception;
 use Icewind\Streams\CallbackWrapper;
 use Icewind\Streams\IteratorDirectory;
+use OC\Cache\CappedMemoryCache;
 use OC\Files\ObjectStore\S3ConnectionTrait;
 use OC\Files\ObjectStore\S3ObjectTrait;
 use OCP\Constants;
@@ -53,9 +55,13 @@ class AmazonS3 extends \OC\Files\Storage\Common {
 	 */
 	private $rescanDelay = 10;
 
+	/** @var CappedMemoryCache|Result[] */
+	private $objectCache;
+
 	public function __construct($parameters) {
 		parent::__construct($parameters);
 		$this->parseParams($parameters);
+		$this->objectCache = new CappedMemoryCache();
 	}
 
 	/**
@@ -83,6 +89,43 @@ class AmazonS3 extends \OC\Files\Storage\Common {
 		return $path;
 	}
 
+	private function clearCache() {
+		$this->objectCache = new CappedMemoryCache();
+	}
+
+	private function invalidateCache($key) {
+		unset($this->objectCache[$key]);
+		$keys = array_keys($this->objectCache->getData());
+		$keyLength = strlen($key);
+		foreach ($keys as $existingKey) {
+			if (substr($existingKey, 0, $keyLength) === $keys) {
+				unset($this->objectCache[$existingKey]);
+			}
+		}
+	}
+
+	/**
+	 * @param $key
+	 * @return Result|boolean
+	 */
+	private function headObject($key) {
+		if (!isset($this->objectCache[$key])) {
+			try {
+				$this->objectCache[$key] = $this->getConnection()->headObject(array(
+					'Bucket' => $this->bucket,
+					'Key' => $key
+				));
+			} catch (S3Exception $e) {
+				if ($e->getStatusCode() >= 500) {
+					throw $e;
+				}
+				$this->objectCache[$key] = false;
+			}
+		}
+
+		return $this->objectCache[$key];
+	}
+
 	/**
 	 * Updates old storage ids (v0.2.1 and older) that are based on key and secret to new ones based on the bucket name.
 	 * TODO Do this in an update.php. requires iterating over all users and loading the mount.json from their home
@@ -152,6 +195,8 @@ class AmazonS3 extends \OC\Files\Storage\Common {
 			return false;
 		}
 
+		$this->invalidateCache($path);
+
 		return true;
 	}
 
@@ -171,10 +216,12 @@ class AmazonS3 extends \OC\Files\Storage\Common {
 			return false;
 		}
 
+		$this->invalidateCache($path);
 		return $this->batchDelete($path);
 	}
 
 	protected function clearBucket() {
+		$this->clearCache();
 		try {
 			$this->getConnection()->clearBucket($this->bucket);
 			return true;
@@ -266,10 +313,7 @@ class AmazonS3 extends \OC\Files\Storage\Common {
 				$stat['size'] = -1; //unknown
 				$stat['mtime'] = time() - $this->rescanDelay * 1000;
 			} else {
-				$result = $this->getConnection()->headObject(array(
-					'Bucket' => $this->bucket,
-					'Key' => $path
-				));
+				$result = $this->headObject($path);
 
 				$stat['size'] = $result['ContentLength'] ? $result['ContentLength'] : 0;
 				if (isset($result['Metadata']['lastmodified'])) {
@@ -290,7 +334,7 @@ class AmazonS3 extends \OC\Files\Storage\Common {
 	public function is_dir($path) {
 		$path = $this->normalizePath($path);
 		try {
-			return $this->isRoot($path) || $this->getConnection()->doesObjectExist($this->bucket, $path . '/');
+			return $this->isRoot($path) || $this->headObject($path . '/');
 		} catch (S3Exception $e) {
 			\OCP\Util::logException('files_external', $e);
 			return false;
@@ -305,10 +349,10 @@ class AmazonS3 extends \OC\Files\Storage\Common {
 		}
 
 		try {
-			if ($this->getConnection()->doesObjectExist($this->bucket, $path)) {
+			if ($this->headObject($path)) {
 				return 'file';
 			}
-			if ($this->getConnection()->doesObjectExist($this->bucket, $path . '/')) {
+			if ($this->headObject($path . '/')) {
 				return 'dir';
 			}
 		} catch (S3Exception $e) {
@@ -336,6 +380,7 @@ class AmazonS3 extends \OC\Files\Storage\Common {
 
 		try {
 			$this->deleteObject($path);
+			$this->invalidateCache($path);
 		} catch (S3Exception $e) {
 			\OCP\Util::logException('files_external', $e);
 			return false;
@@ -435,6 +480,7 @@ class AmazonS3 extends \OC\Files\Storage\Common {
 			return false;
 		}
 
+		$this->invalidateCache($path);
 		return true;
 	}
 
@@ -483,6 +529,8 @@ class AmazonS3 extends \OC\Files\Storage\Common {
 			}
 		}
 
+		$this->invalidateCache($path2);
+
 		return true;
 	}
 
@@ -533,6 +581,7 @@ class AmazonS3 extends \OC\Files\Storage\Common {
 		try {
 			$source = fopen($tmpFile, 'r');
 			$this->writeObject($path, $source);
+			$this->invalidateCache($path);
 			fclose($source);
 
 			unlink($tmpFile);
-- 
cgit v1.2.3