vendor/pimcore/pimcore/models/Document/Listing/Dao.php line 37

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\Document\Listing;
  18. use Doctrine\DBAL\Query\QueryBuilder as DoctrineQueryBuilder;
  19. use Pimcore\Model;
  20. use Pimcore\Model\Document;
  21. use Pimcore\Model\Listing\Dao\QueryBuilderHelperTrait;
  23. /**
  24.  * @internal
  25.  *
  26.  * @property \Pimcore\Model\Document\Listing $model
  27.  */
  28. class Dao extends Model\Listing\Dao\AbstractDao
  29. {
  30.     use QueryBuilderHelperTrait;
  32.     /**
  33.      * Loads a list of objects (all are an instance of Document) for the given parameters an return them
  34.      *
  35.      * @return array
  36.      */
  37.     public function load()
  38.     {
  39.         $documents = [];
  40.         $select = $this->getQueryBuilder(['documents.id', 'documents.type']);
  42.         $documentsData = $this->db->fetchAllAssociative((string) $select, $this->model->getConditionVariables(), $this->model->getConditionVariableTypes());
  44.         foreach ($documentsData as $documentData) {
  45.             if ($documentData['type']) {
  46.                 if ($doc = Document::getById($documentData['id'])) {
  47.                     $documents[] = $doc;
  48.                 }
  49.             }
  50.         }
  52.         $this->model->setDocuments($documents);
  54.         return $documents;
  55.     }
  57.     /**
  58.      * @param string|string[]|null $columns
  59.      *
  60.      * @return DoctrineQueryBuilder
  61.      */
  62.     public function getQueryBuilder(...$columns): DoctrineQueryBuilder
  63.     {
  64.         $queryBuilder = $this->db->createQueryBuilder();
  65.         $queryBuilder->select(...$columns)->from('documents');
  67.         $this->applyListingParametersToQueryBuilder($queryBuilder);
  69.         return $queryBuilder;
  70.     }
  72.     /**
  73.      * Loads a list of document ids for the specicifies parameters, returns an array of ids
  74.      *
  75.      * @return int[]
  76.      */
  77.     public function loadIdList()
  78.     {
  79.         $queryBuilder = $this->getQueryBuilder(['documents.id']);
  80.         $documentIds = $this->db->fetchFirstColumn((string) $queryBuilder, $this->model->getConditionVariables(), $this->model->getConditionVariableTypes());
  82.         return array_map('intval', $documentIds);
  83.     }
  85.     /**
  86.      * @return array
  87.      */
  88.     public function loadIdPathList()
  89.     {
  90.         $queryBuilder = $this->getQueryBuilder(['documents.id', 'CONCAT(documents.path, documents.key) as path']);
  91.         $documentIds = $this->db->fetchAllAssociative((string) $queryBuilder, $this->model->getConditionVariables(), $this->model->getConditionVariableTypes());
  93.         return $documentIds;
  94.     }
  96.     /**
  97.      * @return int
  98.      */
  99.     public function getCount()
  100.     {
  101.         if ($this->model->isLoaded()) {
  102.             return count($this->model->getDocuments());
  103.         } else {
  104.             $idList = $this->loadIdList();
  106.             return count($idList);
  107.         }
  108.     }
  110.     /**
  111.      * @return int
  112.      */
  113.     public function getTotalCount()
  114.     {
  115.         $queryBuilder = $this->getQueryBuilder();
  116.         $this->prepareQueryBuilderForTotalCount($queryBuilder, 'documents.id');
  118.         $amount = (int) $this->db->fetchOne((string) $queryBuilder, $this->model->getConditionVariables(), $this->model->getConditionVariableTypes());
  120.         return $amount;
  121.     }
  122. }