vendor/pimcore/pimcore/models/DataObject/Listing/Dao.php line 73

Open in your IDE?
  1. <?php
  3. /**
  4.  * Pimcore
  5.  *
  6.  * This source file is available under two different licenses:
  7.  * - GNU General Public License version 3 (GPLv3)
  8.  * - Pimcore Commercial License (PCL)
  9.  * Full copyright and license information is available in
  10.  * LICENSE.md which is distributed with this source code.
  11.  *
  12.  *  @copyright  Copyright (c) Pimcore GmbH (http://www.pimcore.org)
  13.  *  @license    http://www.pimcore.org/license     GPLv3 and PCL
  14.  */
  16. namespace Pimcore\Model\DataObject\Listing;
  18. use Doctrine\DBAL\Query\QueryBuilder as DoctrineQueryBuilder;
  19. use Pimcore\Model;
  20. use Pimcore\Model\DataObject;
  21. use Pimcore\Model\Listing\Dao\QueryBuilderHelperTrait;
  23. /**
  24.  * @internal
  25.  *
  26.  * @property \Pimcore\Model\DataObject\Listing $model
  27.  */
  28. class Dao extends Model\Listing\Dao\AbstractDao
  29. {
  30.     use QueryBuilderHelperTrait;
  32.     /**
  33.      * @return string
  34.      */
  35.     public function getTableName()
  36.     {
  37.         return 'objects';
  38.     }
  40.     /**
  41.      * @param string|string[]|null $columns
  42.      *
  43.      * @return DoctrineQueryBuilder
  44.      *
  45.      * @throws \Exception
  46.      */
  47.     public function getQueryBuilder(...$columns): DoctrineQueryBuilder
  48.     {
  49.         $queryBuilder = $this->db->createQueryBuilder();
  50.         $queryBuilder->select(...$columns)->from($this->getTableName());
  52.         // apply joins
  53.         $this->applyJoins($queryBuilder);
  55.         $this->applyListingParametersToQueryBuilder($queryBuilder);
  57.         return $queryBuilder;
  58.     }
  60.     /**
  61.      * Loads a list of objects for the specicifies parameters, returns an array of DataObject\AbstractObject elements
  62.      *
  63.      * @return array
  64.      */
  65.     public function load()
  66.     {
  68.         // load id's
  69.         $list = $this->loadIdList();
  71.         $objects = [];
  72.         foreach ($list as $o_id) {
  73.             if ($object = DataObject::getById($o_id)) {
  74.                 $objects[] = $object;
  75.             }
  76.         }
  78.         $this->model->setObjects($objects);
  80.         return $objects;
  81.     }
  83.     /**
  84.      * @return int
  85.      */
  86.     public function getTotalCount()
  87.     {
  88.         $queryBuilder = $this->getQueryBuilder();
  89.         $this->prepareQueryBuilderForTotalCount($queryBuilder);
  91.         $totalCount = $this->db->fetchOne((string) $queryBuilder, $this->model->getConditionVariables(), $this->model->getConditionVariableTypes());
  93.         return (int) $totalCount;
  94.     }
  96.     /**
  97.      * @return int
  98.      */
  99.     public function getCount()
  100.     {
  101.         if ($this->model->isLoaded()) {
  102.             return count($this->model->getObjects());
  103.         } else {
  104.             $idList = $this->loadIdList();
  106.             return count($idList);
  107.         }
  108.     }
  110.     /**
  111.      * Loads a list of document ids for the specicifies parameters, returns an array of ids
  112.      *
  113.      * @return int[]
  114.      */
  115.     public function loadIdList()
  116.     {
  117.         $queryBuilder = $this->getQueryBuilder([sprintf('%s as o_id', $this->getTableName() . '.o_id'), sprintf('%s as o_type', $this->getTableName() . '.o_type')]);
  118.         $objectIds = $this->db->fetchCol((string) $queryBuilder, $this->model->getConditionVariables(), $this->model->getConditionVariableTypes());
  120.         return array_map('intval', $objectIds);
  121.     }
  123.     /**
  124.      * @param DoctrineQueryBuilder $queryBuilder
  125.      *
  126.      * @return $this
  127.      */
  128.     protected function applyJoins(DoctrineQueryBuilder $queryBuilder)
  129.     {
  130.         return $this;
  131.     }
  132. }