From 26903926a2eed87d6a1664df15eb221426e97dd3 Mon Sep 17 00:00:00 2001 From: Robin Appelman Date: Tue, 10 Jun 2025 16:15:05 +0200 Subject: fix: allow hinting the shard keys for joined tables Signed-off-by: Robin Appelman --- .../QueryBuilder/Partitioned/PartitionedQueryBuilder.php | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'lib/private/DB/QueryBuilder/Partitioned') diff --git a/lib/private/DB/QueryBuilder/Partitioned/PartitionedQueryBuilder.php b/lib/private/DB/QueryBuilder/Partitioned/PartitionedQueryBuilder.php index 2942eeccdf7..8b051561a2e 100644 --- a/lib/private/DB/QueryBuilder/Partitioned/PartitionedQueryBuilder.php +++ b/lib/private/DB/QueryBuilder/Partitioned/PartitionedQueryBuilder.php @@ -444,4 +444,19 @@ class PartitionedQueryBuilder extends ShardedQueryBuilder { 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; + } } -- cgit v1.2.3