84 lines
2.6 KiB
PHP
84 lines
2.6 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Kimai time-tracking app.
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace App\Repository\Loader;
|
|
|
|
use App\Entity\Team;
|
|
use App\Entity\User;
|
|
use Doctrine\ORM\EntityManagerInterface;
|
|
|
|
final class UserLoader implements LoaderInterface
|
|
{
|
|
public function __construct(private EntityManagerInterface $entityManager, private bool $fullyHydrated = false)
|
|
{
|
|
}
|
|
|
|
/**
|
|
* @param array<int|User> $results
|
|
*/
|
|
public function loadResults(array $results): void
|
|
{
|
|
if (empty($results)) {
|
|
return;
|
|
}
|
|
|
|
$ids = array_map(function ($user) {
|
|
if ($user instanceof User) {
|
|
return $user->getId();
|
|
}
|
|
|
|
return $user;
|
|
}, $results);
|
|
|
|
$em = $this->entityManager;
|
|
|
|
$qb = $em->createQueryBuilder();
|
|
/** @var User[] $users */
|
|
$users = $qb->select('PARTIAL user.{id}', 'memberships', 'team')
|
|
->from(User::class, 'user')
|
|
->leftJoin('user.memberships', 'memberships')
|
|
->leftJoin('memberships.team', 'team')
|
|
->andWhere($qb->expr()->in('user.id', $ids))
|
|
->getQuery()
|
|
->execute();
|
|
|
|
$qb = $em->createQueryBuilder();
|
|
$qb->select('PARTIAL user.{id}', 'preferences')
|
|
->from(User::class, 'user')
|
|
->leftJoin('user.preferences', 'preferences')
|
|
->andWhere($qb->expr()->in('user.id', $ids))
|
|
->getQuery()
|
|
->execute();
|
|
|
|
// do not load team members or leads by default, because they will only be used on detail pages
|
|
// and there is no benefit in adding multiple queries for most requests when they are only needed in one place
|
|
if ($this->fullyHydrated) {
|
|
$teamIds = [];
|
|
foreach ($users as $user) {
|
|
foreach ($user->getTeams() as $team) {
|
|
$teamIds[] = $team->getId();
|
|
}
|
|
}
|
|
$teamIds = array_unique($teamIds);
|
|
|
|
if (\count($teamIds) > 0) {
|
|
$qb = $em->createQueryBuilder();
|
|
/** @var Team[] $teams */
|
|
$teams = $qb->select('PARTIAL team.{id}', 'members', 'user')
|
|
->from(Team::class, 'team')
|
|
->leftJoin('team.members', 'members')
|
|
->leftJoin('members.user', 'user')
|
|
->andWhere($qb->expr()->in('team.id', $teamIds))
|
|
->getQuery()
|
|
->execute();
|
|
}
|
|
}
|
|
}
|
|
}
|