1 files changed, 36 insertions, 0 deletions
diff --git a/lib/private/Files/Search/QueryOptimizer/SplitLargeIn.php b/lib/private/Files/Search/QueryOptimizer/SplitLargeIn.php
new file mode 100644
index 00000000000..8aee1975708
--- /dev/null
+++ b/lib/private/Files/Search/QueryOptimizer/SplitLargeIn.php
@@ -0,0 +1,36 @@
+<?php
+
+/**
+ * SPDX-FileCopyrightText: 2024 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+namespace OC\Files\Search\QueryOptimizer;
+
+use OC\Files\Search\SearchBinaryOperator;
+use OC\Files\Search\SearchComparison;
+use OCP\Files\Search\ISearchBinaryOperator;
+use OCP\Files\Search\ISearchComparison;
+use OCP\Files\Search\ISearchOperator;
+
+/**
+ * transform IN (1000+ element) into (IN (1000 elements) OR IN(...))
+ */
+class SplitLargeIn extends ReplacingOptimizerStep {
+	public function processOperator(ISearchOperator &$operator): bool {
+		if (
+			$operator instanceof ISearchComparison
+			&& $operator->getType() === ISearchComparison::COMPARE_IN
+			&& count($operator->getValue()) > 1000
+		) {
+			$chunks = array_chunk($operator->getValue(), 1000);
+			$chunkComparisons = array_map(function (array $values) use ($operator) {
+				return new SearchComparison(ISearchComparison::COMPARE_IN, $operator->getField(), $values, $operator->getExtra());
+			}, $chunks);
+
+			$operator = new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_OR, $chunkComparisons);
+			return true;
+		}
+		parent::processOperator($operator);
+		return false;
+	}
+}