Merge pull request #53427 from nextcloud/shard-key-hint-partition

fix: allow hinting the shard keys for joined tables

Author: ArtificialOwl

lib/private/DB/QueryBuilder/Partitioned/PartitionedQueryBuilder.php

@@ -444,4 +444,19 @@ public function getSQL() {
 	public function getPartitionCount(): int {
 		return count($this->splitQueries) + 1;
 	}
+
+	public function hintShardKey(string $column, mixed $value, bool $overwrite = false): self {
+		if (str_contains($column, '.')) {
+			[$alias, $column] = explode('.', $column);
+			$partition = $this->getPartition($alias);
+			if ($partition) {
+				$this->splitQueries[$partition->name]->query->hintShardKey($column, $value, $overwrite);
+			} else {
+				parent::hintShardKey($column, $value, $overwrite);
+			}
+		} else {
+			parent::hintShardKey($column, $value, $overwrite);
+		}
+		return $this;
+	}
 }