src/Repository/Common/CompanyRepository.php line 21

Open in your IDE?
  1. <?php
  3. namespace Harmonizely\Repository\Common;
  5. use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
  6. use Doctrine\ORM\OptimisticLockException;
  7. use Doctrine\ORM\ORMException;
  8. use Doctrine\Persistence\ManagerRegistry;
  9. use Harmonizely\Entity\Common\CompanyEntity;
  10. use Harmonizely\Repository\Common\Contract\ICompanyRepository;
  11. use Harmonizely\Types\Common\CompanyStatus;
  13. class CompanyRepository extends ServiceEntityRepository implements ICompanyRepository
  14. {
  16.     /**
  17.      * @param ManagerRegistry $registry
  18.      */
  19.     public function __construct(ManagerRegistry $registry)
  20.     {
  21.         parent::__construct($registry, CompanyEntity::class);
  22.     }
  24.     /**
  25.      * @param CompanyEntity $entity
  26.      * @return CompanyEntity
  27.      * @throws ORMException
  28.      * @throws OptimisticLockException
  29.      */
  30.     public function save(CompanyEntity $entity): CompanyEntity
  31.     {
  32.         $this->getEntityManager()->persist($entity);
  33.         $this->getEntityManager()->flush();
  35.         return $entity;
  36.     }
  38.     /**
  39.      * @param string $login
  40.      * @return CompanyEntity|null
  41.      */
  42.     public function getCompanyByLogin(string $login): ?CompanyEntity
  43.     {
  44.         return $this->findOneBy(['login' => $login]);
  45.     }
  47.     /**
  48.      * @return array|string[]
  49.      */
  50.     public function getActiveCompanyLogins(): array
  51.     {
  52.         $qb = $this->createQueryBuilder('company');
  53.         $qb
  54.             ->select('company.login')
  55.             ->where('company.status = :status')
  56.             ->setParameter('status', CompanyStatus::STATUS_ACTIVE);
  58.         $rows = $qb->getQuery()->getScalarResult();
  59.         $res = [];
  60.         foreach ($rows as $row) {
  61.             $res[] = $row['login'];
  62.         }
  64.         return $res;
  65.     }
  67.     /**
  68.      * @return array|string[]
  69.      */
  70.     public function getAllCompanyLogins(): array
  71.     {
  72.         $qb = $this->createQueryBuilder('company');
  73.         $qb
  74.             ->select('company.login')
  75.             ->orderBy('company.id', 'DESC');
  77.         $rows = $qb->getQuery()->getScalarResult();
  78.         $res = [];
  79.         foreach ($rows as $row) {
  80.             $res[] = $row['login'];
  81.         }
  83.         return $res;
  84.     }
  85. }