vendor/pimcore/portal-engine/src/EventSubscriber/SizeEstimationStrategyListener.php line 40

Open in your IDE?
  1. <?php
  3. /**
  4.  * Pimcore
  5.  *
  6.  * This source file is available under following license:
  7.  * - Pimcore Commercial License (PCL)
  8.  *
  9.  *  @copyright  Copyright (c) Pimcore GmbH (http://www.pimcore.org)
  10.  *  @license    http://www.pimcore.org/license     PCL
  11.  */
  13. namespace Pimcore\Bundle\PortalEngineBundle\EventSubscriber;
  15. use Pimcore\Bundle\PortalEngineBundle\Event\DataObject\ExtractMappingEvent;
  16. use Pimcore\Bundle\PortalEngineBundle\Event\DataObject\UpdateIndexDataEvent;
  17. use Pimcore\Bundle\PortalEngineBundle\Service\Download\SizeEstimation\SizeEstimationStrategyInterface;
  18. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  20. class SizeEstimationStrategyListener implements EventSubscriberInterface
  21. {
  22.     protected $sizeEstimationStrategy;
  24.     public function __construct(SizeEstimationStrategyInterface $sizeEstimationStrategy)
  25.     {
  26.         $this->sizeEstimationStrategy = $sizeEstimationStrategy;
  27.     }
  29.     /**
  30.      * @return array
  31.      */
  32.     public static function getSubscribedEvents()
  33.     {
  34.         return [
  35.             ExtractMappingEvent::class => 'onExtractMapping',
  36.             UpdateIndexDataEvent::class => 'onUpdateIndexData'
  37.         ];
  38.     }
  40.     public function onExtractMapping(ExtractMappingEvent $event)
  41.     {
  42.         $mappings = $event->getCustomFieldsMapping();
  44.         $mappings = array_merge($mappings, $this->sizeEstimationStrategy->getCustomDataObjectMappingForIndex($event->getClassDefinition()));
  46.         $event->setCustomFieldsMapping($mappings);
  47.     }
  49.     public function onUpdateIndexData(UpdateIndexDataEvent $event)
  50.     {
  51.         $data = $event->getCustomFields();
  53.         $event->setCustomFields(array_replace($data, $this->sizeEstimationStrategy->getCustomDataObjectDataForIndex($event->getDataObject())));
  54.     }
  55. }