vendor/pimcore/pimcore/models/Asset/Listing/Dao.php line 36

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\Asset\Listing;
  18. use Doctrine\DBAL\Query\QueryBuilder as DoctrineQueryBuilder;
  19. use Pimcore\Model;
  20. use Pimcore\Model\Listing\Dao\QueryBuilderHelperTrait;
  22. /**
  23.  * @internal
  24.  *
  25.  * @property \Pimcore\Model\Asset\Listing $model
  26.  */
  27. class Dao extends Model\Listing\Dao\AbstractDao
  28. {
  29.     use QueryBuilderHelperTrait;
  31.     /**
  32.      * Get the assets from database
  33.      *
  34.      * @return array
  35.      */
  36.     public function load()
  37.     {
  38.         $assets = [];
  40.         $queryBuilder = $this->getQueryBuilder(['assets.id', 'assets.type']);
  41.         $assetsData = $this->db->fetchAllAssociative((string) $queryBuilder, $this->model->getConditionVariables(), $this->model->getConditionVariableTypes());
  43.         foreach ($assetsData as $assetData) {
  44.             if ($assetData['type']) {
  45.                 if ($asset = Model\Asset::getById($assetData['id'])) {
  46.                     $assets[] = $asset;
  47.                 }
  48.             }
  49.         }
  51.         $this->model->setAssets($assets);
  53.         return $assets;
  54.     }
  56.     /**
  57.      * @param string|string[]|null $columns
  58.      *
  59.      * @return DoctrineQueryBuilder
  60.      */
  61.     public function getQueryBuilder(...$columns): DoctrineQueryBuilder
  62.     {
  63.         $queryBuilder = $this->db->createQueryBuilder();
  64.         $queryBuilder->select(...$columns)->from('assets');
  66.         $this->applyListingParametersToQueryBuilder($queryBuilder);
  68.         return $queryBuilder;
  69.     }
  71.     /**
  72.      * Loads a list of document IDs for the specified parameters, returns an array of ids
  73.      *
  74.      * @return int[]
  75.      */
  76.     public function loadIdList()
  77.     {
  78.         $queryBuilder = $this->getQueryBuilder(['assets.id']);
  79.         $assetIds = $this->db->fetchFirstColumn((string) $queryBuilder, $this->model->getConditionVariables(), $this->model->getConditionVariableTypes());
  81.         return array_map('intval', $assetIds);
  82.     }
  84.     /**
  85.      * @return int
  86.      */
  87.     public function getCount()
  88.     {
  89.         if ($this->model->isLoaded()) {
  90.             return count($this->model->getAssets());
  91.         } else {
  92.             $idList = $this->loadIdList();
  94.             return count($idList);
  95.         }
  96.     }
  98.     /**
  99.      * @return int
  100.      */
  101.     public function getTotalCount()
  102.     {
  103.         $queryBuilder = $this->getQueryBuilder();
  104.         $this->prepareQueryBuilderForTotalCount($queryBuilder, 'assets.id');
  106.         $amount = (int) $this->db->fetchOne((string) $queryBuilder, $this->model->getConditionVariables(), $this->model->getConditionVariableTypes());
  108.         return $amount;
  109.     }
  110. }