relationships: fixed counting entities when multiplied by joins [closes

#393]
nextras · Apr 10, 2020 · 8ac61d4 · 8ac61d4
1 parent 3ff59e1
commit 8ac61d4
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 2 deletions.
diff --git a/src/Mapper/Dbal/RelationshipMapperManyHasMany.php b/src/Mapper/Dbal/RelationshipMapperManyHasMany.php
@@ -217,7 +217,7 @@ private function fetchCounts(QueryBuilder $builder, array $values)
 			$result = $this->processMultiCountResult($builder, $values);
 
 		} else {
-			$builder->addSelect('COUNT(%column) as count', "$targetTable.$this->primaryKeyTo");
+			$builder->addSelect('COUNT(DISTINCT %column) AS [count]', "$targetTable.$this->primaryKeyTo");
 			$builder->orderBy(null);
 			$builder->andWhere('%column IN %any', "$targetTable.$this->primaryKeyFrom", $values);
 			$builder->groupBy('%column', "$targetTable.$this->primaryKeyFrom");

diff --git a/src/Mapper/Dbal/RelationshipMapperOneHasMany.php b/src/Mapper/Dbal/RelationshipMapperOneHasMany.php
@@ -229,7 +229,7 @@ private function fetchCounts(QueryBuilder $builder, array $values)
 
 		} else {
 			$builder->orderBy(null);
-			$builder->addSelect('COUNT(%column) AS [count]', "{$sourceTable}.{$targetStoragePrimaryKey}");
+			$builder->addSelect('COUNT(DISTINCT %column) AS [count]', "{$sourceTable}.{$targetStoragePrimaryKey}");
 			$builder->andWhere('%column IN %any', "{$sourceTable}.{$this->joinStorageKey}", $values);
 			$builder->groupBy('%column', "{$sourceTable}.{$this->joinStorageKey}");
 			$result = $this->connection->queryArgs($builder->getQuerySql(), $builder->getQueryParameters());

diff --git a/tests/cases/integration/Relationships/relationships.manyHasMany.phpt b/tests/cases/integration/Relationships/relationships.manyHasMany.phpt
@@ -277,6 +277,21 @@ class RelationshipManyHasManyTest extends DataTestCase
 
 		Assert::same(1, \count($tag->books));
 	}
+
+
+	public function testCountStoredOnManyHasManyRelationshipCondition()
+	{
+		$tag = $this->orm->tags->getById(1);
+		$books = $tag->books->get()->findBy([
+			'this->author->id' => 1,
+		]);
+		Assert::same(1, $books->countStored());
+
+		$books = $tag->books->get()->findBy([
+			'this->author->tagFollowers->author->id' => 1,
+		]);
+		Assert::same(1, $books->countStored());
+	}
 }
 
 

diff --git a/tests/cases/integration/Relationships/relationships.oneHasMany.phpt b/tests/cases/integration/Relationships/relationships.oneHasMany.phpt
@@ -284,6 +284,23 @@ class RelationshipOneHasManyTest extends DataTestCase
 
 		Assert::same(1, \count($author->books));
 	}
+
+
+	public function testCountStoredOnOneHasManyRelationshipCondition()
+	{
+		$publisher = $this->orm->publishers->getById(1);
+		$books = $publisher->books->get()->findBy([
+			'this->tags->id' => 1,
+		]);
+		Assert::same(1, $books->countStored());
+
+		$books = $publisher->books->get()->findBy([
+			ICollection::OR,
+			'this->title' => 'Book 1',
+			'this->tags->id' => 1,
+		]);
+		Assert::same(1, $books->countStored());
+	}
 }