[21] generate a better optimized query for path prefix search filters

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

Author: icewind1991

lib/private/Files/Cache/Cache.php

@@ -800,7 +800,7 @@ public function search($pattern) {
 	 * @param IResult $result
 	 * @return CacheEntry[]
 	 */
-	private function searchResultToCacheEntries(IResult $result): array {
+	protected function searchResultToCacheEntries(IResult $result): array {
 		$files = $result->fetchAll();
 
 		return array_map(function (array $data) {
@@ -837,7 +837,7 @@ public function searchByMime($mimetype) {
 		}, $files);
 	}
 
-	public function searchQuery(ISearchQuery $searchQuery) {
+	protected function buildSearchQuery(ISearchQuery $searchQuery): IQueryBuilder {
 		$builder = $this->getQueryBuilder();
 
 		$query = $builder->selectFileCache('file');
@@ -877,6 +877,12 @@ public function searchQuery(ISearchQuery $searchQuery) {
 			$query->setFirstResult($searchQuery->getOffset());
 		}
 
+		return $query;
+	}
+
+	public function searchQuery(ISearchQuery $searchQuery) {
+		$query = $this->buildSearchQuery($searchQuery);
+
 		$result = $query->execute();
 		$cacheEntries = $this->searchResultToCacheEntries($result);
 		$result->closeCursor();

lib/private/Files/Cache/Wrapper/CacheJail.php

@@ -29,6 +29,7 @@
 namespace OC\Files\Cache\Wrapper;
 
 use OC\Files\Cache\Cache;
+use OC\Files\Cache\QuerySearchHelper;
 use OC\Files\Search\SearchBinaryOperator;
 use OC\Files\Search\SearchComparison;
 use OC\Files\Search\SearchQuery;
@@ -63,6 +64,7 @@ public function __construct($cache, $root) {
 		} else {
 			$this->unjailedRoot = $root;
 		}
+		$this->querySearchHelper = new QuerySearchHelper($this->mimetypeLoader);
 	}
 
 	protected function getRoot() {
@@ -260,7 +262,7 @@ public function search($pattern) {
 			->whereStorageId()
 			->andWhere($query->expr()->orX(
 				$query->expr()->like('path', $query->createNamedParameter($this->getGetUnjailedRoot() . '/%')),
-				$query->expr()->eq('path_hash', $query->createNamedParameter(md5($this->getGetUnjailedRoot()))),
+				$query->expr()->eq('path', $query->createNamedParameter($this->getGetUnjailedRoot())),
 			))
 			->andWhere($query->expr()->iLike('name', $query->createNamedParameter($pattern)));
 
@@ -292,7 +294,7 @@ public function searchByMime($mimetype) {
 			->whereStorageId()
 			->andWhere($query->expr()->orX(
 				$query->expr()->like('path', $query->createNamedParameter($this->getGetUnjailedRoot() . '/%')),
-				$query->expr()->eq('path_hash', $query->createNamedParameter(md5($this->getGetUnjailedRoot()))),
+				$query->expr()->eq('path', $query->createNamedParameter($this->getGetUnjailedRoot())),
 			));
 
 		if (strpos($mimetype, '/')) {
@@ -311,27 +313,21 @@ public function searchByMime($mimetype) {
 		return $this->formatSearchResults($results);
 	}
 
-	public function searchQuery(ISearchQuery $query) {
+	public function searchQuery(ISearchQuery $searchQuery) {
 		if ($this->getGetUnjailedRoot() === '' || $this->getGetUnjailedRoot() === '/') {
-			return parent::searchQuery($query);
+			return parent::searchQuery($searchQuery);
 		}
 
-		$prefixFilter = new SearchComparison(
-			ISearchComparison::COMPARE_LIKE,
-			'path',
-			$this->getGetUnjailedRoot() . '/%'
-		);
-		$rootFilter = new SearchComparison(
-			ISearchComparison::COMPARE_EQUAL,
-			'path',
-			$this->getGetUnjailedRoot()
-		);
-		$operation = new SearchBinaryOperator(
-			ISearchBinaryOperator::OPERATOR_AND,
-			[new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_OR, [$prefixFilter, $rootFilter]) , $query->getSearchOperation()]
-		);
-		$simpleQuery = new SearchQuery($operation, $query->getLimit(), $query->getOffset(), $query->getOrder(), $query->getUser());
-		$results = $this->getCache()->searchQuery($simpleQuery);
+		$query = $this->buildSearchQuery($searchQuery);
+
+		$query->andWhere($query->expr()->orX(
+			$query->expr()->like('path', $query->createNamedParameter($this->getGetUnjailedRoot() . '/%')),
+			$query->expr()->eq('path', $query->createNamedParameter($this->getGetUnjailedRoot())),
+		));
+
+		$result = $query->execute();
+		$results = $this->searchResultToCacheEntries($result);
+		$result->closeCursor();
 		return $this->formatSearchResults($results);
 	}