protected function fetch(QueryBuilder $builder, $hasJoin, array $values, IEntityPreloadContainer $preloadContainer = NULL) { $values = array_values(array_unique(array_filter($values, function ($value) { return $value !== NULL; }))); if (count($values) === 0) { return new EntityContainer([], $preloadContainer); } $primaryKey = $this->targetRepository->getMapper()->getStorageReflection()->getStoragePrimaryKey()[0]; $builder->andWhere('%column IN %any', $primaryKey, $values); $builder->addSelect(($hasJoin ? 'DISTINCT ' : '') . '%table.*', $builder->getFromAlias()); $result = $this->connection->queryArgs($builder->getQuerySQL(), $builder->getQueryParameters()); $entities = []; while ($data = $result->fetch()) { $entity = $this->targetRepository->hydrateEntity($data->toArray()); $entities[$entity->getValue('id')] = $entity; } return new EntityContainer($entities, $preloadContainer); }
private function fetchCounts(QueryBuilder $builder, array $values) { $sourceTable = $builder->getFromAlias(); $targetTable = QueryBuilderHelper::getAlias($this->joinTable); $builder = clone $builder; $builder->leftJoin($sourceTable, '%table', $targetTable, '%column = %column', $this->joinTable, "{$targetTable}.{$this->primaryKeyTo}", "{$sourceTable}." . $this->targetRepository->getMapper()->getStorageReflection()->getStoragePrimaryKey()[0]); $builder->addSelect('%column', "{$targetTable}.{$this->primaryKeyFrom}"); $builder->orderBy(NULL); if ($builder->hasLimitOffsetClause()) { $sqls = []; $args = []; foreach ($values as $value) { $build = clone $builder; $build->andWhere("%column = %any", $this->primaryKeyFrom, $value); $sqls[] = "SELECT %any AS %column, COUNT(*) AS [count] FROM (" . $build->getQuerySql() . ') [temp]'; $args[] = $value; $args[] = $this->primaryKeyFrom; $args = array_merge($args, $build->getQueryParameters()); } $sql = '(' . implode(') UNION ALL (', $sqls) . ')'; $result = $this->connection->queryArgs($sql, $args); } else { $builder->addSelect('COUNT(%column) as count', $this->primaryKeyTo); $builder->andWhere('%column IN %any', $this->primaryKeyFrom, $values); $builder->groupBy('%column', $this->primaryKeyFrom); $result = $this->connection->queryArgs($builder->getQuerySql(), $builder->getQueryParameters()); } $counts = []; foreach ($result as $row) { $counts[$row->{$this->primaryKeyFrom}] = $row->count; } return $counts; }
private function normalizeAndAddJoins(array $levels, $sourceEntity, QueryBuilder $builder, &$distinctNeeded = FALSE, &$value = NULL) { $column = array_pop($levels); $sourceMapper = $this->mapper; $sourceAlias = $builder->getFromAlias(); $sourceReflection = $sourceMapper->getStorageReflection(); $sourceEntityMeta = $this->metadataStorage->get($sourceEntity ?: $sourceMapper->getRepository()->getEntityClassNames()[0]); foreach ($levels as $levelIndex => $level) { $property = $sourceEntityMeta->getProperty($level); if ($property->relationship === NULL) { throw new InvalidArgumentException("Entity {$sourceEntityMeta->className}::\${$level} does not contain a relationship."); } $targetMapper = $this->model->getRepository($property->relationship->repository)->getMapper(); $targetReflection = $targetMapper->getStorageReflection(); $targetEntityMetadata = $this->metadataStorage->get($property->relationship->entity); $relType = $property->relationship->type; if ($relType === Relationship::ONE_HAS_MANY || $relType === Relationship::ONE_HAS_ONE_DIRECTED && !$property->relationship->isMain) { $targetColumn = $targetReflection->convertEntityToStorageKey($property->relationship->property); $sourceColumn = $sourceReflection->getStoragePrimaryKey()[0]; $distinctNeeded = TRUE; } elseif ($relType === Relationship::MANY_HAS_MANY) { if ($property->relationship->isMain) { list($joinTable, list($inColumn, $outColumn)) = $sourceMapper->getManyHasManyParameters($property, $targetMapper); } else { $sourceProperty = $targetEntityMetadata->getProperty($property->relationship->property); list($joinTable, list($outColumn, $inColumn)) = $targetMapper->getManyHasManyParameters($sourceProperty, $sourceMapper); } $sourceColumn = $sourceReflection->getStoragePrimaryKey()[0]; $builder->leftJoin($sourceAlias, $joinTable, self::getAlias($joinTable), "[{$sourceAlias}.{$sourceColumn}] = [{$joinTable}.{$inColumn}]"); $sourceAlias = $joinTable; $sourceColumn = $outColumn; $targetColumn = $targetReflection->getStoragePrimaryKey()[0]; $distinctNeeded = TRUE; } else { $targetColumn = $targetReflection->getStoragePrimaryKey()[0]; $sourceColumn = $sourceReflection->convertEntityToStorageKey($level); } $targetTable = $targetMapper->getTableName(); $targetAlias = $level . str_repeat('_', $levelIndex); $builder->leftJoin($sourceAlias, $targetTable, $targetAlias, "[{$sourceAlias}.{$sourceColumn}] = [{$targetAlias}.{$targetColumn}]"); $sourceAlias = $targetAlias; $sourceMapper = $targetMapper; $sourceReflection = $targetReflection; $sourceEntityMeta = $targetEntityMetadata; } $sourceEntityMeta->getProperty($column); // check if property exists if ($column === 'id' && count($sourceEntityMeta->getPrimaryKey()) > 1) { $pair = []; foreach ($sourceEntityMeta->getPrimaryKey() as $column) { $column = $sourceReflection->convertEntityToStorageKey($column); $pair[] = "{$sourceAlias}.{$column}"; } if (!isset($value[0][0])) { $value = [$value]; } return '(' . implode(', ', $pair) . ')'; } else { $column = $sourceReflection->convertEntityToStorageKey($column); return "{$sourceAlias}.{$column}"; } }
private function normalizeAndAddJoins(array $levels, $sourceEntity, QueryBuilder $builder, &$distinctNeeded = false, &$value = null, &$modifier = '%any') { $column = array_pop($levels); $sourceMapper = $this->mapper; $sourceAlias = $builder->getFromAlias(); $sourceReflection = $sourceMapper->getStorageReflection(); $sourceEntityMeta = $this->metadataStorage->get($sourceEntity ?: $sourceMapper->getRepository()->getEntityClassNames()[0]); foreach ($levels as $levelIndex => $level) { $property = $sourceEntityMeta->getProperty($level); if ($property->relationship === null) { throw new InvalidArgumentException("Entity {$sourceEntityMeta->className}::\${$level} does not contain a relationship."); } $targetMapper = $this->model->getRepository($property->relationship->repository)->getMapper(); $targetReflection = $targetMapper->getStorageReflection(); $targetEntityMetadata = $this->metadataStorage->get($property->relationship->entity); $relType = $property->relationship->type; if ($relType === Relationship::ONE_HAS_MANY || $relType === Relationship::ONE_HAS_ONE && !$property->relationship->isMain) { $targetColumn = $targetReflection->convertEntityToStorageKey($property->relationship->property); $sourceColumn = $sourceReflection->getStoragePrimaryKey()[0]; $distinctNeeded = $relType === Relationship::ONE_HAS_MANY; } elseif ($relType === Relationship::MANY_HAS_MANY) { if ($property->relationship->isMain) { list($joinTable, list($inColumn, $outColumn)) = $sourceMapper->getManyHasManyParameters($property, $targetMapper); } else { $sourceProperty = $targetEntityMetadata->getProperty($property->relationship->property); list($joinTable, list($outColumn, $inColumn)) = $targetMapper->getManyHasManyParameters($sourceProperty, $sourceMapper); } $sourceColumn = $sourceReflection->getStoragePrimaryKey()[0]; $builder->leftJoin($sourceAlias, $joinTable, self::getAlias($joinTable), "[{$sourceAlias}.{$sourceColumn}] = [{$joinTable}.{$inColumn}]"); $sourceAlias = $joinTable; $sourceColumn = $outColumn; $targetColumn = $targetReflection->getStoragePrimaryKey()[0]; $distinctNeeded = true; } else { $targetColumn = $targetReflection->getStoragePrimaryKey()[0]; $sourceColumn = $sourceReflection->convertEntityToStorageKey($level); } $targetTable = $targetMapper->getTableName(); $targetAlias = $level . str_repeat('_', $levelIndex); $builder->leftJoin($sourceAlias, $targetTable, $targetAlias, "[{$sourceAlias}.{$sourceColumn}] = [{$targetAlias}.{$targetColumn}]"); $sourceAlias = $targetAlias; $sourceMapper = $targetMapper; $sourceReflection = $targetReflection; $sourceEntityMeta = $targetEntityMetadata; } $targetProperty = $sourceEntityMeta->getProperty($column); if ($targetProperty->isPrimary && $targetProperty->isVirtual) { // primary-proxy $primaryKey = $sourceEntityMeta->getPrimaryKey(); if (count($primaryKey) > 1) { // composite primary key $modifier = '%any'; list($expression, $value) = $this->processMultiColumn($sourceReflection, $primaryKey, $value, $sourceAlias); return $expression; } else { $column = reset($primaryKey); } } list($expression, $modifier, $value) = $this->processColumn($sourceReflection, $column, $value, $sourceAlias); return $expression; }
private function fetchCounts(QueryBuilder $builder, array $values) { $targetStoragePrimaryKey = $this->targetMapper->getStorageReflection()->getStoragePrimaryKey()[0]; $sourceTable = $builder->getFromAlias(); $builder = clone $builder; $builder->addSelect("{$sourceTable}.{$this->joinStorageKey}"); $builder->orderBy(NULL); if ($builder->hasLimitOffsetClause()) { $sqls = []; $args = []; foreach ($values as $value) { $build = clone $builder; $build->andWhere("{$sourceTable}.{$this->joinStorageKey} = %any", $value); $sqls[] = "SELECT {$value} as {$this->joinStorageKey}, COUNT(*) AS count FROM (" . $build->getQuerySql() . ') temp'; $args = array_merge($args, $build->getQueryParameters()); } $sql = '(' . implode(') UNION ALL (', $sqls) . ')'; $result = $this->connection->queryArgs($sql, $args); } else { $builder->addSelect("COUNT({$sourceTable}.{$targetStoragePrimaryKey}) AS count"); $builder->andWhere("{$sourceTable}.{$this->joinStorageKey} IN %any", $values); $builder->groupBy("{$sourceTable}.{$this->joinStorageKey}"); $result = $this->connection->queryArgs($builder->getQuerySql(), $builder->getQueryParameters()); } $counts = []; foreach ($result as $row) { $counts[$row->{$this->joinStorageKey}] = $row->count; } return $counts; }