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

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.     {
  67.         // load id's
  68.         $list = $this->loadIdList();
  70.         $objects = [];
  71.         foreach ($list as $o_id) {
  72.             if ($object = DataObject::getById($o_id)) {
  73.                 $objects[] = $object;
  74.             }
  75.         }
  77.         $this->model->setObjects($objects);
  79.         return $objects;
  80.     }
  82.     /**
  83.      * @return int
  84.      */
  85.     public function getTotalCount()
  86.     {
  87.         $queryBuilder = $this->getQueryBuilder();
  88.         $this->prepareQueryBuilderForTotalCount($queryBuilder, $this->getTableName() . '.o_id');
  90.         $totalCount = $this->db->fetchOne((string) $queryBuilder, $this->model->getConditionVariables(), $this->model->getConditionVariableTypes());
  92.         return (int) $totalCount;
  93.     }
  95.     /**
  96.      * @return int
  97.      */
  98.     public function getCount()
  99.     {
  100.         if ($this->model->isLoaded()) {
  101.             return count($this->model->getObjects());
  102.         } else {
  103.             $idList = $this->loadIdList();
  105.             return count($idList);
  106.         }
  107.     }
  109.     /**
  110.      * Loads a list of document ids for the specicifies parameters, returns an array of ids
  111.      *
  112.      * @return int[]
  113.      */
  114.     public function loadIdList()
  115.     {
  116.         $queryBuilder = $this->getQueryBuilder([sprintf('%s as o_id', $this->getTableName() . '.o_id'), sprintf('%s as o_type', $this->getTableName() . '.o_type')]);
  117.         $objectIds = $this->db->fetchFirstColumn((string) $queryBuilder, $this->model->getConditionVariables(), $this->model->getConditionVariableTypes());
  119.         return array_map('intval', $objectIds);
  120.     }
  122.     /**
  123.      * @param DoctrineQueryBuilder $queryBuilder
  124.      *
  125.      * @return $this
  126.      */
  127.     protected function applyJoins(DoctrineQueryBuilder $queryBuilder)
  128.     {
  129.         return $this;
  130.     }
  131. }